Link all challenges to benchmark python hook (#4786)

tests/challenges/kubernetes/test_kubernetes_template_challenge_a.py

@@ -1,23 +1,26 @@
+from typing import Any, Dict
+
 import pytest
 import yaml
 from pytest_mock import MockerFixture
 
-from autogpt.agent import Agent
-from autogpt.commands.file_operations import read_file
+from autogpt.workspace import Workspace
 from tests.challenges.challenge_decorator.challenge_decorator import challenge
-from tests.challenges.utils import get_workspace_path_from_agent, run_interaction_loop
+from tests.challenges.utils import get_workspace_path, run_challenge
 
 CYCLE_COUNT = 3
 OUTPUT_LOCATION = "kube.yaml"
+USER_INPUTS = ["Write a simple kubernetes deployment file and save it as a kube.yaml."]
 
 
 @challenge()
 def test_kubernetes_template_challenge_a(
-    kubernetes_agent: Agent,
     monkeypatch: pytest.MonkeyPatch,
     patched_api_requestor: MockerFixture,
     level_to_run: int,
     challenge_name: str,
+    workspace: Workspace,
+    patched_make_workspace: pytest.fixture,
 ) -> None:
     """
     Test the challenge_a function in a given agent by mocking user inputs
@@ -28,16 +31,23 @@ def test_kubernetes_template_challenge_a(
         monkeypatch (pytest.MonkeyPatch)
         level_to_run (int)
     """
-    run_interaction_loop(
-        monkeypatch, kubernetes_agent, CYCLE_COUNT, challenge_name, level_to_run
+    run_challenge(
+        challenge_name,
+        level_to_run,
+        monkeypatch,
+        USER_INPUTS[level_to_run - 1],
+        CYCLE_COUNT,
     )
 
-    file_path = get_workspace_path_from_agent(kubernetes_agent, OUTPUT_LOCATION)
-    content = read_file(file_path, kubernetes_agent)
+    file_path = get_workspace_path(workspace, OUTPUT_LOCATION)
+    with open(file_path, "r") as file:
+        content_string = file.read()
 
     for word in ["apiVersion", "kind", "metadata", "spec"]:
-        assert word in content, f"Expected the file to contain {word}"
+        assert word in content_string, f"Expected the file to contain {word}"
 
-    content = yaml.safe_load(content)
+    yaml_as_dict: Dict[str, Any] = yaml.safe_load(content_string)
     for word in ["Service", "Deployment", "Pod"]:
-        assert word in content["kind"], f"Expected the file to contain {word}"
+        assert word in yaml_as_dict.get(
+            "kind", ""
+        ), f"Expected the file to contain {word}"