Adjust add_model_content unit tests

Author: smlindauer

src/sasctl/_services/model_repository.py

@@ -366,7 +366,9 @@ def create_model(
         )
 
     @classmethod
-    def add_model_content(cls, model, file, name, role=None, content_type=None):
+    def add_model_content(
+        cls, model, file, name, role=None, content_type="multipart/form-data"
+    ):
         """Add additional files to the model.
 
         Parameters
@@ -379,9 +381,9 @@ def add_model_content(cls, model, file, name, role=None, content_type=None):
         name : str
             Name of the file related to the model.
         role : str
-            Role of the model file, such as 'Python pickle'.
+            Role of the model file, such as 'Python pickle'. Default value is None.
         content_type : str
-            an HTTP Content-Type value
+            An HTTP Content-Type value. Default value is multipart/form-data.
 
         Returns
         -------
@@ -397,7 +399,7 @@ def add_model_content(cls, model, file, name, role=None, content_type=None):
             model = cls.get_model(model)
             id_ = model["id"]
 
-        if content_type is None and isinstance(file, bytes):
+        if content_type is "multipart/form-data" and isinstance(file, bytes):
             content_type = "application/octet-stream"
         elif isinstance(file, dict):
             import json

tests/unit/test_model_repository.py

@@ -183,17 +183,23 @@ def test_add_model_content():
 
             # Basic upload of text data
             mr.add_model_content(None, text_data, 'test.txt')
-            assert post.call_args[1]['files'] == {'test.txt': text_data}
+            assert post.call_args[1]['files'] == {'files': ('test.txt', text_data, 'multipart/form-data')}
 
             # Upload of text data with content type
             mr.add_model_content(None, text_data, 'test.txt', content_type='application/text')
-            assert post.call_args[1]['files'] == {'test.txt': ('test.txt', text_data, 'application/text')}
+            assert post.call_args[1]['files'] == {'files': ('test.txt', text_data, 'application/text')}
+
+            # Upload of dict data without content type
+            import json
+            dict_data = {'data': text_data}
+            mr.add_model_content(None, dict_data, 'dict.json')
+            assert post.call_args[1]['files'] == {'files': ('dict.json', json.dumps(dict_data), 'multipart/form-data')}
 
             # Upload of binary data should include content type
             binary_data = 'Test binary file contents'.encode()
             mr.add_model_content(None, binary_data, 'test.pkl')
-            assert post.call_args[1]['files'] == {'test.pkl': ('test.pkl', binary_data, 'application/octet-stream')}
+            assert post.call_args[1]['files'] == {'files': ('test.pkl', binary_data, 'application/octet-stream')}
 
             # Should be able to customize content type
             mr.add_model_content(None, binary_data, 'test.pkl', content_type='application/image')
-            assert post.call_args[1]['files'] == {'test.pkl': ('test.pkl', binary_data, 'application/image')}
+            assert post.call_args[1]['files'] == {'files': ('test.pkl', binary_data, 'application/image')}