oxibonsai-runtime 0.1.4

Inference runtime, sampling, tokenizer, and server for OxiBonsai
Documentation 
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

//! Integration tests for the web UI module.
//!
//! Tests cover:
//! - Static properties of the embedded HTML
//! - Router construction
//! - HTTP endpoint behaviour via `axum::test` helpers

use axum::{
    body::Body,
    http::{Request, StatusCode},
};
use oxibonsai_runtime::web_ui::{create_ui_router, CHAT_UI_HTML};
use tower::ServiceExt; // for `oneshot`

// ─── Static HTML tests ────────────────────────────────────────────────────────

#[test]
fn ui_html_is_nonempty() {
    assert!(
        !CHAT_UI_HTML.is_empty(),
        "CHAT_UI_HTML constant must not be empty"
    );
}

#[test]
fn ui_html_contains_doctype() {
    assert!(
        CHAT_UI_HTML.contains("<!DOCTYPE html>"),
        "HTML must contain a DOCTYPE declaration"
    );
}

#[test]
fn ui_html_contains_fetch() {
    assert!(
        CHAT_UI_HTML.contains("fetch"),
        "HTML must use fetch() to call the API"
    );
}

#[test]
fn ui_router_creates_ok() {
    // Should not panic.
    let _router = create_ui_router();
}

// ─── Async HTTP endpoint tests ────────────────────────────────────────────────

#[tokio::test]
async fn ui_health_endpoint_returns_200() {
    let app = create_ui_router();
    let req = Request::builder()
        .method("GET")
        .uri("/ui/health")
        .body(Body::empty())
        .expect("failed to build request");

    let response = app
        .oneshot(req)
        .await
        .expect("handler should not return a transport error");

    assert_eq!(
        response.status(),
        StatusCode::OK,
        "/ui/health must return 200"
    );
}

#[tokio::test]
async fn ui_serve_html_returns_200() {
    let app = create_ui_router();
    let req = Request::builder()
        .method("GET")
        .uri("/ui")
        .body(Body::empty())
        .expect("failed to build request");

    let response = app
        .oneshot(req)
        .await
        .expect("handler should not return a transport error");

    assert_eq!(response.status(), StatusCode::OK, "GET /ui must return 200");

    let content_type = response
        .headers()
        .get("content-type")
        .expect("/ui should set Content-Type")
        .to_str()
        .expect("Content-Type should be valid UTF-8");

    assert!(
        content_type.starts_with("text/html"),
        "Content-Type should be text/html, got: {}",
        content_type
    );
}