stoobly-agent 1.10.1__py3-none-any.whl → 1.10.2__py3-none-any.whl

stoobly_agent/test/app/proxy/replay/body_parser_service_test.py

@@ -1,5 +1,6 @@
 import pdb
 import pytest
+import base64
 
 from stoobly_agent.app.proxy.replay.body_parser_service import decode_response, encode_response
 
@@ -11,13 +12,36 @@ def multipart_string():
 def content_type():
   return "multipart/form-data; boundary=ce560532019a77d83195f9e9873e16a1"
 
+@pytest.fixture
+def image_data():
+  """A small 1x1 pixel PNG image in base64 format for testing."""
+  # This is a minimal 1x1 pixel PNG image
+  png_base64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg=="
+  return base64.b64decode(png_base64)
+
+@pytest.fixture
+def multipart_with_image_string(image_data):
+  """Multipart form data containing both text fields and an image file."""
+  boundary = "ce560532019a77d83195f9e9873e16a1"
+  
+  # Create multipart form data with text field and image
+  result = f'--{boundary}\r\nContent-Disposition: form-data; name="title"\r\n\r\nTest Image Upload\r\n'.encode('utf-8')
+  result += f'--{boundary}\r\nContent-Disposition: form-data; name="image"; filename="test.png"\r\nContent-Type: image/png\r\n\r\n'.encode('utf-8')
+  result += image_data
+  result += f'\r\n--{boundary}--\r\n'.encode('utf-8')
+
+  with open('/tmp/multi', 'wb') as fp:
+    fp.write(result)
+
+  return result
+
 class TestMultipart():
 
   def test_decodes_response(self, multipart_string: bytes, content_type: str):
     multidict = decode_response(multipart_string, content_type)
 
-    assert multidict.get('author') == 'John Smith'
-    assert multidict.get('file') == 'Hello World'
+    assert multidict.get('author') == b'John Smith'
+    assert multidict.get('file') == b'Hello World'
 
   def test_encodes_response(self, content_type: str):
     expected_params = { 'author': 'John Smith', 'file': 'Hello World'}
@@ -25,4 +49,72 @@ class TestMultipart():
 
     multidict = decode_response(multipart_string, content_type)
     for key in expected_params:
-      assert multidict.get(key) == expected_params[key]
+      assert multidict.get(key) == expected_params[key].encode('utf-8')
+
+  def test_decodes_multipart_with_image(self, multipart_with_image_string: bytes, content_type: str, image_data: bytes):
+    """Test decoding multipart form data that contains both text fields and binary image data."""
+    multidict = decode_response(multipart_with_image_string, content_type)
+
+    # Check text field
+    assert multidict.get('title') == b'Test Image Upload'
+    
+    # Check image field - should contain the binary image data
+    image_field = multidict.get('image')
+    assert image_field is not None
+    assert isinstance(image_field, bytes)
+    assert image_field == image_data
+    assert len(image_field) == len(image_data)
+
+  def test_encodes_multipart_with_image(self, content_type: str, image_data: bytes):
+    """Test encoding multipart form data that contains both text fields and binary image data."""
+    expected_params = {
+      'title': 'Test Image Upload',
+      'image': image_data
+    }
+    
+    multipart_string = encode_response(expected_params, content_type)
+    multidict = decode_response(multipart_string, content_type)
+    
+    # Verify text field
+    assert multidict.get('title') == b'Test Image Upload'
+    
+    # Verify image field
+    decoded_image = multidict.get('image')
+    assert decoded_image is not None
+    assert isinstance(decoded_image, bytes)
+    assert decoded_image == image_data
+
+  def test_decode_error_handling(self):
+    """Test decode function error handling with malformed multipart data."""
+    from stoobly_agent.app.proxy.replay.multipart import decode
+    
+    # Test with invalid content type
+    headers = {'content-type': 'invalid/type'}
+    result = decode(headers, b'some content')
+    assert result == b'some content'  # Should return original content
+    
+    # Test with missing boundary
+    headers = {'content-type': 'multipart/form-data'}
+    result = decode(headers, b'some content')
+    assert result == b'some content'  # Should return original content
+    
+    # Test with malformed multipart data
+    headers = {'content-type': 'multipart/form-data; boundary=test123'}
+    malformed_content = b'--test123\r\nincomplete part'
+    result = decode(headers, malformed_content)
+    # Should return original content or handle gracefully
+    assert result is not None  # Should not crash
+
+  def test_decode_return_type_consistency(self):
+    """Test that decode function returns consistent types."""
+    from stoobly_agent.app.proxy.replay.multipart import decode
+    
+    # Test with invalid input types
+    headers = {'content-type': 'multipart/form-data; boundary=test123'}
+    
+    # Should return original content for invalid input
+    result = decode(headers, None)
+    assert result is None
+    
+    result = decode(headers, 123)  # int instead of bytes/str
+    assert result == 123

