feat add tests

Author: Nigma-Ks

backend/config/paths.py

@@ -7,7 +7,7 @@ def get_tmp_thread_files(uid):
 
 def get_thread_filepath(uid, filepath):
     base_prefix = TMP_FILE_DIR + "/"
-    new_prefix = f"{TMP_FILE_DIR}/{uid}_"
+    new_prefix = f"{TMP_FILE_DIR}/{uid}"
     return filepath.replace(base_prefix, new_prefix)
 
 

backend/config/tests/test_filepaths.py

@@ -0,0 +1,31 @@
+import pytest
+
+from backend.config.paths import (
+    get_tmp_thread_files,
+    RESULTS_DIR,
+    UPLOAD_DIR,
+    get_thread_filepath,
+    TMP_FILE_DIR,
+)
+
+
+@pytest.mark.parametrize(
+    "uid, filepath, expected",
+    [
+        ("", RESULTS_DIR, RESULTS_DIR),
+        ("123", UPLOAD_DIR, TMP_FILE_DIR + "/123uploads"),
+    ],
+)
+def test_get_thread_filepath(uid, filepath, expected):
+    assert get_thread_filepath(uid, filepath) == expected
+
+
+@pytest.mark.parametrize(
+    "uid, expected",
+    [
+        ("", [UPLOAD_DIR, RESULTS_DIR]),
+        ("123", [TMP_FILE_DIR + "/123uploads", TMP_FILE_DIR + "/123results"]),
+    ],
+)
+def test_get_created_tmp_files(uid, expected):
+    assert get_tmp_thread_files(uid) == expected

backend/file_utils/tests/test_file_utils.py

@@ -0,0 +1,72 @@
+import shutil
+
+import pytest, os, csv
+
+from backend.config.paths import TMP_FILE_DIR
+from backend.file_utils.csv_methods_writer import write_launch_info_to_csv
+from backend.file_utils.files import reset_dirs, save_upload_file
+from unittest.mock import Mock
+from fastapi import UploadFile
+from io import BytesIO
+
+
+@pytest.mark.parametrize(
+    "filepaths",
+    [
+        ([TMP_FILE_DIR + "/test0"]),
+        ([TMP_FILE_DIR + "/test1", TMP_FILE_DIR + "/test2"]),
+    ],
+)
+def test_removing_tmp_files(filepaths):
+    for path in filepaths:
+        os.makedirs(path)
+    reset_dirs(filepaths)
+    for path in filepaths:
+        assert not os.path.exists(path)
+
+
+def test_save_upload_file():
+    test_content = b"Hello, World!" * 1000
+    upload_dir = TMP_FILE_DIR + "/test_save_upload/"
+    dest_path = upload_dir + "test.txt"
+
+    mock_file = BytesIO(test_content)
+    upload_file = Mock(spec=UploadFile)
+    upload_file.file = mock_file
+
+    save_upload_file(upload_file, dest_path)
+
+    assert os.path.exists(dest_path)
+    with open(dest_path, "rb") as f:
+        saved_content = f.read()
+    shutil.rmtree(upload_dir)
+    assert saved_content == test_content
+
+
+def test_write_launch_info_to_csv():
+    upload_dir = TMP_FILE_DIR + "/test_write_launch_info_to_csv/"
+    os.makedirs(upload_dir)
+    output_file = upload_dir + "launch_info.csv"
+    parsed_methods = [
+        "ManuallyCollected.dll,BinSearchMain",
+        "ManuallyCollected.dll,BellmanFord",
+        "ManuallyCollected.dll,BinaryMaze1BFS",
+    ]
+
+    write_launch_info_to_csv(parsed_methods=parsed_methods, output_file=output_file)
+
+    assert os.path.exists(output_file)
+
+    with open(output_file, "r") as f:
+        reader = csv.reader(f)
+        rows = list(reader)
+
+    assert rows[0] == ["dll", "method"]
+
+    expected_data = [
+        ["ManuallyCollected.dll", "BinSearchMain"],
+        ["ManuallyCollected.dll", "BellmanFord"],
+        ["ManuallyCollected.dll", "BinaryMaze1BFS"],
+    ]
+    shutil.rmtree(upload_dir)
+    assert rows[1:] == expected_data

backend/launch_service/app_setup.py

@@ -10,19 +10,19 @@
 from backend.utils.methods_handler import Methods
 
 IMAGE_NAME = "pysymgym-test"
+URL = "https://raw.githubusercontent.com/PySymGym/PySymGym/main/maps/DotNet/Maps/dataset.json"
 
 
-def fetch_dataset():
-    url = "https://raw.githubusercontent.com/PySymGym/PySymGym/main/maps/DotNet/Maps/dataset.json"
+def fetch_dataset(url, data_upload_file):
     print(f"Downloading dataset from {url} ...")
     resp = requests.get(url)
     resp.raise_for_status()
 
