def testNew(self): mc = ModelCollection() with self.assertRaises(TypeError): mc.new() mc._model = MockModel self.assertIsInstance(mc.new(), MockModel)
def test_Build(self): mc = ModelCollection() items = [{'name':"hello"}, {'name':"world"}] with self.assertRaises(TypeError): mc._build(items) mc._model = MockModel entities = mc._build(items) for entity in entities: self.assertTrue(hasattr(entity, 'name')) self.assertTrue(entity.name in ['hello', 'world'])