Skip to content
This repository was archived by the owner on Jul 4, 2025. It is now read-only.

task: expand e2e Hardware #2076

Merged
merged 2 commits into from
Mar 11, 2025
Merged

task: expand e2e Hardware #2076

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
4d33e22
test: add new e2e
Mar 6, 2025
50cffd1
Merge branch 'dev' into harry/e2e-hardware
LeVinhGithub Mar 10, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
251 changes: 251 additions & 0 deletions engine/e2e-test/api/hardware/test_api_get_hardware.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,251 @@
import pytest
import requests
from utils.test_runner import start_server, stop_server
import jsonschema
from utils.logger import log_response
from utils.assertion import assert_equal


class TestApiGetHardware:

@pytest.fixture(autouse=True)
def setup_and_teardown(self):
# Setup
success = start_server()
if not success:
raise Exception("Failed to start server")

yield

# Teardown
stop_server()

def test_api_get_hardware_successfully(self):
hardware_url = f"http://localhost:3928/v1/hardware"
hardware_response = requests.get(hardware_url)
json_data_hardware = hardware_response.json()
log_response(json_data_hardware, "test_api_get_hardware_successfully")
assert_equal(hardware_response.status_code,200)

schema = {
"type": "object",
"properties": {
"cpu": {
"type": "object",
"properties": {
"arch": {
"type": "string",
"example": "amd64",
"description": "The architecture of the CPU."
},
"cores": {
"type": "integer",
"example": 8,
"description": "The number of CPU cores available."
},
"instructions": {
"type": "array",
"items": {
"type": "string"
},
"example": [
"fpu",
"mmx",
"sse",
"sse2",
"sse3",
"ssse3",
"sse4_1",
"sse4_2",
"pclmulqdq",
"avx",
"avx2",
"aes",
"f16c"
],
"description": "A list of supported CPU instruction sets."
},
"model": {
"type": "string",
"example": "AMD Ryzen Threadripper PRO 5955WX 16-Cores",
"description": "The model name of the CPU."
}
},
"required": [
"arch",
"cores",
"instructions",
"model"
]
},
"gpus": {
"type": "array",
"items": {
"type": "object",
"properties": {
"activated": {
"type": "boolean",
"example": True,
"description": "Indicates if the GPU is currently activated."
},
"additional_information": {
"type": "object",
"properties": {
"compute_cap": {
"type": "string",
"example": "8.6",
"description": "The compute capability of the GPU."
},
"driver_version": {
"type": "string",
"example": "535.183",
"description": "The version of the installed driver."
}
},
"required": [
"compute_cap",
"driver_version"
]
},
"free_vram": {
"type": "integer",
"example": 23983,
"description": "The amount of free VRAM in MB."
},
"id": {
"type": "string",
"example": "0",
"description": "Unique identifier for the GPU."
},
"name": {
"type": "string",
"example": "NVIDIA GeForce RTX 3090",
"description": "The name of the GPU model."
},
"total_vram": {
"type": "integer",
"example": 24576,
"description": "The total VRAM available in MB."
},
"uuid": {
"type": "string",
"example": "GPU-5206045b-2a1c-1e7d-6c60-d7c367d02376",
"description": "The universally unique identifier for the GPU."
},
"version": {
"type": "string",
"example": "12.2",
"description": "The version of the GPU."
}
},
"required": [
"activated",
"additional_information",
"free_vram",
"id",
"name",
"total_vram",
"uuid",
"version"
]
}
},
"os": {
"type": "object",
"properties": {
"name": {
"type": "string",
"example": "Ubuntu 24.04.1 LTS",
"description": "The name of the operating system."
},
"version": {
"type": "string",
"example": "24.04.1 LTS (Noble Numbat)",
"description": "The version of the operating system."
}
},
"required": [
"name",
"version"
]
},
"power": {
"type": "object",
"properties": {
"battery_life": {
"type": "integer",
"example": 0,
"description": "The percentage of battery life remaining."
},
"charging_status": {
"type": "string",
"example": "",
"description": "The charging status of the device."
},
"is_power_saving": {
"type": "boolean",
"example": False,
"description": "Indicates if the power-saving mode is enabled."
}
},
"required": [
"battery_life",
"charging_status",
"is_power_saving"
]
},
"ram": {
"type": "object",
"properties": {
"available": {
"type": "integer",
"example": 11100,
"description": "The amount of available RAM in MB."
},
"total": {
"type": "integer",
"example": 15991,
"description": "The total RAM in MB."
},
"type": {
"type": "string",
"example": "",
"description": "The type of RAM."
}
},
"required": [
"available",
"total",
"type"
]
},
"storage": {
"type": "object",
"properties": {
"available": {
"type": "integer",
"example": 0,
"description": "The amount of available storage in MB."
},
"total": {
"type": "integer",
"example": 0,
"description": "The total storage in MB."
},
"type": {
"type": "string",
"example": "",
"description": "The type of storage."
}
},
"required": [
"available",
"total",
"type"
]
}
}
}

# Validate response schema
jsonschema.validate(instance=json_data_hardware, schema=schema)
1 change: 1 addition & 0 deletions engine/e2e-test/runner/cortex-llamacpp-e2e-nightly.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
from api.thread.test_api_delete_thread import TestApiDeleteThread
from api.thread.test_api_get_thread import TestApiGetThread
from api.thread.test_api_get_list_thread import TestApiGetListThread
from api.hardware.test_api_get_hardware import TestApiGetHardware
from api.assistants.test_api_create_assistant import TestApiCreateAssistant
from api.assistants.test_api_get_list_assistant import TestApiGetListAssistant
from api.assistants.test_api_get_assistant import TestApiGetAssistant
Expand Down
2 changes: 1 addition & 1 deletion engine/e2e-test/runner/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,12 +36,12 @@
from api.thread.test_api_delete_thread import TestApiDeleteThread
from api.thread.test_api_get_thread import TestApiGetThread
from api.thread.test_api_get_list_thread import TestApiGetListThread
from api.hardware.test_api_get_hardware import TestApiGetHardware
from api.assistants.test_api_create_assistant import TestApiCreateAssistant
from api.assistants.test_api_get_list_assistant import TestApiGetListAssistant
from api.assistants.test_api_get_assistant import TestApiGetAssistant
from api.assistants.test_api_delete_assistant import TestApiDeleteAssistant


###
from cli.engines.test_cli_engine_get import TestCliEngineGet
from cli.engines.test_cli_engine_install import TestCliEngineInstall
Expand Down
Loading