remove unnecessary tests from blueprint and snapshot smoke tests

Author: sid-rl

tests/smoketests/sdk/test_async_blueprint.py

@@ -221,8 +221,7 @@ async def test_create_devbox_from_blueprint(self, async_sdk_client: AsyncRunloop
 
         try:
             # Create devbox from the blueprint
-            devbox = await async_sdk_client.devbox.create_from_blueprint_id(
-                blueprint_id=blueprint.id,
+            devbox = await blueprint.create_devbox(
                 name=unique_name("sdk-async-devbox-from-blueprint"),
                 launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
             )
@@ -244,42 +243,6 @@ async def test_create_devbox_from_blueprint(self, async_sdk_client: AsyncRunloop
         finally:
             await blueprint.delete()
 
-    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT * 2)
-    async def test_create_multiple_devboxes_from_blueprint(self, async_sdk_client: AsyncRunloopSDK) -> None:
-        """Test creating multiple devboxes from the same blueprint."""
-        # Create a blueprint
-        blueprint = await async_sdk_client.blueprint.create(
-            name=unique_name("sdk-async-blueprint-multi-devbox"),
-            dockerfile="FROM ubuntu:20.04\nRUN apt-get update && apt-get install -y curl",
-        )
-
-        try:
-            # Create first devbox
-            devbox1 = await async_sdk_client.devbox.create_from_blueprint_id(
-                blueprint_id=blueprint.id,
-                name=unique_name("sdk-async-devbox-1"),
-                launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-            )
-
-            # Create second devbox
-            devbox2 = await async_sdk_client.devbox.create_from_blueprint_id(
-                blueprint_id=blueprint.id,
-                name=unique_name("sdk-async-devbox-2"),
-                launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-            )
-
-            try:
-                assert devbox1.id != devbox2.id
-                info1 = await devbox1.get_info()
-                info2 = await devbox2.get_info()
-                assert info1.status == "running"
-                assert info2.status == "running"
-            finally:
-                await devbox1.shutdown()
-                await devbox2.shutdown()
-        finally:
-            await blueprint.delete()
-
 
 class TestAsyncBlueprintErrorHandling:
     """Test async blueprint error handling scenarios."""

tests/smoketests/sdk/test_async_snapshot.py

@@ -220,8 +220,7 @@ async def test_restore_devbox_from_snapshot(self, async_sdk_client: AsyncRunloop
 
             try:
                 # Create new devbox from snapshot
-                restored_devbox = await async_sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
+                restored_devbox = await snapshot.create_devbox(
                     name=unique_name("sdk-async-restored-devbox"),
                     launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
                 )
@@ -242,60 +241,6 @@ async def test_restore_devbox_from_snapshot(self, async_sdk_client: AsyncRunloop
         finally:
             await source_devbox.shutdown()
 
-    @pytest.mark.timeout(FOUR_MINUTE_TIMEOUT)
-    async def test_multiple_devboxes_from_snapshot(self, async_sdk_client: AsyncRunloopSDK) -> None:
-        """Test creating multiple devboxes from the same snapshot."""
-        # Create source devbox
-        source_devbox = await async_sdk_client.devbox.create(
-            name=unique_name("sdk-async-source-multi"),
-            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-        )
-
-        try:
-            # Create content
-            await source_devbox.file.write(file_path="/tmp/async_shared.txt", contents="Async shared content")
-
-            # Create snapshot
-            snapshot = await source_devbox.snapshot_disk(
-                name=unique_name("sdk-async-snapshot-multi"),
-            )
-
-            try:
-                # Create first devbox from snapshot
-                devbox1 = await async_sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
-                    name=unique_name("sdk-async-restored-1"),
-                    launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-                )
-
-                # Create second devbox from snapshot
-                devbox2 = await async_sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
-                    name=unique_name("sdk-async-restored-2"),
-                    launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-                )
-
-                try:
-                    # Both should be running
-                    assert devbox1.id != devbox2.id
-                    info1 = await devbox1.get_info()
-                    info2 = await devbox2.get_info()
-                    assert info1.status == "running"
-                    assert info2.status == "running"
-
-                    # Both should have the snapshot content
-                    content1 = await devbox1.file.read(file_path="/tmp/async_shared.txt")
-                    content2 = await devbox2.file.read(file_path="/tmp/async_shared.txt")
-                    assert content1 == "Async shared content"
-                    assert content2 == "Async shared content"
-                finally:
-                    await devbox1.shutdown()
-                    await devbox2.shutdown()
-            finally:
-                await snapshot.delete()
-        finally:
-            await source_devbox.shutdown()
-
 
 class TestAsyncSnapshotListing:
     """Test async snapshot listing and retrieval operations."""
@@ -365,53 +310,3 @@ async def test_list_snapshots_by_devbox(self, async_sdk_client: AsyncRunloopSDK)
                 await snapshot.delete()
         finally:
             await devbox.shutdown()
-
-
-class TestAsyncSnapshotEdgeCases:
-    """Test async snapshot edge cases and special scenarios."""
-
-    @pytest.mark.timeout(FOUR_MINUTE_TIMEOUT)
-    async def test_snapshot_preserves_file_permissions(self, async_sdk_client: AsyncRunloopSDK) -> None:
-        """Test that snapshot preserves file permissions."""
-        # Create devbox
-        devbox = await async_sdk_client.devbox.create(
-            name=unique_name("sdk-async-devbox-permissions"),
-            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-        )
-
-        try:
-            # Create executable file
-            await devbox.file.write(file_path="/tmp/test_async_exec.sh", contents="#!/bin/bash\necho 'Hello'")
-            await devbox.cmd.exec(command="chmod +x /tmp/test_async_exec.sh")
-
-            # Verify it's executable
-            result = await devbox.cmd.exec(command="test -x /tmp/test_async_exec.sh && echo 'executable'")
-            stdout = await result.stdout(num_lines=1)
-            assert "executable" in stdout
-
-            # Create snapshot
-            snapshot = await devbox.snapshot_disk(
-                name=unique_name("sdk-async-snapshot-permissions"),
-            )
-
-            try:
-                # Restore from snapshot
-                restored_devbox = await async_sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
-                    name=unique_name("sdk-async-restored-permissions"),
-                    launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-                )
-
-                try:
-                    # Verify file is still executable
-                    result = await restored_devbox.cmd.exec(
-                        command="test -x /tmp/test_async_exec.sh && echo 'still_executable'"
-                    )
-                    stdout = await result.stdout(num_lines=1)
-                    assert "still_executable" in stdout
-                finally:
-                    await restored_devbox.shutdown()
-            finally:
-                await snapshot.delete()
-        finally:
-            await devbox.shutdown()

tests/smoketests/sdk/test_blueprint.py

@@ -221,8 +221,7 @@ def test_create_devbox_from_blueprint(self, sdk_client: RunloopSDK) -> None:
 
         try:
             # Create devbox from the blueprint
-            devbox = sdk_client.devbox.create_from_blueprint_id(
-                blueprint_id=blueprint.id,
+            devbox = blueprint.create_devbox(
                 name=unique_name("sdk-devbox-from-blueprint"),
                 launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
             )
@@ -244,40 +243,6 @@ def test_create_devbox_from_blueprint(self, sdk_client: RunloopSDK) -> None:
         finally:
             blueprint.delete()
 
-    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT * 2)
-    def test_create_multiple_devboxes_from_blueprint(self, sdk_client: RunloopSDK) -> None:
-        """Test creating multiple devboxes from the same blueprint."""
-        # Create a blueprint
-        blueprint = sdk_client.blueprint.create(
-            name=unique_name("sdk-blueprint-multi-devbox"),
-            dockerfile="FROM ubuntu:20.04\nRUN apt-get update && apt-get install -y curl",
-        )
-
-        try:
-            # Create first devbox
-            devbox1 = sdk_client.devbox.create_from_blueprint_id(
-                blueprint_id=blueprint.id,
-                name=unique_name("sdk-devbox-1"),
-                launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-            )
-
-            # Create second devbox
-            devbox2 = sdk_client.devbox.create_from_blueprint_id(
-                blueprint_id=blueprint.id,
-                name=unique_name("sdk-devbox-2"),
-                launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-            )
-
-            try:
-                assert devbox1.id != devbox2.id
-                assert devbox1.get_info().status == "running"
-                assert devbox2.get_info().status == "running"
-            finally:
-                devbox1.shutdown()
-                devbox2.shutdown()
-        finally:
-            blueprint.delete()
-
 
 class TestBlueprintErrorHandling:
     """Test blueprint error handling scenarios."""

tests/smoketests/sdk/test_snapshot.py

@@ -219,8 +219,7 @@ def test_restore_devbox_from_snapshot(self, sdk_client: RunloopSDK) -> None:
 
             try:
                 # Create new devbox from snapshot
-                restored_devbox = sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
+                restored_devbox = snapshot.create_devbox(
                     name=unique_name("sdk-restored-devbox"),
                     launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
                 )
@@ -241,58 +240,6 @@ def test_restore_devbox_from_snapshot(self, sdk_client: RunloopSDK) -> None:
         finally:
             source_devbox.shutdown()
 
-    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT * 2)
-    def test_multiple_devboxes_from_snapshot(self, sdk_client: RunloopSDK) -> None:
-        """Test creating multiple devboxes from the same snapshot."""
-        # Create source devbox
-        source_devbox = sdk_client.devbox.create(
-            name=unique_name("sdk-source-multi"),
-            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-        )
-
-        try:
-            # Create content
-            source_devbox.file.write(file_path="/tmp/shared.txt", contents="Shared content")
-
-            # Create snapshot
-            snapshot = source_devbox.snapshot_disk(
-                name=unique_name("sdk-snapshot-multi"),
-            )
-
-            try:
-                # Create first devbox from snapshot
-                devbox1 = sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
-                    name=unique_name("sdk-restored-1"),
-                    launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-                )
-
-                # Create second devbox from snapshot
-                devbox2 = sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
-                    name=unique_name("sdk-restored-2"),
-                    launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-                )
-
-                try:
-                    # Both should be running
-                    assert devbox1.id != devbox2.id
-                    assert devbox1.get_info().status == "running"
-                    assert devbox2.get_info().status == "running"
-
-                    # Both should have the snapshot content
-                    content1 = devbox1.file.read(file_path="/tmp/shared.txt")
-                    content2 = devbox2.file.read(file_path="/tmp/shared.txt")
-                    assert content1 == "Shared content"
-                    assert content2 == "Shared content"
-                finally:
-                    devbox1.shutdown()
-                    devbox2.shutdown()
-            finally:
-                snapshot.delete()
-        finally:
-            source_devbox.shutdown()
-
 
 class TestSnapshotListing:
     """Test snapshot listing and retrieval operations."""
