Fix field init tests. Thanks Josh Bohde

newsapps · Aug 28, 2012 · 179ab3f · 179ab3f
1 parent 7b37078
commit 179ab3f
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/content_gfk/tests/fields.py b/tests/content_gfk/tests/fields.py
@@ -9,16 +9,17 @@
 class ContentTypeFieldTestCase(TestCase):
 
     def test_init(self):
+        # Test that you have to use a dict some other resources
         with self.assertRaises(ValueError):
-            # Test that you have to use a dict some other resources
             GenericForeignKeyField(((Note, NoteResource)), 'nofield')
 
-            # Test that you must register some other resources
+        # Test that you must register some other resources
+        with self.assertRaises(ValueError):
             GenericForeignKeyField({}, 'nofield')
 
-            # Test that the resources you raise must be models
-            GenericForeignKeyField({'str': 'ing'}, 'nofield')
-
+        # Test that the resources you raise must be models
+        with self.assertRaises(ValueError):
+            GenericForeignKeyField({NoteResource: Note}, 'nofield')
 
     def test_get_related_resource(self):
         gfk_field = GenericForeignKeyField({