stoobly_agent/test/config/data_dir_test.py

@@ -17,14 +17,9 @@ class TestDataDir():
   def original_cwd(self) -> str:
     return os.getcwd()
 
-  @pytest.fixture(scope='class')
-  def home_dir(self) -> str:
-    return os.path.expanduser("~")
-
-  def test_in_home(self, original_cwd: str, home_dir: str):
-    # A previous test can put us in 'stoobly_agent/test/app/models/schemas/.stoobly'
+  def test_in_home(self, original_cwd: str):
     os.chdir(original_cwd)
-    data_dir_path = os.path.join(home_dir, DATA_DIR_NAME)
+    data_dir_path = os.path.join(original_cwd, DATA_DIR_NAME)
     os.environ[ENV] = NONE
 
     result = DataDir.instance().path

stoobly_agent/test/test_helper.py

@@ -1,29 +1,40 @@
 import os
 import pdb
+import shutil
+import tempfile
 
 from stoobly_agent import VERSION
 from stoobly_agent.config.constants.env_vars import ENV
 from stoobly_agent.app.settings.constants.mode import TEST
 
 from stoobly_agent.app.settings import Settings
-from stoobly_agent.config.data_dir import DataDir
+from stoobly_agent.config.data_dir import DATA_DIR_NAME, DataDir
 from stoobly_agent.lib.orm import ORM
 from stoobly_agent.lib.orm.migrate_service import migrate
 
 DETERMINISTIC_GET_REQUEST_URL = 'https://dog.ceo/api/breeds/list/all'
 NON_DETERMINISTIC_GET_REQUEST_URL = 'https://www.google.com'
 
-def reset():
+def reset(dir_name = 'stoobly-agent-test'):
   os.environ[ENV] = TEST
 
-  DataDir.instance().remove() # Clean data dir for testing
+  # Create a temporary folder using tmpfile
+  temp_dir = os.path.join('/tmp', dir_name)
+  if os.path.exists(temp_dir):
+    shutil.rmtree(temp_dir)
+  os.makedirs(temp_dir)
+  os.chdir(temp_dir)
+
+  data_dir: DataDir = DataDir.instance()
+  data_dir.create()
 
   ORM.instance().initialize_db()
   migrate(VERSION)
 
-  Settings.instance().reset_and_load()
+  settings: Settings = Settings.instance()
+  settings.reset_and_load()
 
-  return Settings.instance()
+  return settings
 
 def assert_orm_request_equivalent(request_1, request_2):
   assert request_1.latency == request_2.latency

{stoobly_agent-1.10.1.dist-info → stoobly_agent-1.10.2.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: stoobly-agent
-Version: 1.10.1
+Version: 1.10.2
 Summary: Record, mock, and test HTTP(s) requests. CLI agent for Stoobly
 License: Apache-2.0
+License-File: LICENSE
 Author: Matt Le
 Author-email: themathewle@gmail.com
 Requires-Python: >=3.10,<4.0
@@ -12,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: click (>=8.1.0,<9.0.0)
 Requires-Dist: diff-match-patch (>=v20241021,<20241022)
 Requires-Dist: distro (>=1.9.0,<1.10)