Update tests for studydocs

Author: temparus

amivapi/tests/studydocs/test_studydocs_auth.py

@@ -14,7 +14,7 @@
 
 
 class StudydocsAuthTest(WebTest):
-    """Studycos Test class."""
+    """Studydocs Test class."""
 
     def test_uploader_added(self):
         """Test that the uploader is correctly added to a document."""
@@ -25,6 +25,7 @@ def test_uploader_added(self):
                             headers={'content-type': 'multipart/form-data'},
                             data={
                                 'title': 'test',
+                                'type': 'cheat sheets',
                                 'files': [
                                     (BytesIO(test_file_content), 'test.zip')
                                 ]
@@ -41,6 +42,7 @@ def test_can_edit_own_doc(self):
                             headers={'content-type': 'multipart/form-data'},
                             data={
                                 'title': 'test',
+                                'type': 'cheat sheets',
                                 'files': [
                                     (BytesIO(test_file_content), 'test.zip')
                                 ]
@@ -65,6 +67,7 @@ def test_can_not_edit_others_docs(self):
                             headers={'content-type': 'multipart/form-data'},
                             data={
                                 'title': 'test',
+                                'type': 'cheat sheets',
                                 'files': [
                                     (BytesIO(test_file_content), 'test.zip')
                                 ]

amivapi/tests/studydocs/test_summary.py

@@ -17,14 +17,17 @@ def _load_data(self):
         self.load_fixture({
             'studydocuments': [{
                 'title': 'first',
+                'type': 'lecture documents',
                 'lecture': 'a',
                 'professor': 'a',
             }, {
                 'title': 'second',
+                'type': 'cheat sheets',
                 'lecture': 'a',
                 'professor': 'b',
             }, {
                 'title': 'third',
+                'type': 'lecture documents',
                 'professor': 'b',
             }]
         })
@@ -38,6 +41,10 @@ def test_full_summary(self):
 
         assert '_summary' in response
         assert response['_summary'] == {
+            'type': {
+                'lecture documents': 2,
+                'cheat sheets': 1,
+            },
             'lecture': {
                 'a': 2,
             },
@@ -57,6 +64,10 @@ def test_filtered_summary(self):
 
         assert '_summary' in response
         assert response['_summary'] == {
+            'type': {
+                'lecture documents': 1,  # The document with title `third` is ignored
+                'cheat sheets': 1,
+            },
             'lecture': {
                 'a': 2,
             },