def test_init_add_garage_instance(self):
cesar = Cesar(rnd_name())
garage = Garage(rnd_town(), rnd_int())
for value in [1000, 'some string', [garage]]:
with self.assertRaises(TypeError):
cesar.add_garage(value)
def test_init_add_garage(self):
garage1 = Garage(rnd_town(), rnd_int())
garage2 = Garage(rnd_town(), rnd_int())
cesar = Cesar(rnd_name(), [garage1])
self.assertFalse(cesar.add_garage(garage1))
self.assertTrue(cesar.add_garage(garage2))
self.assertEqual(cesar.register_id, garage1.owner)
self.assertEqual(cesar.register_id, garage2.owner)
self.assertTrue(garage1.number in cesar.garages)
self.assertTrue(garage2.number in cesar.garages)