Add test case for view

Add a test case where the attribute is not defined on either view or model
Javex · Apr 3, 2014 · e15ec14 · e15ec14
1 parent 7b3ac63
commit e15ec14
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_views.py b/tests/test_views.py
@@ -146,6 +146,11 @@ def test_iter_head_cols_missing_info_attr(self):
         assert cols == [{'label': 'Test Text',
                          'css_class': 'column-test_text'}]
 
+    def test_iter_head_cols_wrong_attr(self):
+        self.View.list_display = ('doesnotexist',)
+        with pytest.raises(AttributeError):
+            list(self.view.iter_head_cols())
+
     def test_iter_list_cols_default(self, obj):
         cols = list(self.view.iter_list_cols(obj))
         assert cols == ['ModelStr']