@@ -362,49 +309,3 @@ def test_list_snapshots_by_devbox(self, sdk_client: RunloopSDK) -> None:
                 snapshot.delete()
         finally:
             devbox.shutdown()
-
-
-class TestSnapshotEdgeCases:
-    """Test snapshot edge cases and special scenarios."""
-
-    @pytest.mark.timeout(TWO_MINUTE_TIMEOUT * 2)
-    def test_snapshot_preserves_file_permissions(self, sdk_client: RunloopSDK) -> None:
-        """Test that snapshot preserves file permissions."""
-        # Create devbox
-        devbox = sdk_client.devbox.create(
-            name=unique_name("sdk-devbox-permissions"),
-            launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-        )
-
-        try:
-            # Create executable file
-            devbox.file.write(file_path="/tmp/test_exec.sh", contents="#!/bin/bash\necho 'Hello'")
-            devbox.cmd.exec(command="chmod +x /tmp/test_exec.sh")
-
-            # Verify it's executable
-            result = devbox.cmd.exec(command="test -x /tmp/test_exec.sh && echo 'executable'")
-            assert "executable" in result.stdout(num_lines=1)
-
-            # Create snapshot
-            snapshot = devbox.snapshot_disk(
-                name=unique_name("sdk-snapshot-permissions"),
-            )
-
-            try:
-                # Restore from snapshot
-                restored_devbox = sdk_client.devbox.create_from_snapshot(
-                    snapshot_id=snapshot.id,
-                    name=unique_name("sdk-restored-permissions"),
-                    launch_parameters={"resource_size_request": "SMALL", "keep_alive_time_seconds": 60 * 5},
-                )
-
-                try:
-                    # Verify file is still executable
-                    result = restored_devbox.cmd.exec(command="test -x /tmp/test_exec.sh && echo 'still_executable'")
-                    assert "still_executable" in result.stdout(num_lines=1)
-                finally:
-                    restored_devbox.shutdown()
-            finally:
-                snapshot.delete()
-        finally:
-            devbox.shutdown()