예제 #1
0
파일: test_model.py 프로젝트: masom/Puck
    def testNew(self):
        mc = ModelCollection()

        with self.assertRaises(TypeError):
            mc.new()

        mc._model = MockModel
        self.assertIsInstance(mc.new(), MockModel)
예제 #2
0
파일: test_model.py 프로젝트: masom/Puck
    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'])