class JomFactoryTestCase(TestCase): def setUp(self): self.factory = JomFactory() def tearDown(self): self.factory = None def testRegisterJomDescriptor(self): descriptor = self.factory.register(SimpleModelJomDescriptor) self.assertTrue(isinstance(descriptor, SimpleModelJomDescriptor)) def testGetForName(self): name = SimpleModel.__name__ descriptor = self.factory.register(SimpleModelJomDescriptor) self.assertEqual(descriptor, self.factory.getForName(name)) def testGetForModel(self): descriptor = self.factory.register(SimpleModelJomDescriptor) self.assertEqual(descriptor, self.factory.getForModel(SimpleModel)) def testGetJomInstance(self): instance = SimpleModel.objects.create(name = "foo") descriptor = self.factory.register(SimpleModelJomDescriptor) self.assertEqual(descriptor, self.factory.getJomInstance(instance).descriptor) instance.delete() def testGetJomClass(self): descriptor = self.factory.register(SimpleModelJomDescriptor) self.assertEqual(descriptor, self.factory.getJomClass(SimpleModel).descriptor)