-    with open(DATASET_FILE, "wb") as f:
+    with open(data_upload_file, "wb") as f:
         f.write(resp.content)
 
-    print(f"Dataset saved to {DATASET_FILE}")
-    return DATASET_FILE
+    print(f"Dataset saved to {data_upload_file}")
+    return data_upload_file
 
 
 def build_container():
@@ -35,13 +35,15 @@ def build_container():
     os.makedirs(RESOURCES_DIR, exist_ok=True)
 
 
-def update_frontend_selection_options():
+def update_frontend_selection_options(dataset_file, selection_options_file):
     print("Updating frontend selection options...")
-    selection_tree = Methods.parse_dataset_file_for_front_selection(DATASET_FILE)
-    Methods.write_selection_dataset_to_front_file(selection_tree, METHODS_TS_FILE)
+    selection_tree = Methods.parse_dataset_file_for_front_selection(dataset_file)
+    Methods.write_selection_dataset_to_front_file(
+        selection_tree, selection_options_file
+    )
 
 
 if __name__ == "__main__":
     build_container()
-    fetch_dataset()
-    update_frontend_selection_options()
+    fetch_dataset(URL, DATASET_FILE)
+    update_frontend_selection_options(DATASET_FILE, METHODS_TS_FILE)

backend/launch_service/tests/test_setup_functions.py

@@ -0,0 +1,42 @@
+import os
+import shutil
+from unittest.mock import Mock, patch
+
+from backend.config.paths import TMP_FILE_DIR, DOCKER_DIR, RESOURCES_DIR
+from backend.launch_service.app_setup import fetch_dataset, IMAGE_NAME, build_container
+
+
+def test_fetch_dataset():
+    url = "https://example.com/dataset.csv"
+    test_content = b"dll,method\nManuallyCollected.dll,BinSearchMain"
+    upload_dir = TMP_FILE_DIR + "/test_dataset_upload/"
+    os.makedirs(upload_dir)
+    data_file = upload_dir + "dataset.json"
+
+    with patch("requests.get") as mock_get:
+        mock_response = Mock()
+        mock_response.content = test_content
+        mock_response.raise_for_status = Mock()
+        mock_get.return_value = mock_response
+
+        result = fetch_dataset(url, data_file)
+
+    assert result == data_file
+    assert os.path.exists(data_file)
+    with open(data_file, "rb") as f:
+        assert f.read() == test_content
+    shutil.rmtree(upload_dir)
+    mock_get.assert_called_once_with(url)
+    mock_response.raise_for_status.assert_called_once()
+
+
+def test_successful_build():
+    with patch("subprocess.run") as mock_run:
+        mock_run.return_value = Mock(returncode=0)
+
+        with patch("os.makedirs") as mock_makedirs:
+            build_container()
+
+    expected_command = ["docker", "build", "--no-cache", "-t", IMAGE_NAME, DOCKER_DIR]
+    mock_run.assert_called_once_with(expected_command, check=True)
+    mock_makedirs.assert_called_once_with(RESOURCES_DIR, exist_ok=True)

backend/utils/results_sender.py

@@ -18,7 +18,7 @@ def send_folder_by_email(
     to_email: str,
     folder_path: str,
     experiment_name: str,
-    model_name: str,
+    model_file_name: str,
 ):
     load_dotenv()
     from_email = require_env("EMAIL")
@@ -34,14 +34,15 @@ def send_folder_by_email(
         for file in folder.rglob("*"):
             zipf.write(file, file.relative_to(folder))
 
+    model_name = model_file_name[:-5]
     msg = EmailMessage()
     msg["From"] = from_email
     msg["To"] = to_email
-    msg["Subject"] = f"Results of {experiment_name} with {model_name[:-5]}"
+    msg["Subject"] = f"Results of {experiment_name} with {model_name}"
 
     message = (
         f"\n\n"
-        f"The results of the experiment '{experiment_name}' using the model '{model_name[:-5]}' are ready. "
+        f"The results of the experiment '{experiment_name}' using the model '{model_name}' are ready. "
         f"Please find the details in the attached ZIP file.\n\n"
         f"Contents of the ZIP:\n"
         f" - Model run results: artifact_run_ai folder\n"

backend/utils/tests/resources/AI.csv

@@ -0,0 +1,4 @@
+method,tests,errors,coverage,total_time_sec
+BinSearchMain,1,1,100.0,0
+Switches1,29,0,100.0,96
+Switches2,1,0,15.0,97

backend/utils/tests/resources/ExecutionTreeContributedCoverage.csv

@@ -0,0 +1,4 @@
+method,tests,errors,coverage,total_time_sec
+BinSearchMain,1,1,100.0,0
+Switches1,30,0,100.0,3
+Switches2,2,0,38.0,96

backend/utils/tests/resources/launch_info.csv

@@ -0,0 +1,4 @@
+dll,method
+ManuallyCollected.dll,BinSearchMain
+ManuallyCollected.dll,Switches1
+ManuallyCollected.dll,Switches2