| import os |
| import time |
| import subprocess |
| from typing import Dict, Any |
| from flow_modules.aiflows.InterpreterFlowModule import InterpreterAtomicFlow |
|
|
|
|
| class CodeTestingAtomicFlow(InterpreterAtomicFlow): |
| """This class inherits from InterpreterAtomicFlow and is used to test code. |
| |
| *Input Interface*: |
| - `temp_code_file_location`: Location of the file containing the code to be tested. |
| |
| *Output Interface*: |
| - `feedback`: Feedback from the test (i.e. test results). |
| |
| *Configuration Parameters*: |
| - `input_interface`: the input interface of the atomic flow. |
| - `output_interface`: the output interface of the atomic flow. |
| """ |
| def _open_file_and_wait_for_file_update(self, file_location, check_interval=1): |
| """Opens the file in VSCode and waits for the file to be updated. |
| :param file_location: The location of the file to be opened. |
| :type file_location: str |
| :param check_interval: The interval at which the file is checked for updates. |
| :type check_interval: int |
| """ |
| process = subprocess.Popen(["code", "--wait", file_location]) |
| while True: |
| if process.poll() is not None: |
| break |
| time.sleep(check_interval) |
| def _prepare_code_and_lang(self, input_data: Dict[str, Any]): |
| """Prepares the code and language to be passed to the interpreter. |
| :param input_data: The input data to the atomic flow. |
| :type input_data: Dict[str, Any] |
| """ |
| file_location = input_data["temp_code_file_location"] |
| input_data["language"] = "python" |
| with open(file_location, "r") as file: |
| code_str = file.read() |
| input_data["code"] = code_str |
|
|
| def _check_input(self, input_data: Dict[str, Any]): |
| """Checks if the input data contains the required keys. |
| :param input_data: The input data to the atomic flow. |
| :type input_data: Dict[str, Any] |
| :raises AssertionError: If the input data does not contain the required keys. |
| """ |
| assert "temp_code_file_location" in input_data, "temp_code_file_location not passed to CodeTestingAtomicFlow" |
|
|
| def _delete_file(self, file_location): |
| """Deletes the file at the given location. |
| :param file_location: The location of the file to be deleted. |
| :type file_location: str |
| """ |
| if os.path.exists(file_location): |
| os.remove(file_location) |
|
|
| def run( |
| self, |
| input_data: Dict[str, Any]): |
| """Runs the atomic flow. |
| :param input_data: The input data to the atomic flow. |
| :type input_data: Dict[str, Any] |
| :return: The output data of the atomic flow. |
| :rtype: Dict[str, Any] |
| """ |
| self._check_input(input_data) |
| file_loc = input_data["temp_code_file_location"] |
| self._open_file_and_wait_for_file_update(file_loc) |
| self._prepare_code_and_lang(input_data) |
| self._process_input_data(input_data) |
| response = self._call() |
| self._delete_file(file_loc) |
| if response.strip() == '': |
| response = "test passed" |
| return {"feedback": response} |
|
|
