class ModelInspectTest(TestCase):
def setUp(self):
self.om = OtherModel.objects.create()
ctype = ContentType.objects.get_for_model(OtherModel)
self.mti = ModelToInspect.objects.create(foreign=self.om, one=self.om, content_type=ctype)
self.mti.many.add(self.om)
self.lm = LinkedModel.objects.create(toinspect=self.mti)
self.im = InspectModel(self.mti)
def test_fields(self):
# 21 fields + the automatically generated id field
self.assertEqual(len(self.im.fields), 22)
self.assertFalse("attribute" in self.im.fields)
self.assertFalse("_hidden" in self.im.fields)
def test_relation_fields(self):
# 2 'local' fields + a OneToOneField on LinkedModel
self.assertEqual(len(self.im.relation_fields), 5)
self.assertTrue("foreign" in self.im.relation_fields)
self.assertTrue("content_type" in self.im.relation_fields)
self.assertTrue("genericforeign" in self.im.relation_fields)
self.assertTrue("linkedmodel" in self.im.relation_fields)
self.assertTrue("one" in self.im.relation_fields)
self.assertFalse("many" in self.im.relation_fields)
def test_many_fields(self):
# 1 local + 1 on the ManyRelatedModel
self.assertEqual(len(self.im.many_fields), 2)
self.assertTrue("manyrelatedmodel_set" in self.im.many_fields)
self.assertTrue("many" in self.im.many_fields)
self.assertFalse("one" in self.im.many_fields)
def test_attributes(self):
self.assertEqual(len(self.im.attributes), 1)
def test_properties(self):
self.assertEqual(len(self.im.properties), 2)
def test_methods(self):
self.assertEqual(len(self.im.methods), 2)
self.assertFalse("method_args" in self.im.methods)
self.assertFalse("_hidden_method" in self.im.methods)
def test_items(self):
# make sure all the items are indeed part of a ModelToInspect instance
items = [getattr(self.mti, f) for f in self.im.items]
self.assertEqual(len(items), 34)
def test_multiple_calls(self):
"""Multiple calls to get_FOO"""
self.im.update_fields()
self.assertEqual(len(self.im.fields), 22)
self.assertEqual(len(self.im.relation_fields), 5)
self.assertEqual(len(self.im.many_fields), 2)
self.im.update_attributes()
self.assertEqual(len(self.im.attributes), 1)
self.im.update_methods()
self.assertEqual(len(self.im.methods), 2)
self.assertEqual(len(self.im.items), 34)
class ModelInspectTest(TestCase):
def setUp(self):
self.om = OtherModel.objects.create()
ctype = ContentType.objects.get_for_model(OtherModel)
self.mti = ModelToInspect.objects.create(
foreign=self.om, one=self.om, content_type=ctype)
self.mti.many.add(self.om)
self.lm = LinkedModel.objects.create(toinspect=self.mti)
self.im = InspectModel(self.mti)
def test_fields(self):
# 21 fields + the automatically generated id field
self.assertEqual(len(self.im.fields), 22)
self.assertFalse('attribute' in self.im.fields)
self.assertFalse('_hidden' in self.im.fields)
def test_relation_fields(self):
# 2 'local' fields + a OneToOneField on LinkedModel + GenericFK + CT.
self.assertEqual(len(self.im.relation_fields), 5)
self.assertTrue('foreign' in self.im.relation_fields)
self.assertTrue('content_type' in self.im.relation_fields)
self.assertTrue('genericforeign' in self.im.relation_fields)
self.assertTrue('linkedmodel' in self.im.relation_fields)
self.assertTrue('one' in self.im.relation_fields)
self.assertFalse('many' in self.im.relation_fields)
def test_many_fields(self):
# 1 local + 1 on the ManyRelatedModel
self.assertEqual(len(self.im.many_fields), 2)
self.assertTrue('manyrelatedmodel_set' in self.im.many_fields)
self.assertTrue('many' in self.im.many_fields)
self.assertFalse('one' in self.im.many_fields)
def test_attributes(self):
self.assertEqual(len(self.im.attributes), 1)
def test_properties(self):
self.assertEqual(len(self.im.properties), 2)
def test_methods(self):
self.assertEqual(len(self.im.methods), 2)
self.assertFalse('method_args' in self.im.methods)
self.assertFalse('_hidden_method' in self.im.methods)
def test_items(self):
# make sure all the items are indeed part of a ModelToInspect instance
items = [getattr(self.mti, f) for f in self.im.items]
self.assertEqual(len(items), 34)
def test_multiple_calls(self):
"""Multiple calls to get_FOO"""
self.im.update_fields()
self.assertEqual(len(self.im.fields), 22)
self.assertEqual(len(self.im.relation_fields), 5)
self.assertEqual(len(self.im.many_fields), 2)
self.im.update_attributes()
self.assertEqual(len(self.im.attributes), 1)
self.im.update_methods()
self.assertEqual(len(self.im.methods), 2)
self.assertEqual(len(self.im.items), 34)
