1use chrono::Utc;
 2
 3use crate::{
 4    db::{Database, NewUserParams},
 5    test_both_dbs,
 6};
 7use std::sync::Arc;
 8
 9test_both_dbs!(
10    test_accepted_tos,
11    test_accepted_tos_postgres,
12    test_accepted_tos_sqlite
13);
14
15async fn test_accepted_tos(db: &Arc<Database>) {
16    let user_id = db
17        .create_user(
18            "user1@example.com",
19            None,
20            false,
21            NewUserParams {
22                github_login: "user1".to_string(),
23                github_user_id: 1,
24            },
25        )
26        .await
27        .unwrap()
28        .user_id;
29
30    let user = db.get_user_by_id(user_id).await.unwrap().unwrap();
31    assert!(user.accepted_tos_at.is_none());
32
33    let accepted_tos_at = Utc::now().naive_utc();
34    db.set_user_accepted_tos_at(user_id, Some(accepted_tos_at))
35        .await
36        .unwrap();
37
38    let user = db.get_user_by_id(user_id).await.unwrap().unwrap();
39    assert!(user.accepted_tos_at.is_some());
40    assert_eq!(user.accepted_tos_at, Some(accepted_tos_at));
41
42    db.set_user_accepted_tos_at(user_id, None).await.unwrap();
43
44    let user = db.get_user_by_id(user_id).await.unwrap().unwrap();
45    assert!(user.accepted_tos_at.is_none());
46}