Function make_app

src/app/tests.rs:24–83

(upstream_url: &str)

use crate::stats::Stats;

fn make_app(upstream_url: &str) -> axum::Router {
    let mut key_map = BTreeMap::new();
    key_map.insert(
        "vk-test-1".to_string(),
        ResolvedKey {
            source: KeySource::Static {
                real_key: "sk-real-1".to_string(),
            },
            provider: Provider::Openai,
        },
    );
    key_map.insert(
        "vk-test-2".to_string(),
        ResolvedKey {
            source: KeySource::Static {
                real_key: "sk-real-2".to_string(),
            },
            provider: Provider::Openai,
        },
    );

    let patterns = vec![
        DlpPattern {
            name: "ssn".to_string(),
            regex: r"\b\d{3}-\d{2}-\d{4}\b".to_string(),
            action: DlpAction::Block,
        },
        DlpPattern {
            name: "email".to_string(),
            regex: r"\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,}\b".to_string(),
            action: DlpAction::Block,
        },
        DlpPattern {
            name: "credit_card".to_string(),
            regex: r"\b(?:\d[ -]*?){13,19}\b".to_string(),
            action: DlpAction::Block,
        },
    ];

    let mut upstream_urls = BTreeMap::new();
    upstream_urls.insert(Provider::Openai, upstream_url.to_string());
    upstream_urls.insert(Provider::Anthropic, upstream_url.to_string());

    let state = AppState {
        key_manager: Arc::new(KeyManager::new(key_map)),
        dlp_scanner: Arc::new(DlpScanner::new(&patterns, false).unwrap()),
        proxy_client: Arc::new(ProxyClient::with_upstream_urls(
            upstream_urls,
            "2023-06-01".to_string(),
        )),
        oauth_registry: Arc::new(OAuthRegistry::new(Default::default())),
        email_enabled: false,
        email_policy: None,
        email_accounts: Arc::new(BTreeMap::new()),
        email_service: Arc::new(EmailService::mock_disabled()),
        stats: Arc::new(Stats::new(None)),
    };