agent/test_run_python_file.py at main · bluesky-o/agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from functions.run_python_file import run_python_file


def test_run_python_file():
    print("Test 1: Run main.py without arguments")
    result = run_python_file("calculator", "main.py")
    print(result)
    print("-" * 60)

    print("Test 2: Run main.py with arguments")
    result = run_python_file("calculator", "main.py", ["3 + 5"])
    print(result)
    print("-" * 60)

    print("Test 3: Run tests.py")
    result = run_python_file("calculator", "tests.py")
    print(result)
    print("-" * 60)

    print("Test 4: Attempt path traversal")
    result = run_python_file("calculator", "../main.py")
    print(result)
    print("-" * 60)

    print("Test 5: Non-existent file")
    result = run_python_file("calculator", "nonexistent.py")
    print(result)
    print("-" * 60)

    print("Test 6: Non-Python file")
    result = run_python_file("calculator", "lorem.txt")
    print(result)
    print("-" * 60)


if __name__ == "__main__":
    test_run_python_file()