spoolman2/tests_integration/tests/spool/test_find.py

"""Integration tests for the Spool API endpoint."""

from typing import Any

import httpx

URL = "http://spoolman:8000"


def test_find_spools(random_filament: dict[str, Any]):  # noqa: PLR0915
    """Test finding spools in the database."""
    # Setup
    result = httpx.post(
        f"{URL}/api/v1/spool",
        json={
            "filament_id": random_filament["id"],
            "remaining_weight": 1000,
            "location": "The Pantry",
            "lot_nr": "123456789",
        },
    )
    result.raise_for_status()
    spool_1 = result.json()

    result = httpx.post(
        f"{URL}/api/v1/spool",
        json={
            "filament_id": random_filament["id"],
            "remaining_weight": 1000,
            "location": "Living Room",
            "lot_nr": "987654321",
        },
    )
    result.raise_for_status()
    spool_2 = result.json()

    result = httpx.post(
        f"{URL}/api/v1/spool",
        json={
            "filament_id": random_filament["id"],
            "remaining_weight": 1000,
            "archived": True,
        },
    )
    result.raise_for_status()
    spool_3 = result.json()

    added_spools_by_id = {
        spool_1["id"]: spool_1,
        spool_2["id"]: spool_2,
        spool_3["id"]: spool_3,
    }

    # Execute - find all spools
    result = httpx.get(f"{URL}/api/v1/spool")
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 2
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]
        assert spool["archived"] is False

    # Execute - find all spools, including archived
    result = httpx.get(f"{URL}/api/v1/spool?allow_archived=true")
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 3
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]

    # Execute - find spools by filament name
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"filament_name": random_filament["name"]},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 2
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]

    # Execute - find spools by filament id
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"filament_id": random_filament["id"]},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 2
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]

    # Execute - find spools by filament material
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"filament_material": random_filament["material"]},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 2
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]

    # Execute - find spools by filament vendor name
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"vendor_name": random_filament["vendor"]["name"]},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 2
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]

    # Execute - find spools by filament vendor id
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"vendor_id": random_filament["vendor"]["id"]},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 2
    for spool in spools:
        assert spool == added_spools_by_id[spool["id"]]

    # Execute - find spools by location
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"location": "The Pantry"},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 1
    assert spools[0] == added_spools_by_id[spool_1["id"]]

    # Execute - find spools by lot nr
    result = httpx.get(
        f"{URL}/api/v1/spool",
        params={"lot_nr": "123456789"},
    )
    result.raise_for_status()

    # Verify
    spools = result.json()
    assert len(spools) == 1
    assert spools[0] == added_spools_by_id[spool_1["id"]]

    # Clean up
    httpx.delete(f"{URL}/api/v1/spool/{spool_1['id']}").raise_for_status()
    httpx.delete(f"{URL}/api/v1/spool/{spool_2['id']}").raise_for_status()
    httpx.delete(f"{URL}/api/v1/spool/{spool_3['id']}").raise_for_status()