def test_classmethod(self, mock_get_roll_call):
mock_get_roll_call.return_value = [
Car('Ford'),
]
self.assertEquals(get_roll_call(), [
Car('Ford'),
])
def test_context_manager(self, mock_open_car):
car = Car()
car.closed = 'Foo'
def enter_car(car):
car.closed = 'Bar'
return car
mock_open_car.return_value.__enter__ = enter_car
#mock_open_car.return_value.__exit__ = exit_car
states = close_car(car)
self.assertEquals(states, ['Bar', 'Bar'])
def test_context_manager(self, mock_open_car):
car = Car()
car.closed = 'Foo'
def enter_car(car):
car.closed = 'Bar'
return car
mock_open_car.return_value.__enter__ = enter_car
#mock_open_car.return_value.__exit__ = exit_car
states = close_car(car)
self.assertEquals(states, ['Bar', 'Bar'])
def get_roll_call():
return Car.roll_call()
def get_car_wheels():
return Car().wheels
def get_car_make(make=None):
car = Car()
if make:
car = Car.for_make(make)
return car.get_make()
def get_roll_call():
return Car.roll_call()
def get_car_make(make=None):
car = Car()
if make:
car = Car.for_make(make)
return car.get_make()
