Issue #25: More test cases

Author: boonya

tests/test_context_view.py

@@ -116,15 +116,25 @@ def get_data(self):
         }
 
 
-class EmptyModelContext(views.ContextView):
-    """Badly initiated context."""
-    pass
+class ModelUndefinedContext(views.ContextView):
+    """Model undefined"""
+
+    def get_data(self):
+        pass
 
 
 class WrongModelContext(views.ContextView):
-    """Badly initiated context."""
+    """Wrong model defined."""
     __model__ = type
 
+    def get_data(self):
+        pass
+
+
+class GetterUndefinedContext(views.ContextView):
+    """Getter undefined."""
+    __model__ = Profile
+
 
 class TestContextView(unittest.TestCase):
     main_photo = Photo(id=1, title='main photo', path='path/to/the/main/photo',
@@ -141,14 +151,23 @@ class TestContextView(unittest.TestCase):
                       main_photo=main_photo,
                       photos=[main_photo, photo2, photo3])
 
-    def test_wrong_class_defined(self):
-        with self.assertRaises(TypeError):
-            EmptyModelContext("it doesn't matter")
-
     def test_wrong_model_passed(self):
         with self.assertRaises(TypeError):
             PublicProfile("invalid argument")
 
+    def test_model_undefined(self):
+        with self.assertRaises(TypeError):
+            ModelUndefinedContext("it doesn't matter")
+
+    def test_wrong_model_defined(self):
+        with self.assertRaises(TypeError):
+            WrongModelContext("it doesn't matter")
+
+    def test_getter_undefined(self):
+        contextual_view = GetterUndefinedContext(self.profile)
+        with self.assertRaises(NotImplementedError):
+            contextual_view.get_data()
+
     def test_context_view(self):
         public_context = PublicProfile(self.profile)
         private_context = PrivateProfile(self.profile)