worker-service 0.2.0

Worker Service - A worker administration microservice that interoperates with the worker-matcher crate
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

//! Common test utilities for integration tests

use worker_service::{
    config::Config,
    db::create_pool,
    search::SearchEngine,
    matching::ProbabilisticMatcher,
    api::rest::{AppState, create_router},
};
use axum::Router;

/// Create a test application state for integration tests
pub fn create_test_app_state() -> AppState {
    // Load test configuration
    let config = Config::from_env().expect("Failed to load test config");

    // Create database pool
    let db_pool = create_pool(&config.database)
        .expect("Failed to create database pool");

    // Create search engine
    let search_engine = SearchEngine::new(&config.search.index_path)
        .expect("Failed to create search engine");

    // Create matcher
    let matcher = ProbabilisticMatcher::new(config.matching.clone());

    // Create application state
    AppState::new(db_pool, search_engine, matcher, config)
}

/// Create a test router with test application state
pub fn create_test_router() -> Router {
    let state = create_test_app_state();
    create_router(state)
}

/// Create a unique test worker name to avoid conflicts
pub fn unique_worker_name(suffix: &str) -> String {
    use chrono::Utc;
    let timestamp = Utc::now().timestamp_micros();
    format!("TestWorker{}_{}", suffix, timestamp)
}