unit test changes

Author: sid-rl

tests/sdk/test_async_clients.py

@@ -237,7 +237,7 @@ async def test_list(self, mock_async_client: AsyncMock, object_view: MockObjectV
             name="test",
             search="query",
             starting_after="obj_000",
-            state="ready",
+            state="READ_ONLY",
         )
 
         assert len(objects) == 1
@@ -282,14 +282,15 @@ async def test_upload_from_text(self, mock_async_client: AsyncMock, object_view:
         mock_async_client._client = http_client
 
         client = AsyncStorageObjectOps(mock_async_client)
-        obj = await client.upload_from_text("test content", "test.txt", metadata={"key": "value"})
+        obj = await client.upload_from_text("test content", name="test.txt", metadata={"key": "value"})
 
         assert isinstance(obj, AsyncStorageObject)
         assert obj.id == "obj_123"
         mock_async_client.objects.create.assert_awaited_once_with(
             name="test.txt",
             content_type="text",
             metadata={"key": "value"},
+            ttl_ms=None,
         )
         http_client.put.assert_awaited_once_with(object_view.upload_url, content="test content")
         mock_async_client.objects.complete.assert_awaited_once()
@@ -306,14 +307,15 @@ async def test_upload_from_bytes(self, mock_async_client: AsyncMock, object_view
         mock_async_client._client = http_client
 
         client = AsyncStorageObjectOps(mock_async_client)
-        obj = await client.upload_from_bytes(b"test content", "test.bin", content_type="binary")
+        obj = await client.upload_from_bytes(b"test content", name="test.bin", content_type="binary")
 
         assert isinstance(obj, AsyncStorageObject)
         assert obj.id == "obj_123"
         mock_async_client.objects.create.assert_awaited_once_with(
             name="test.bin",
             content_type="binary",
             metadata=None,
+            ttl_ms=None,
         )
         http_client.put.assert_awaited_once_with(object_view.upload_url, content=b"test content")
         mock_async_client.objects.complete.assert_awaited_once()

tests/sdk/test_clients.py

@@ -225,7 +225,7 @@ def test_list(self, mock_client: Mock, object_view: MockObjectView) -> None:
             name="test",
             search="query",
             starting_after="obj_000",
-            state="ready",
+            state="READ_ONLY",
         )
 
         assert len(objects) == 1
@@ -264,14 +264,15 @@ def test_upload_from_text(self, mock_client: Mock, object_view: MockObjectView)
         mock_client._client = http_client
 
         client = StorageObjectOps(mock_client)
-        obj = client.upload_from_text("test content", "test.txt", metadata={"key": "value"})
+        obj = client.upload_from_text("test content", name="test.txt", metadata={"key": "value"})
 
         assert isinstance(obj, StorageObject)
         assert obj.id == "obj_123"
         mock_client.objects.create.assert_called_once_with(
             name="test.txt",
             content_type="text",
             metadata={"key": "value"},
+            ttl_ms=None,
         )
         http_client.put.assert_called_once_with(object_view.upload_url, content="test content")
         mock_client.objects.complete.assert_called_once()
@@ -286,14 +287,15 @@ def test_upload_from_bytes(self, mock_client: Mock, object_view: MockObjectView)
         mock_client._client = http_client
 
         client = StorageObjectOps(mock_client)
-        obj = client.upload_from_bytes(b"test content", "test.bin", content_type="binary")
+        obj = client.upload_from_bytes(b"test content", name="test.bin", content_type="binary")
 
         assert isinstance(obj, StorageObject)
         assert obj.id == "obj_123"
         mock_client.objects.create.assert_called_once_with(
             name="test.bin",
             content_type="binary",
             metadata=None,
+            ttl_ms=None,
         )
         http_client.put.assert_called_once_with(object_view.upload_url, content=b"test content")
         mock_client.objects.complete.assert_called_once()

tests/smoketests/sdk/test_async_storage_object.py

@@ -98,7 +98,7 @@ async def test_upload_from_text(self, async_sdk_client: AsyncRunloopSDK) -> None
         text_content = "Hello from async upload_from_text!"
         obj = await async_sdk_client.storage_object.upload_from_text(
             text_content,
-            unique_name("sdk-async-text-upload"),
+            name=unique_name("sdk-async-text-upload"),
             metadata={"source": "upload_from_text"},
         )
 
@@ -117,7 +117,7 @@ async def test_upload_from_bytes(self, async_sdk_client: AsyncRunloopSDK) -> Non
         bytes_content = b"Binary content from async SDK"
         obj = await async_sdk_client.storage_object.upload_from_bytes(
             bytes_content,
-            unique_name("sdk-async-bytes-upload"),
+            name=unique_name("sdk-async-bytes-upload"),
             content_type="text",
             metadata={"source": "upload_from_bytes"},
         )
@@ -142,7 +142,7 @@ async def test_upload_from_file(self, async_sdk_client: AsyncRunloopSDK) -> None
         try:
             obj = await async_sdk_client.storage_object.upload_from_file(
                 tmp_path,
-                unique_name("sdk-async-file-upload"),
+                name=unique_name("sdk-async-file-upload"),
                 metadata={"source": "upload_from_file"},
             )
 
@@ -171,7 +171,7 @@ async def test_upload_from_dir(self, async_sdk_client: AsyncRunloopSDK) -> None:
 
             obj = await async_sdk_client.storage_object.upload_from_dir(
                 tmp_path,
-                unique_name("sdk-async-dir-upload"),
+                name=unique_name("sdk-async-dir-upload"),
                 metadata={"source": "upload_from_dir"},
             )
 
@@ -206,7 +206,7 @@ async def test_download_as_text(self, async_sdk_client: AsyncRunloopSDK) -> None
         content = "Async text content to download"
         obj = await async_sdk_client.storage_object.upload_from_text(
             content,
-            unique_name("sdk-async-download-text"),
+            name=unique_name("sdk-async-download-text"),
         )
 
         try:
@@ -221,7 +221,7 @@ async def test_download_as_bytes(self, async_sdk_client: AsyncRunloopSDK) -> Non
         content = b"Async bytes content to download"
         obj = await async_sdk_client.storage_object.upload_from_bytes(
             content,
-            unique_name("sdk-async-download-bytes"),
+            name=unique_name("sdk-async-download-bytes"),
             content_type="text",
         )
 
@@ -237,7 +237,7 @@ async def test_get_download_url(self, async_sdk_client: AsyncRunloopSDK) -> None
         """Test getting download URL."""
         obj = await async_sdk_client.storage_object.upload_from_text(
             "Content for async URL",
-            unique_name("sdk-async-download-url"),
+            name=unique_name("sdk-async-download-url"),
         )
 
         try:
@@ -266,7 +266,7 @@ async def test_get_storage_object_by_id(self, async_sdk_client: AsyncRunloopSDK)
         # Create an object
         created = await async_sdk_client.storage_object.upload_from_text(
             "Content for async retrieval",
-            unique_name("sdk-async-storage-retrieve"),
+            name=unique_name("sdk-async-storage-retrieve"),
         )
 
         try:
@@ -286,7 +286,7 @@ async def test_list_storage_objects_by_content_type(self, async_sdk_client: Asyn
         # Create object with specific content type
         obj = await async_sdk_client.storage_object.upload_from_text(
             "Text content",
-            unique_name("sdk-async-storage-list-type"),
+            name=unique_name("sdk-async-storage-list-type"),
         )
 
         try:
@@ -310,7 +310,7 @@ async def test_mount_storage_object_to_devbox(self, async_sdk_client: AsyncRunlo
         # Create storage object with content
         obj = await async_sdk_client.storage_object.upload_from_text(
             "Async mounted content from SDK",
-            unique_name("sdk-async-mount-object"),
+            name=unique_name("sdk-async-mount-object"),
         )
 
         try:
@@ -342,7 +342,7 @@ async def test_access_mounted_storage_object(self, async_sdk_client: AsyncRunloo
         # Create storage object
         obj = await async_sdk_client.storage_object.upload_from_text(
             "Async content to mount and access",
-            unique_name("sdk-async-mount-access"),
+            name=unique_name("sdk-async-mount-access"),
         )
 
         try:
@@ -385,7 +385,7 @@ async def test_storage_object_large_content(self, async_sdk_client: AsyncRunloop
 
         obj = await async_sdk_client.storage_object.upload_from_text(
             large_content,
-            unique_name("sdk-async-storage-large"),
+            name=unique_name("sdk-async-storage-large"),
         )
 
         try:
@@ -404,7 +404,7 @@ async def test_storage_object_binary_content(self, async_sdk_client: AsyncRunloo
 
         obj = await async_sdk_client.storage_object.upload_from_bytes(
             binary_content,
-            unique_name("sdk-async-storage-binary"),
+            name=unique_name("sdk-async-storage-binary"),
             content_type="binary",
         )
 
@@ -420,7 +420,7 @@ async def test_storage_object_empty_content(self, async_sdk_client: AsyncRunloop
         """Test uploading empty content."""
         obj = await async_sdk_client.storage_object.upload_from_text(
             "",
-            unique_name("sdk-async-storage-empty"),
+            name=unique_name("sdk-async-storage-empty"),
         )
 
         try:

tests/smoketests/sdk/test_storage_object.py

@@ -98,7 +98,7 @@ def test_upload_from_text(self, sdk_client: RunloopSDK) -> None:
         text_content = "Hello from upload_from_text!"
         obj = sdk_client.storage_object.upload_from_text(
             text_content,
-            unique_name("sdk-text-upload"),
+            name=unique_name("sdk-text-upload"),
             metadata={"source": "upload_from_text"},
         )
 
@@ -117,7 +117,7 @@ def test_upload_from_bytes(self, sdk_client: RunloopSDK) -> None:
         bytes_content = b"Binary content from SDK"
         obj = sdk_client.storage_object.upload_from_bytes(
             bytes_content,
-            unique_name("sdk-bytes-upload"),
+            name=unique_name("sdk-bytes-upload"),
             content_type="text",
             metadata={"source": "upload_from_bytes"},
         )
@@ -142,7 +142,7 @@ def test_upload_from_file(self, sdk_client: RunloopSDK) -> None:
         try:
             obj = sdk_client.storage_object.upload_from_file(
                 tmp_path,
-                unique_name("sdk-file-upload"),
+                name=unique_name("sdk-file-upload"),
                 metadata={"source": "upload_from_file"},
             )
 
@@ -171,7 +171,7 @@ def test_upload_from_dir(self, sdk_client: RunloopSDK) -> None:
 
             obj = sdk_client.storage_object.upload_from_dir(
                 tmp_path,
-                unique_name("sdk-dir-upload"),
+                name=unique_name("sdk-dir-upload"),
                 metadata={"source": "upload_from_dir"},
             )
 
@@ -206,7 +206,7 @@ def test_download_as_text(self, sdk_client: RunloopSDK) -> None:
         content = "Text content to download"
         obj = sdk_client.storage_object.upload_from_text(
             content,
-            unique_name("sdk-download-text"),
+            name=unique_name("sdk-download-text"),
         )
 
         try:
@@ -221,7 +221,7 @@ def test_download_as_bytes(self, sdk_client: RunloopSDK) -> None:
         content = b"Bytes content to download"
         obj = sdk_client.storage_object.upload_from_bytes(
             content,
-            unique_name("sdk-download-bytes"),
+            name=unique_name("sdk-download-bytes"),
             content_type="text",
         )
 
@@ -237,7 +237,7 @@ def test_get_download_url(self, sdk_client: RunloopSDK) -> None:
         """Test getting download URL."""
         obj = sdk_client.storage_object.upload_from_text(
             "Content for URL",
-            unique_name("sdk-download-url"),
+            name=unique_name("sdk-download-url"),
         )
 
         try:
@@ -266,7 +266,7 @@ def test_get_storage_object_by_id(self, sdk_client: RunloopSDK) -> None:
         # Create an object
         created = sdk_client.storage_object.upload_from_text(
             "Content for retrieval",
-            unique_name("sdk-storage-retrieve"),
+            name=unique_name("sdk-storage-retrieve"),
         )
 
         try:
@@ -286,7 +286,7 @@ def test_list_storage_objects_by_content_type(self, sdk_client: RunloopSDK) -> N
         # Create object with specific content type
         obj = sdk_client.storage_object.upload_from_text(
             "Text content",
-            unique_name("sdk-storage-list-type"),
+            name=unique_name("sdk-storage-list-type"),
         )
 
         try:
@@ -310,7 +310,7 @@ def test_mount_storage_object_to_devbox(self, sdk_client: RunloopSDK) -> None:
         # Create storage object with content
         obj = sdk_client.storage_object.upload_from_text(
             "Mounted content from SDK",
-            unique_name("sdk-mount-object"),
+            name=unique_name("sdk-mount-object"),
         )
 
         try:
@@ -342,7 +342,7 @@ def test_access_mounted_storage_object(self, sdk_client: RunloopSDK) -> None:
         # Create storage object
         obj = sdk_client.storage_object.upload_from_text(
             "Content to mount and access",
-            unique_name("sdk-mount-access"),
+            name=unique_name("sdk-mount-access"),
         )
 
         try:
@@ -384,7 +384,7 @@ def test_storage_object_large_content(self, sdk_client: RunloopSDK) -> None:
 
         obj = sdk_client.storage_object.upload_from_text(
             large_content,
-            unique_name("sdk-storage-large"),
+            name=unique_name("sdk-storage-large"),
         )
 
         try:
@@ -403,7 +403,7 @@ def test_storage_object_binary_content(self, sdk_client: RunloopSDK) -> None:
 
         obj = sdk_client.storage_object.upload_from_bytes(
             binary_content,
-            unique_name("sdk-storage-binary"),
+            name=unique_name("sdk-storage-binary"),
             content_type="binary",
         )
 
@@ -419,7 +419,7 @@ def test_storage_object_empty_content(self, sdk_client: RunloopSDK) -> None:
         """Test uploading empty content."""
         obj = sdk_client.storage_object.upload_from_text(
             "",
-            unique_name("sdk-storage-empty"),
+            name=unique_name("sdk-storage-empty"),
         )
 
         try: