scalebox-sdk 0.1.25__py3-none-any.whl → 1.0.2__py3-none-any.whl

scalebox/test/test_code_interpreter_sync_comprehensive.py

@@ -106,8 +106,9 @@ result = {"x": x, "y": y}
 print(result)
 """
 
-        execution = self.sandbox.run_code(code, language="python")
+        execution = self.sandbox.run_code(code, envs={"CI_TEST": "sync_test1"})
         print(execution.to_json())
+        # time.sleep(1000)
         assert isinstance(execution, Execution)
         assert execution.error is None
         assert len(execution.logs.stdout) > 0
@@ -147,7 +148,7 @@ print(f"标准差: {std_val:.3f}")
 }
 """
 
-        execution = self.sandbox.run_code(code, language="python")
+        execution = self.sandbox.run_code(code)
         print(execution.to_json())
         assert execution.error is None
         assert any("圆的面积" in line for line in execution.logs.stdout)
@@ -197,7 +198,7 @@ result = {
 print(f"\\n处理结果: {json.dumps(result, indent=2)}")
 """
 
-        execution = self.sandbox.run_code(code, language="python")
+        execution = self.sandbox.run_code(code)
         print(execution.to_json())
         assert execution.error is None
         assert any("原始数据" in line for line in execution.logs.stdout)
@@ -365,10 +366,7 @@ print(f"结果: {result}")
         assert self.sandbox is not None
 
         # 创建Python上下文
-        python_context = self.sandbox.create_code_context(
-            language="python", 
-            cwd="/tmp"
-        )
+        python_context = self.sandbox.create_code_context(language="python", cwd="/tmp")
         print(python_context)
         assert isinstance(python_context, Context)
         assert python_context.id is not None
@@ -799,6 +797,55 @@ print("生成HTML格式结果")
 from IPython.display import HTML
 HTML(html_content)
 """
+        # code = """
+        # import sys
+        # try:
+        #     import matplotlib
+        #     matplotlib.use('Agg')  # Non-interactive backend
+        #     import matplotlib.pyplot as plt
+        #     import numpy as np
+        #     from IPython.display import display
+        #
+        #     print("✅ All imports successful")
+        #
+        #     # Generate a simple plot
+        #     x = np.linspace(0, 2 * np.pi, 100)
+        #     y = np.sin(x)
+        #
+        #     fig, ax = plt.subplots(figsize=(8, 6))
+        #     ax.plot(x, y, 'b-', linewidth=2, label='sin(x)')
+        #     ax.set_xlabel('X axis')
+        #     ax.set_ylabel('Y axis')
+        #     ax.set_title('Sine Wave - Test Plot')
+        #     ax.legend()
+        #     ax.grid(True, alpha=0.3)
+        #
+        #     plt.tight_layout()
+        #
+        #     # 🎯 KEY: Save to bytes and display as Image
+        #     import io
+        #     from IPython.display import Image
+        #
+        #     buf = io.BytesIO()
+        #     plt.savefig(buf, format='png', bbox_inches='tight', dpi=100)
+        #     buf.seek(0)
+        #     plt.close(fig)
+        #
+        #     # Display the image data
+        #     display(Image(data=buf.getvalue()))
+        #     #
+        #     # print("SUCCESS: Plot generated and displayed")
+        #     # "plot_created"
+        #
+        # except ImportError as e:
+        #     print(f"SKIP: matplotlib not available - {e}")
+        #     "matplotlib_not_installed"
+        # except Exception as e:
+        #     print(f"ERROR: {e}")
+        #     import traceback
+        #     traceback.print_exc()
+        #     "error_occurred"
+        # """
 
         execution = self.sandbox.run_code(code, language="python")
         print(execution.to_json())
@@ -3292,68 +3339,68 @@ print(f"\\n完成 {len(results)} 个API调用")
             self.test_file_operations_simulation, "File Operations Simulation"
         )
 
-        # 性能测试
-        self.run_test(
-            self.test_performance_simple_calculations, "Performance Simple Calculations"
-        )
-        self.run_test(
-            self.test_performance_concurrent_simulation,
-            "Performance Concurrent Simulation",
-        )
-
-        # 结果格式测试
-        self.run_test(self.test_text_result, "Text Result Format")
-        self.run_test(self.test_html_result, "HTML Result Format")
-        self.run_test(self.test_markdown_result, "Markdown Result Format")
-        self.run_test(self.test_svg_result, "SVG Result Format")
-        self.run_test(self.test_image_results, "Image Result Formats (PNG/JPEG)")
-        self.run_test(self.test_latex_result, "LaTeX Result Format")
-        self.run_test(self.test_json_data_result, "JSON Data Result Format")
-        self.run_test(self.test_javascript_result, "JavaScript Result Format")
-        self.run_test(self.test_chart_data_result, "Chart Data Result Format")
-        self.run_test(self.test_mixed_format_result, "Mixed Format Result")
-
-        # R语言测试
-        self.run_test(
-            self.test_r_language_basic_execution, "R Language Basic Execution"
-        )
-        self.run_test(self.test_r_language_data_analysis, "R Language Data Analysis")
-        self.run_test(self.test_r_language_visualization, "R Language Visualization")
-        self.run_test(self.test_r_language_statistics, "R Language Statistics")
-        self.run_test(
-            self.test_r_language_context_management, "R Language Context Management"
-        )
-
-        # Node.js/JavaScript 测试
-        self.run_test(self.test_nodejs_basic_execution, "Node.js Basic Execution")
-        self.run_test(self.test_nodejs_async_promises, "Node.js Async Promises")
-        self.run_test(self.test_nodejs_data_processing, "Node.js Data Processing")
-        self.run_test(self.test_nodejs_chart_data, "Node.js Chart Data Generation")
-        self.run_test(self.test_nodejs_context_management, "Node.js Context Management")
-
-        # Bash 测试
-        self.run_test(self.test_bash_basic_execution, "Bash Basic Execution")
-        self.run_test(self.test_bash_file_operations, "Bash File Operations")
-        self.run_test(self.test_bash_pipelines_and_grep, "Bash Pipelines and Grep")
-        # self.run_test(self.test_bash_env_and_exit_codes, "Bash Env and Exit Codes")
-        self.run_test(self.test_bash_context_management, "Bash Context Management")
-
-        # IJAVA 测试
-        self.run_test(self.test_ijava_basic_execution, "IJAVA Basic Execution")
-        self.run_test(self.test_ijava_oop_features, "IJAVA OOP Features")
-        self.run_test(self.test_ijava_collections, "IJAVA Collections")
-        self.run_test(self.test_ijava_file_io, "IJAVA File I/O")
-        self.run_test(self.test_ijava_context_management, "IJAVA Context Management")
-
-        # Deno 测试
-        self.run_test(self.test_deno_basic_execution, "Deno Basic Execution")
-        self.run_test(self.test_deno_typescript_features, "Deno TypeScript Features")
-        self.run_test(self.test_deno_async_await, "Deno Async/Await")
-        self.run_test(self.test_deno_file_operations, "Deno File Operations")
-        self.run_test(self.test_deno_context_management, "Deno Context Management")
-
-        # 高级功能测试
-        self.run_test(self.test_web_request_simulation, "Web Request Simulation")
+        # # 性能测试
+        # self.run_test(
+        #     self.test_performance_simple_calculations, "Performance Simple Calculations"
+        # )
+        # self.run_test(
+        #     self.test_performance_concurrent_simulation,
+        #     "Performance Concurrent Simulation",
+        # )
+
+        # # 结果格式测试
+        # self.run_test(self.test_text_result, "Text Result Format")
+        # self.run_test(self.test_html_result, "HTML Result Format")
+        # self.run_test(self.test_markdown_result, "Markdown Result Format")
+        # self.run_test(self.test_svg_result, "SVG Result Format")
+        # self.run_test(self.test_image_results, "Image Result Formats (PNG/JPEG)")
+        # self.run_test(self.test_latex_result, "LaTeX Result Format")
+        # self.run_test(self.test_json_data_result, "JSON Data Result Format")
+        # self.run_test(self.test_javascript_result, "JavaScript Result Format")
+        # self.run_test(self.test_chart_data_result, "Chart Data Result Format")
+        # self.run_test(self.test_mixed_format_result, "Mixed Format Result")
+        #
+        # # R语言测试
+        # self.run_test(
+        #     self.test_r_language_basic_execution, "R Language Basic Execution"
+        # )
+        # self.run_test(self.test_r_language_data_analysis, "R Language Data Analysis")
+        # self.run_test(self.test_r_language_visualization, "R Language Visualization")
+        # self.run_test(self.test_r_language_statistics, "R Language Statistics")
+        # self.run_test(
+        #     self.test_r_language_context_management, "R Language Context Management"
+        # )
+        #
+        # # Node.js/JavaScript 测试
+        # self.run_test(self.test_nodejs_basic_execution, "Node.js Basic Execution")
+        # self.run_test(self.test_nodejs_async_promises, "Node.js Async Promises")
+        # self.run_test(self.test_nodejs_data_processing, "Node.js Data Processing")
+        # self.run_test(self.test_nodejs_chart_data, "Node.js Chart Data Generation")
+        # self.run_test(self.test_nodejs_context_management, "Node.js Context Management")
+        #
+        # # Bash 测试
+        # self.run_test(self.test_bash_basic_execution, "Bash Basic Execution")
+        # self.run_test(self.test_bash_file_operations, "Bash File Operations")
+        # self.run_test(self.test_bash_pipelines_and_grep, "Bash Pipelines and Grep")
+        # # self.run_test(self.test_bash_env_and_exit_codes, "Bash Env and Exit Codes")
+        # self.run_test(self.test_bash_context_management, "Bash Context Management")
+        #
+        # # IJAVA 测试
+        # self.run_test(self.test_ijava_basic_execution, "IJAVA Basic Execution")
+        # self.run_test(self.test_ijava_oop_features, "IJAVA OOP Features")
+        # self.run_test(self.test_ijava_collections, "IJAVA Collections")
+        # self.run_test(self.test_ijava_file_io, "IJAVA File I/O")
+        # self.run_test(self.test_ijava_context_management, "IJAVA Context Management")
+        #
+        # # Deno 测试
+        # self.run_test(self.test_deno_basic_execution, "Deno Basic Execution")
+        # # self.run_test(self.test_deno_typescript_features, "Deno TypeScript Features")
+        # self.run_test(self.test_deno_async_await, "Deno Async/Await")
+        # self.run_test(self.test_deno_file_operations, "Deno File Operations")
+        # self.run_test(self.test_deno_context_management, "Deno Context Management")
+        #
+        # # 高级功能测试
+        # self.run_test(self.test_web_request_simulation, "Web Request Simulation")
 
     def cleanup(self):
         """清理资源"""

