1use crate::{
 2    db::{Database, NewUserParams},
 3    test_both_dbs,
 4};
 5use pretty_assertions::assert_eq;
 6use std::sync::Arc;
 7
 8test_both_dbs!(
 9    test_get_user_flags,
10    test_get_user_flags_postgres,
11    test_get_user_flags_sqlite
12);
13
14async fn test_get_user_flags(db: &Arc<Database>) {
15    let user_1 = db
16        .create_user(
17            "user1@example.com",
18            false,
19            NewUserParams {
20                github_login: "user1".to_string(),
21                github_user_id: 1,
22            },
23        )
24        .await
25        .unwrap()
26        .user_id;
27
28    let user_2 = db
29        .create_user(
30            "user2@example.com",
31            false,
32            NewUserParams {
33                github_login: "user2".to_string(),
34                github_user_id: 2,
35            },
36        )
37        .await
38        .unwrap()
39        .user_id;
40
41    const FEATURE_FLAG_ONE: &str = "brand-new-ux";
42    const FEATURE_FLAG_TWO: &str = "cool-feature";
43    const FEATURE_FLAG_THREE: &str = "feature-enabled-for-everyone";
44
45    let feature_flag_one = db.create_user_flag(FEATURE_FLAG_ONE, false).await.unwrap();
46    let feature_flag_two = db.create_user_flag(FEATURE_FLAG_TWO, false).await.unwrap();
47    db.create_user_flag(FEATURE_FLAG_THREE, true).await.unwrap();
48
49    db.add_user_flag(user_1, feature_flag_one).await.unwrap();
50    db.add_user_flag(user_1, feature_flag_two).await.unwrap();
51
52    db.add_user_flag(user_2, feature_flag_one).await.unwrap();
53
54    let mut user_1_flags = db.get_user_flags(user_1).await.unwrap();
55    user_1_flags.sort();
56    assert_eq!(
57        user_1_flags,
58        &[FEATURE_FLAG_ONE, FEATURE_FLAG_TWO, FEATURE_FLAG_THREE]
59    );
60
61    let mut user_2_flags = db.get_user_flags(user_2).await.unwrap();
62    user_2_flags.sort();
63    assert_eq!(user_2_flags, &[FEATURE_FLAG_ONE, FEATURE_FLAG_THREE]);
64}