Added code_execution_config to tests.

Author: asaharland

tests/test_kubectl_executor_agent.py

@@ -5,7 +5,13 @@
 
 def test_execute_code_blocks_when_read_only_then_allow_read_execution():
     """Allow read commands when read_only=False."""
-    executor_agent = KubectlExecutorAgent(name="test_kait", read_only=True)
+    executor_agent = KubectlExecutorAgent(
+        name="test_kait",
+        read_only=True,
+        code_execution_config={
+            "use_docker": False,
+        },
+    )
 
     executor_agent.run_code = MagicMock(return_value=(0, "", ""))
 
@@ -17,7 +23,13 @@ def test_execute_code_blocks_when_read_only_then_allow_read_execution():
 
 def test_execute_code_blocks_when_read_only_then_prevent_create_execution():
     """Prevent CUD commands when read_only=True."""
-    executor_agent = KubectlExecutorAgent(name="test_kait", read_only=True)
+    executor_agent = KubectlExecutorAgent(
+        name="test_kait",
+        read_only=True,
+        code_execution_config={
+            "use_docker": False,
+        },
+    )
 
     code_blocks = [("bash", "kubectl create deployment nginx --image nginx")]
 
@@ -27,7 +39,13 @@ def test_execute_code_blocks_when_read_only_then_prevent_create_execution():
 
 def test_execute_code_blocks_when_not_read_only_then_allow_create_execution():
     """Allow CRUD commands when read_only=False."""
-    executor_agent = KubectlExecutorAgent(name="test_kait", read_only=False)
+    executor_agent = KubectlExecutorAgent(
+        name="test_kait",
+        read_only=False,
+        code_execution_config={
+            "use_docker": False,
+        },
+    )
 
     executor_agent.run_code = MagicMock(return_value=(0, "", ""))
 
@@ -39,7 +57,13 @@ def test_execute_code_blocks_when_not_read_only_then_allow_create_execution():
 
 def test_execute_code_blocks_when_blocking_command_used_then_prevent_execution():
     """Prevent commands that block execution."""
-    executor_agent = KubectlExecutorAgent(name="test_kait", read_only=False)
+    executor_agent = KubectlExecutorAgent(
+        name="test_kait",
+        read_only=False,
+        code_execution_config={
+            "use_docker": False,
+        },
+    )
 
     executor_agent.run_code = MagicMock(return_value=(0, "", ""))