llm-shield-api 0.1.0

Production-grade REST API for LLM Shield
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! Integration tests for scanners endpoint

use axum::http::StatusCode;
use llm_shield_api::models::response::ListScannersResponse;
use llm_shield_api::router::create_router_with_state;

mod common;
use common::{create_test_state, get_request, parse_json};

#[tokio::test]
async fn test_list_scanners_endpoint() {
    let state = create_test_state();
    let app = create_router_with_state(state);

    let (status, body) = get_request(app, "/v1/scanners").await;

    assert_eq!(status, StatusCode::OK);

    let response: ListScannersResponse = parse_json(&body);
    assert_eq!(response.total_count, 3); // test_scanner_1, test_scanner_2, test_failing_scanner
    assert_eq!(response.scanners.len(), 3);

    // Verify scanner metadata is present
    assert!(response.scanners.iter().any(|s| s.name == "test_scanner_1"));
    assert!(response.scanners.iter().any(|s| s.name == "test_scanner_2"));
    assert!(response
        .scanners
        .iter()
        .any(|s| s.name == "test_failing_scanner"));
}

#[tokio::test]
async fn test_list_scanners_metadata() {
    let state = create_test_state();
    let app = create_router_with_state(state);

    let (status, body) = get_request(app, "/v1/scanners").await;

    assert_eq!(status, StatusCode::OK);

    let response: ListScannersResponse = parse_json(&body);

    // Verify each scanner has required metadata
    for scanner in &response.scanners {
        assert!(!scanner.name.is_empty());
        assert!(!scanner.scanner_type.is_empty());
        assert!(!scanner.version.is_empty());
        assert!(!scanner.description.is_empty());
    }
}

#[tokio::test]
async fn test_list_scanners_types() {
    let state = create_test_state();
    let app = create_router_with_state(state);

    let (status, body) = get_request(app, "/v1/scanners").await;

    assert_eq!(status, StatusCode::OK);

    let response: ListScannersResponse = parse_json(&body);

    // All test scanners are Input type
    for scanner in &response.scanners {
        assert_eq!(scanner.scanner_type, "Input");
    }
}