API: Added method for adding and removing extra_marks (#15)

Author: ANUSHIYAPRIYA

Changelog.md

@@ -1,5 +1,6 @@
 # Changelog
 ##[unreleased]
+- Added new methods for adding and removing extra marks (#15)
 - Fixed bug which caused file uploads to be PUT requests by default instead of POST (#22)
 
 ##[v0.1.0]

markusapi/markusapi.py

@@ -309,6 +309,42 @@ def update_marking_state(self, assignment_id: int, group_id: int, new_marking_st
         path = Markus._get_path(assignments=assignment_id, groups=group_id, update_marking_state=None)
         return self._submit_request(params, path, "PUT")
 
+    def create_extra_marks(
+        self,
+        assignment_id: int,
+        group_id: int,
+        extra_marks: float,
+        description: str
+    ) -> ResponseType:
+        """
+        Create new extra mark for the particular group.
+        Mark specified in extra_marks will be created
+        """
+        params = {
+            'extra_marks': extra_marks,
+            'description': description
+        }
+        path = Markus._get_path(assignments=assignment_id, groups=group_id, create_extra_marks=None)
+        return self._submit_request(params, path, 'POST')
+
+    def remove_extra_marks(
+        self,
+        assignment_id: int,
+        group_id: int,
+        extra_marks: float,
+        description: str
+    ) -> ResponseType:
+        """
+        Remove the extra mark for the particular group.
+        Mark specified in extra_marks will be removed
+        """
+        params = {
+            'extra_marks': extra_marks,
+            'description': description
+        }
+        path = Markus._get_path(assignments=assignment_id, groups=group_id, remove_extra_marks=None)
+        return self._submit_request(params, path, 'DELETE')
+
     def get_files_from_repo(
         self, assignment_id: int, group_id: int, filename: Optional[str] = None, collected: bool = True
     ) -> Optional[bytes]:

markusapi/tests/test_markusapi.py

@@ -298,6 +298,32 @@ def test_upload_folder_to_repo(self, _get_path, _decode_json_response, _submit_r
         assert path == _get_path.return_value
         assert params.keys() == {"folder_path"}
 
+    @given(kwargs=strategies_from_signature(Markus.create_extra_marks))
+    @patch.object(Markus, '_submit_request', return_value=DUMMY_RETURNS['_submit_request'])
+    @patch.object(Markus, '_get_path', return_value=DUMMY_RETURNS['path'])
+    def test_create_extra_marks(self, _get_path, _submit_request, kwargs):
+        dummy_markus().create_extra_marks(**kwargs)
+        params = {
+            'extra_marks': kwargs['extra_marks'],
+            'description': kwargs['description']
+        }
+        _get_path.assert_called_with(assignments=kwargs['assignment_id'], groups=kwargs['group_id'],
+                                    create_extra_marks=None)
+        _submit_request.assert_called_with(params, _get_path.return_value, 'POST')
+
+    @given(kwargs=strategies_from_signature(Markus.remove_extra_marks))
+    @patch.object(Markus, '_submit_request', return_value=DUMMY_RETURNS['_submit_request'])
+    @patch.object(Markus, '_get_path', return_value=DUMMY_RETURNS['path'])
+    def test_remove_extra_marks(self, _get_path, _submit_request, kwargs):
+        dummy_markus().remove_extra_marks(**kwargs)
+        params = {
+            'extra_marks': kwargs['extra_marks'],
+            'description': kwargs['description']
+        }
+        _get_path.assert_called_with(assignments=kwargs['assignment_id'], groups=kwargs['group_id'],
+                                    remove_extra_marks=None)
+        _submit_request.assert_called_with(params, _get_path.return_value, 'DELETE')
+
     @given(kwargs=strategies_from_signature(Markus.upload_file_to_repo), filename=file_name_strategy())
     @patch.object(Markus, "_submit_request", return_value=DUMMY_RETURNS["_submit_request"])
     @patch.object(Markus, "_decode_json_response", return_value=[DUMMY_RETURNS["_decode_json_response"]])