class test_model_group(unittest.TestCase):
def setUp(self):
self.mg = ModelGroup()
self.om = self.mg.create_multi_observer(MockMultiObserver)
self.os = self.mg.create_single_observer(MockSingleObserver)
def test_a_add_model(self):
a1 = BlahModel()
self.mg.add_model(a1)
for a in a1.__model_attrs_map__:
self.assert_(self.om in a1.__model_observers__[a])
def test_b_preadd_model(self):
a1 = BlahModel()
mg = ModelGroup()
mg.add_model(a1)
om = mg.create_multi_observer(MockMultiObserver)
for a in a1.__model_attrs_map__:
self.assert_(om in a1.__model_observers__[a])
def test_c_set_current(self):
a1 = BlahModel()
om = self.mg.create_multi_observer(MockMultiObserver)
self.mg.add_model(a1)
self.mg.set_current(a1)
self.assertEqual(a1, self.om._current)
self.assertEqual(a1, om._current)
def test_d_set_model(self):
a1 = BlahModel()
os = self.mg.create_single_observer(MockSingleObserver)
self.mg.add_model(a1)
self.mg.set_current(a1)
self.assertEqual(a1, self.os._model)
self.assertEqual(a1, os._model)
class test_model_group(unittest.TestCase):
def setUp(self):
self.mg = ModelGroup()
self.om = self.mg.create_multi_observer(MockMultiObserver)
self.os = self.mg.create_single_observer(MockSingleObserver)
def test_a_add_model(self):
a1 = BlahModel()
self.mg.add_model(a1)
for a in a1.__model_attrs_map__:
self.assert_(self.om in a1.__model_observers__[a])
def test_b_preadd_model(self):
a1 = BlahModel()
mg = ModelGroup()
mg.add_model(a1)
om = mg.create_multi_observer(MockMultiObserver)
for a in a1.__model_attrs_map__:
self.assert_(om in a1.__model_observers__[a])
def test_c_set_current(self):
a1 = BlahModel()
om = self.mg.create_multi_observer(MockMultiObserver)
self.mg.add_model(a1)
self.mg.set_current(a1)
self.assertEqual(a1, self.om._current)
self.assertEqual(a1, om._current)
def test_d_set_model(self):
a1 = BlahModel()
os = self.mg.create_single_observer(MockSingleObserver)
self.mg.add_model(a1)
self.mg.set_current(a1)
self.assertEqual(a1, self.os._model)
self.assertEqual(a1, os._model)
