bestool-alertd 6.0.0

(Internal) BES tooling: Alert daemon
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

use std::{collections::HashMap, sync::Arc};

use jiff::Timestamp;
use tokio::sync::mpsc;

use crate::{alert::InternalContext, scheduler::Scheduler};

use super::ServerState;

pub async fn create_test_state() -> Arc<ServerState> {
	let db_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set for tests");
	let pool = bestool_postgres::pool::create_pool(&db_url, "bestool-alertd-test")
		.await
		.unwrap();
	let ctx = Arc::new(InternalContext {
		pg_pool: pool,
		http_client: reqwest::Client::new(),
		canopy_client: None,
	});
	let scheduler = Arc::new(Scheduler::new(
		vec![],
		ctx.clone(),
		None,
		true, // dry_run
		None, // server_kind
	));

	let (reload_tx, _reload_rx) = mpsc::channel::<()>(10);

	Arc::new(ServerState {
		reload_tx,
		started_at: Timestamp::now(),
		pid: std::process::id(),
		event_manager: None,
		internal_context: ctx,
		email_config: None,
		dry_run: true,
		scheduler,
		watchdog_timeout: Some(std::time::Duration::from_secs(600)),
		task_endpoints: Arc::new(HashMap::new()),
	})
}