mirror of
https://github.com/bec-project/bec_atlas.git
synced 2025-07-14 07:01:48 +02:00
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def logged_in_client(backend):
|
|
client, _ = backend
|
|
response = client.post(
|
|
"/api/v1/user/login", json={"username": "admin@bec_atlas.ch", "password": "admin"}
|
|
)
|
|
assert response.status_code == 200
|
|
token = response.json()
|
|
assert isinstance(token, str)
|
|
assert len(token) > 20
|
|
return client
|
|
|
|
|
|
@pytest.mark.timeout(60)
|
|
@pytest.mark.parametrize("realm, num_deployments", [("test", 0), ("demo_beamline_1", 1)])
|
|
def test_get_deployment_by_realm(logged_in_client, realm, num_deployments):
|
|
"""
|
|
Test that the login endpoint returns a token.
|
|
"""
|
|
client = logged_in_client
|
|
response = client.get("/api/v1/deployments/realm", params={"realm": realm})
|
|
assert response.status_code == 200
|
|
deployments = response.json()
|
|
assert len(deployments) == num_deployments
|
|
|
|
|
|
@pytest.mark.timeout(60)
|
|
def test_get_deployment_by_id(logged_in_client):
|
|
"""
|
|
Test that the login endpoint returns a token.
|
|
"""
|
|
client = logged_in_client
|
|
|
|
deployments = client.get(
|
|
"/api/v1/deployments/realm", params={"realm": "demo_beamline_1"}
|
|
).json()
|
|
deployment_id = deployments[0]["_id"]
|
|
|
|
response = client.get("/api/v1/deployments/id", params={"deployment_id": deployment_id})
|
|
assert response.status_code == 200
|
|
deployment = response.json()
|
|
assert deployment["_id"] == deployment_id
|
|
assert deployment["realm_id"] == "demo_beamline_1"
|
|
|
|
|
|
@pytest.mark.timeout(60)
|
|
def test_get_deployment_by_id_wrong_id(logged_in_client):
|
|
"""
|
|
Test that the login endpoint returns a token.
|
|
"""
|
|
client = logged_in_client
|
|
|
|
response = client.get("/api/v1/deployments/id", params={"deployment_id": "wrong_id"})
|
|
assert response.status_code == 400
|
|
assert response.json() == {"detail": "Invalid deployment id"}
|