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)
