src/services: Init unit-tests for kill & suite service

Signed-off-by: Kamoltat Sirivadhna <[email protected]>

Author: kamoltat

Diff for: pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+pythonpath = src

Diff for: src/services/suite.py

@@ -66,7 +66,7 @@ def make_run_name(run_dic):
             str(run_dic["timestamp"]),
             run_dic["suite"],
             run_dic["ceph_branch"],
-            run_dic["kernel_branch"] or "-",
+            run_dic["kernel_branch"] or "distro",
             run_dic["flavor"],
             worker,
         ]

Diff for: src/services/test_kill.py

@@ -0,0 +1,36 @@
+import pytest
+from fastapi.testclient import TestClient
+from main import app
+from services.helpers import get_token, get_username, get_run_details
+from teuthology.kill import main as teuth_main
+from unittest.mock import patch
+from routes.kill import Depends as depends
+import json
+
+client = TestClient(app)
+
+def override_get_token():
+    return {"access_token": "token_123", "token_type": "bearer"}
+
+async def override_get_username():
+    return {"user": "kamoltat"}
+
+async def override_get_run_details():
+    return {"id": "7451978", "user": "kamoltat"}
+
+def test_missing_access_token():
+    response = client.post("/kill", json={"run": "run1"})
+    assert response.status_code == 401
+    assert response.json() == {"detail": "You need to be logged in"}
+
+def test_kill_run_success():
+    app.dependency_overrides[get_token] = override_get_token
+    app.dependency_overrides[get_username] = override_get_username
+    app.dependency_overrides[get_run_details] = override_get_run_details
+    body = {
+        "owner": "kamoltat" ,
+        "run": "run1"
+    }
+    response = client.post("/kill", json=json.dumps(body))
+    assert response.status_code == 200
+    assert response.json() == {"kill": "success"}

Diff for: src/services/test_suite.py

@@ -0,0 +1,71 @@
+from fastapi.testclient import TestClient
+from main import app
+from suite import *
+
+"""
+Created a TestClient instance and define a dictionary run_dic with test data for the make_run_name function,
+then called the function with the test data and assert that the result matches the expected output
+"""
+
+def test_make_run_name():
+    client = TestClient(app)
+    run_dic = {
+        "user": "testuser",
+        "timestamp": "2022-03-21_14:30:00",
+        "suite": "rados",
+        "ceph_branch": "ceph1",
+        "kernel_branch": "kernel1",
+        "flavor": "test-flavor",
+        "machine_type": "test-machine"
+    }
+    expected_result = "testuser-2022-03-21_14:30:00-rados-ceph1-kernel1-test-flavor-test-machine"
+    assert make_run_name(run_dic) == expected_result
+
+"""
+    Test the `make_run_name` function with an input dictionary containing a single worker machine type.
+"""
+def test_make_run_name_with_single_worker():
+    run_dic = {
+        "user": "test_user",
+        "timestamp": "2022-03-21_14:30:00",
+        "suite": "rados",
+        "ceph_branch": "ceph1",
+        "kernel_branch": "kernel1",
+        "flavor": "test-flavor",
+        "machine_type": "worker1"
+    }
+    expected_run_name = "test_user-2022-03-21_14:30:00-rados-ceph1-kernel1-test-flavor-worker1"
+    assert make_run_name(run_dic) == expected_run_name
+
+"""
+Test the `make_run_name` function with a multi-machine type input dictionary.
+"""
+
+def test_make_run_name_with_multi_worker():
+    run_dic = {
+        "user": "test_user",
+        "timestamp": "2022-03-21_14:30:00",
+        "suite": "rados",
+        "ceph_branch": "ceph1",
+        "kernel_branch": "kernel1",
+        "flavor": "test-flavor",
+        "machine_type": "worker1,worker2,worker3"
+    }
+    expected_run_name = "test_user-2022-03-21_14:30:00-rados-ceph1-kernel1-test-flavor-multi"
+    assert make_run_name(run_dic) == expected_run_name
+
+"""
+Test the function for no kernel branch
+"""
+def test_make_run_name_with_no_kernel_branch():
+    run_dic = {
+        "user": "teuthology",
+        "timestamp": "2022-03-21_14:30:00",
+        "suite": "rados",
+        "ceph_branch": "ceph1",
+        "kernel_branch": None,
+        "flavor": "test-flavor",
+        "machine_type": "test-machine"
+    }
+    expected_run_name = "teuthology-2022-03-21_14:30:00-rados-ceph1-distro-test-flavor-test-machine"
+    assert make_run_name(run_dic) == expected_run_name