scalebox/test/test_connect_pause_async.py

@@ -0,0 +1,300 @@
+#!/usr/bin/env python3
+"""
+Test cases for connect and pause functionality (asynchronous version).
+
+This test suite validates:
+- Creating a sandbox
+- Adding and deleting files before pause
+- Pausing the sandbox
+- Connecting to the paused sandbox
+- Verifying file operations after connect
+"""
+
+import asyncio
+import time
+
+from scalebox.sandbox.filesystem.filesystem import WriteInfo, FileType
+from scalebox.sandbox_async.main import AsyncSandbox
+
+
+class ConnectPauseAsyncTest:
+    """Test suite for connect and pause functionality (async)."""
+
+    def __init__(self):
+        self.sandbox = None
+        self.sandbox_id = None
+        self.test_results = []
+
+    def log_test_result(self, test_name: str, success: bool, message: str = ""):
+        """记录测试结果"""
+        status = "✅ PASS" if success else "❌ FAIL"
+        result = {
+            "test": test_name,
+            "success": success,
+            "message": message,
+        }
+        self.test_results.append(result)
+        print(f"{status} {test_name}: {message}")
+
+    async def test_connect_pause_with_files(self):
+        """测试connect和pause功能，包含文件操作"""
+        try:
+            # 1. 创建沙箱
+            print("Step 1: Creating sandbox...")
+            self.sandbox = await AsyncSandbox.create(
+                template="base",
+                timeout=3600,
+                metadata={"test": "connect_pause_async"},
+                envs={"TEST_ENV": "async_test"},
+            )
+            self.sandbox_id = self.sandbox.sandbox_id
+            assert self.sandbox is not None
+            assert self.sandbox_id is not None
+            print(f"Created sandbox with ID: {self.sandbox_id}")
+
+            # 2. 在pause之前增加文件
+            print("Step 2: Adding files before pause...")
+            test_files = {
+                "/tmp/test_file1.txt": "Content of test file 1",
+                "/tmp/test_file2.txt": "Content of test file 2",
+                "/tmp/test_file3.bin": b"Binary content for test file 3",
+            }
+
+            for file_path, content in test_files.items():
+                result = await self.sandbox.files.write(file_path, content)
+                assert isinstance(result, WriteInfo)
+                exists = await self.sandbox.files.exists(file_path)
+                assert exists
+                print(f"Created file: {file_path}")
+
+            # 3. 在pause之前删除一个文件
+            print("Step 3: Deleting a file before pause...")
+            file_to_delete = "/tmp/test_file2.txt"
+            await self.sandbox.files.remove(file_to_delete)
+            exists = await self.sandbox.files.exists(file_to_delete)
+            assert not exists
+            print(f"Deleted file: {file_to_delete}")
+
+            # 4. 创建目录和文件
+            print("Step 4: Creating directory and nested files...")
+            await self.sandbox.files.make_dir("/tmp/test_dir")
+            await self.sandbox.files.write(
+                "/tmp/test_dir/nested_file.txt", "Nested file content"
+            )
+            assert await self.sandbox.files.exists("/tmp/test_dir")
+            assert await self.sandbox.files.exists("/tmp/test_dir/nested_file.txt")
+            print("Created directory and nested file")
+
+            # 5. Pause沙箱
+            print("Step 5: Pausing sandbox...")
+            await self.sandbox.beta_pause()
+            print("Sandbox paused successfully")
+
+            # 等待一小段时间确保pause完成
+            await asyncio.sleep(2)
+
+            # 6. Connect到暂停的沙箱
+            print("Step 6: Connecting to paused sandbox...")
+            connected_sandbox = await AsyncSandbox.connect(self.sandbox_id)
+            assert connected_sandbox.sandbox_id == self.sandbox_id
+            print("Connected to sandbox successfully")
+
+            # 等待一小段时间确保connect完成
+            await asyncio.sleep(2)
+
+            # 7. 在connect之后校验文件操作结果
+            print("Step 7: Verifying file operations after connect...")
+
+            # 校验应该存在的文件
+            exists = await connected_sandbox.files.exists("/tmp/test_file1.txt")
+            assert exists, "test_file1.txt should exist"
+            file1_content = await connected_sandbox.files.read(
+                "/tmp/test_file1.txt", format="text"
+            )
+            assert "Content of test file 1" in file1_content
+            print("✅ Verified test_file1.txt exists and content is correct")
+
+            exists = await connected_sandbox.files.exists("/tmp/test_file3.bin")
+            assert exists, "test_file3.bin should exist"
+            file3_content = await connected_sandbox.files.read(
+                "/tmp/test_file3.bin", format="bytes"
+            )
+            assert bytes(file3_content) == b"Binary content for test file 3"
+            print("✅ Verified test_file3.bin exists and content is correct")
+
+            # 校验应该不存在的文件（被删除的）
+            exists = await connected_sandbox.files.exists("/tmp/test_file2.txt")
+            assert not exists, "test_file2.txt should not exist"
+            print("✅ Verified test_file2.txt was deleted correctly")
+
+            # 校验目录和嵌套文件
+            exists = await connected_sandbox.files.exists("/tmp/test_dir")
+            assert exists, "test_dir should exist"
+            exists = await connected_sandbox.files.exists(
+                "/tmp/test_dir/nested_file.txt"
+            )
+            assert exists, "nested_file.txt should exist"
+            nested_content = await connected_sandbox.files.read(
+                "/tmp/test_dir/nested_file.txt", format="text"
+            )
+            assert "Nested file content" in nested_content
+            print("✅ Verified directory and nested file exist and content is correct")
+
+            # 8. 在connect后的沙箱中执行一些操作验证功能正常
+            print("Step 8: Verifying sandbox functionality after connect...")
+            result = await connected_sandbox.commands.run(
+                "echo 'Test command after connect'"
+            )
+            assert result.exit_code == 0
+            assert "Test command after connect" in result.stdout
+            print("✅ Verified sandbox commands work after connect")
+
+            # 9. 创建新文件验证写入功能正常
+            print("Step 9: Testing file write after connect...")
+            new_file_path = "/tmp/new_file_after_connect.txt"
+            await connected_sandbox.files.write(
+                new_file_path, "New file created after connect"
+            )
+            exists = await connected_sandbox.files.exists(new_file_path)
+            assert exists
+            new_content = await connected_sandbox.files.read(
+                new_file_path, format="text"
+            )
+            assert "New file created after connect" in new_content
+            print("✅ Verified file write works after connect")
+
+            self.log_test_result(
+                "Connect and Pause with Files", True, "All verifications passed"
+            )
+            return True
+
+        except Exception as e:
+            self.log_test_result("Connect and Pause with Files", False, str(e))
+            print(f"Test failed with error: {e}")
+            import traceback
+
+            traceback.print_exc()
+            return False
+
+    async def test_connect_pause_multiple_operations(self):
+        """测试多次pause和connect操作"""
+        try:
+            # 1. 创建沙箱
+            print("Test 2: Creating sandbox for multiple operations test...")
+            sandbox = await AsyncSandbox.create(
+                template="base",
+                timeout=3600,
+                metadata={"test": "connect_pause_multiple_async"},
+            )
+            sandbox_id = sandbox.sandbox_id
+            print(f"Created sandbox with ID: {sandbox_id}")
+
+            # 2. 第一次操作：创建文件
+            print("First operation: Creating files...")
+            await sandbox.files.write("/tmp/multi_test1.txt", "First batch")
+            await sandbox.files.write("/tmp/multi_test2.txt", "First batch")
+
+            # 3. 第一次pause
+            print("First pause...")
+            await sandbox.beta_pause()
+            await asyncio.sleep(2)
+
+            # 4. 第一次connect
+            print("First connect...")
+            sandbox = await AsyncSandbox.connect(sandbox_id)
+            await asyncio.sleep(2)
+
+            # 5. 验证第一次的文件存在
+            assert await sandbox.files.exists("/tmp/multi_test1.txt")
+            assert await sandbox.files.exists("/tmp/multi_test2.txt")
+
+            # 6. 第二次操作：删除一个文件，创建新文件
+            print("Second operation: Deleting and creating files...")
+            await sandbox.files.remove("/tmp/multi_test1.txt")
+            await sandbox.files.write("/tmp/multi_test3.txt", "Second batch")
+
+            # 7. 第二次pause
+            print("Second pause...")
+            await sandbox.beta_pause()
+            await asyncio.sleep(2)
+
+            # 8. 第二次connect
+            print("Second connect...")
+            sandbox = await AsyncSandbox.connect(sandbox_id)
+            await asyncio.sleep(2)
+
+            # 9. 验证第二次操作的结果
+            exists = await sandbox.files.exists("/tmp/multi_test1.txt")
+            assert not exists, "multi_test1.txt should be deleted"
+            exists = await sandbox.files.exists("/tmp/multi_test2.txt")
+            assert exists, "multi_test2.txt should exist"
+            exists = await sandbox.files.exists("/tmp/multi_test3.txt")
+            assert exists, "multi_test3.txt should exist"
+            content = await sandbox.files.read("/tmp/multi_test3.txt", format="text")
+            assert "Second batch" in content
+
+            # 清理
+            await sandbox.kill()
+
+            self.log_test_result(
+                "Multiple Connect and Pause Operations",
+                True,
+                "All verifications passed",
+            )
+            return True
+
+        except Exception as e:
+            self.log_test_result("Multiple Connect and Pause Operations", False, str(e))
+            print(f"Test failed with error: {e}")
+            import traceback
+
+            traceback.print_exc()
+            return False
+
+    async def cleanup(self):
+        """清理资源"""
+        if self.sandbox:
+            try:
+                await self.sandbox.kill()
+                print("Sandbox cleaned up successfully")
+            except Exception as e:
+                print(f"Error cleaning up sandbox: {e}")
+
+    def print_summary(self):
+        """打印测试摘要"""
+        total_tests = len(self.test_results)
+        passed_tests = sum(1 for r in self.test_results if r["success"])
+        failed_tests = total_tests - passed_tests
+
+        print("\n" + "=" * 60)
+        print("Connect and Pause Test Report (Async)")
+        print("=" * 60)
+        print(f"总测试数: {total_tests}")
+        print(f"通过数: {passed_tests}")
+        print(f"失败数: {failed_tests}")
+        print(f"成功率: {(passed_tests/total_tests*100):.1f}%")
+
+        if failed_tests > 0:
+            print(f"\n失败的测试:")
+            for result in self.test_results:
+                if not result["success"]:
+                    print(f"  ❌ {result['test']}: {result['message']}")
+
+        print("=" * 60)
+
+
+async def main():
+    """主函数"""
+    test_suite = ConnectPauseAsyncTest()
+
+    try:
+        # 运行测试
+        await test_suite.test_connect_pause_with_files()
+        await test_suite.test_connect_pause_multiple_operations()
+    finally:
+        await test_suite.cleanup()
+        test_suite.print_summary()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())