test_loadtester.py

import os
import subprocess
import json

def test_loadtester_get():
    url = "https://httpbin.org/get"
    auth_header = "none"
    method = "GET"
    data = "none"
    num_users = "1"
    num_requests = "1"

    env = os.environ.copy()
    env["URL"] = url
    env["AUTH_HEADER"] = auth_header
    env["DATA"] = data
    env["METHOD"] = method
    env["NUM_USERS"] = num_users
    env["NUM_REQUESTS"] = num_requests

    result = subprocess.run(["python", "LoadTester.py"], env=env, text=True, capture_output=True)

    if result.stderr:
        raise ValueError("stderr:", result.stderr) 
    print("stdout:", result.stdout)
    assert "Média de resposta" in result.stdout, "Média de resposta não encontrada na saída"
    assert "Taxa de sucesso" in result.stdout, "Taxa de saída não encontrada na saída"

def test_loadtester_post():
    url = "https://httpbin.org/post"
    auth_header = "none"
    method = "POST"
    data = json.dumps({"key": "value"})
    num_users = "1"
    num_requests = "1"

    env = os.environ.copy()
    env["URL"] = url
    env["AUTH_HEADER"] = auth_header
    env["DATA"] = data
    env["METHOD"] = method
    env["NUM_USERS"] = num_users
    env["NUM_REQUESTS"] = num_requests

    result = subprocess.run(["python", "LoadTester.py"], env=env, text=True, capture_output=True)

    if result.stderr:
        raise ValueError("stderr:", result.stderr) 
    print("stdout:", result.stdout)
    assert "Média de resposta" in result.stdout, "Média de resposta não encontrada na saída"
    assert "Taxa de sucesso" in result.stdout, "Taxa de saída não encontrada na saída"

if __name__ == "__main__":
    test_loadtester_get()
    print("LoadTester GET teste concluído com sucesso.")
    test_loadtester_post()
    print("LoadTester POST teste concluído com sucesso.")