AtCoderProblems/atcoder-problems-backend/sql-client/tests/test_accepted_count.rs at a730898f047ed09cbf3ed54ab425eb9da5a8dcd6 · hotate29/AtCoderProblems · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
use sql_client::accepted_count::AcceptedCountClient;
use sql_client::models::{Submission, UserProblemCount};
use sql_client::PgPool;

mod utils;

#[sqlx::test]
async fn test_accepted_count(pool: PgPool) {
    utils::initialize(&pool).await;

    let submissions = [
        Submission {
            id: 1,
            user_id: "user1".to_owned(),
            problem_id: "problem1".to_owned(),
            ..Default::default()
        },
        Submission {
            id: 2,
            user_id: "user1".to_owned(),
            problem_id: "problem1".to_owned(),
            ..Default::default()
        },
        Submission {
            id: 3,
            user_id: "user1".to_owned(),
            problem_id: "problem2".to_owned(),
            ..Default::default()
        },
        Submission {
            id: 4,
            user_id: "user2".to_owned(),
            problem_id: "problem1".to_owned(),
            ..Default::default()
        },
        Submission {
            id: 5,
            user_id: "user2".to_owned(),
            problem_id: "problem2".to_owned(),
            ..Default::default()
        },
        Submission {
            id: 6,
            user_id: "user2".to_owned(),
            problem_id: "problem3".to_owned(),
            ..Default::default()
        },
    ];
    pool.update_accepted_count(&submissions).await.unwrap();

    let accepted_count = pool.load_accepted_count().await.unwrap();
    assert_eq!(
        accepted_count,
        vec![
            UserProblemCount {
                user_id: "user2".to_owned(),
                problem_count: 3
            },
            UserProblemCount {
                user_id: "user1".to_owned(),
                problem_count: 2
            }
        ]
    );

    let accepted_count_1st_to_2nd = pool.load_accepted_count_in_range(0..2).await.unwrap();
    assert_eq!(
        accepted_count_1st_to_2nd,
        vec![
            UserProblemCount {
                user_id: "user2".to_owned(),
                problem_count: 3
            },
            UserProblemCount {
                user_id: "user1".to_owned(),
                problem_count: 2
            }
        ]
    );

    let accepted_count_2nd_to_2nd = pool.load_accepted_count_in_range(1..2).await.unwrap();
    assert_eq!(
        accepted_count_2nd_to_2nd,
        vec![UserProblemCount {
            user_id: "user1".to_owned(),
            problem_count: 2
        }]
    );

    assert_eq!(
        pool.load_accepted_count_in_range(0..10)
            .await
            .unwrap()
            .len(),
        2
    );

    assert_eq!(pool.get_users_accepted_count("user1").await.unwrap(), 2);
    assert_eq!(pool.get_users_accepted_count("user2").await.unwrap(), 3);
    assert_eq!(pool.get_users_accepted_count("USER1").await.unwrap(), 2);
    assert_eq!(pool.get_users_accepted_count("USER2").await.unwrap(), 3);
    assert_eq!(pool.get_accepted_count_rank(3).await.unwrap(), 0);
    assert_eq!(pool.get_accepted_count_rank(2).await.unwrap(), 1);

    assert!(pool
        .get_users_accepted_count("non_existing_user")
        .await
        .is_none());
}