implement kwargs (fix ucfopen#631) for upload_comment

canvasapi/submission.py

@@ -177,7 +177,11 @@ def upload_comment(self, file: FileOrPathLike, **kwargs):
         ).start()
 
         if response[0]:
-            self.edit(comment={"file_ids": [response[1]["id"]]})
+            if "comment" in kwargs:
+                kwargs["comment"].update({"file_ids": [response[1]["id"]]})
+            else:
+                kwargs["comment"] = {"file_ids": [response[1]["id"]]}
+            self.edit(**kwargs)
         return response
 
 

tests/test_submission.py

@@ -122,6 +122,25 @@ def test_upload_comment(self, m):
         finally:
             cleanup_file(filename)
 
+    def test_upload_comment_with_kwargs(self, m):
+        register_uris(
+            {"submission": ["upload_comment", "upload_comment_final", "edit"]}, m
+        )
+
+        filename = "testfile_submission_{}".format(uuid.uuid4().hex)
+
+        try:
+            with open(filename, "w+") as file:
+                response = self.submission.upload_comment(
+                    file, comment={"attempt": 1, "text_comment": "This is just a test."}
+                )
+
+            self.assertTrue(response[0])
+            self.assertIsInstance(response[1], dict)
+            self.assertIn("url", response[1])
+        finally:
+            cleanup_file(filename)
+
 
 class TestGroupedSubmission(unittest.TestCase):
     def setUp(self):