Skip to content

Commit

Permalink
Use more execute_batch() in stats tests
Browse files Browse the repository at this point in the history 
Change-Id: I4010d96c866ebbf3f521542b66c3ed291029ebe3
  • Loading branch information
@vmiklos
vmiklos committed Sep 1, 2024
1 parent 64dd2b9 commit 34a4b07
Showing 1 changed file with 40 additions and 91 deletions.
131 changes: 40 additions & 91 deletions src/stats/tests.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,8 @@ fn test_handle_progress() {
let mut ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-10", "254651"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2020-05-10', '254651');",
)
.unwrap();
}
Expand Down Expand Up @@ -60,14 +59,9 @@ fn test_handle_capital_progress() {
let mut ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_citycounts (date, city, count) values (?1, ?2, ?3)"#,
["2020-05-10", "Budapest_02", "200"],
)
.unwrap();
conn.execute(
r#"insert into stats_citycounts (date, city, count) values (?1, ?2, ?3)"#,
["2020-05-10", "Budapest_11", "11"],
conn.execute_batch(
"insert into stats_citycounts (date, city, count) values ('2020-05-10', 'Budapest_02', '200');
insert into stats_citycounts (date, city, count) values ('2020-05-10', 'Budapest_11', '11');",
)
.unwrap();
}
Expand Down Expand Up @@ -131,15 +125,15 @@ fn test_handle_topusers() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
let mut stmt = conn
.prepare("insert into stats_topusers (date, user, count) values (?1, ?2, ?3)")
.unwrap();
for i in 0..20 {
conn.execute(
r#"insert into stats_topusers (date, user, count) values (?1, ?2, ?3)"#,
[
"2020-05-10",
&format!("user{}", i + 1),
&format!("{}", 20 - i),
],
)
stmt.execute([
"2020-05-10",
&format!("user{}", i + 1),
&format!("{}", 20 - i),
])
.unwrap();
}
}
Expand Down Expand Up @@ -192,14 +186,9 @@ fn test_hanle_daily_new() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-04-26", "251250"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-04-27", "251614"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2020-04-26', '251250');
insert into stats_counts (date, count) values ('2020-04-27', '251614');",
)
.unwrap();
}
Expand All @@ -218,14 +207,9 @@ fn test_handle_invalid_addr_cities() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_invalid_addr_cities_counts (date, count) values (?1, ?2)"#,
["2020-04-26", "1"],
)
.unwrap();
conn.execute(
r#"insert into stats_invalid_addr_cities_counts (date, count) values (?1, ?2)"#,
["2020-04-27", "2"],
conn.execute_batch(
"insert into stats_invalid_addr_cities_counts (date, count) values ('2020-04-26', '1');
insert into stats_invalid_addr_cities_counts (date, count) values ('2020-04-27', '2');",
)
.unwrap();
}
Expand Down Expand Up @@ -256,24 +240,11 @@ fn test_handle_monthly_new() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2019-05-01", "199518"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2019-06-01", "203317"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-01", "253027"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-10", "254651"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2019-05-01', '199518');
insert into stats_counts (date, count) values ('2019-06-01', '203317');
insert into stats_counts (date, count) values ('2020-05-01', '253027');
insert into stats_counts (date, count) values ('2020-05-10', '254651');",
)
.unwrap();
}
Expand All @@ -293,14 +264,9 @@ fn test_handle_monthly_new_empty_month_range() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-01", "253027"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-10", "254651"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2020-05-01', '253027');
insert into stats_counts (date, count) values ('2020-05-10', '254651');",
)
.unwrap();
}
Expand All @@ -316,19 +282,10 @@ fn test_handle_monthly_new_incomplete_last_month() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2019-05-01", "199518"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2019-06-01", "203317"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-01", "253027"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2019-05-01', '199518');
insert into stats_counts (date, count) values ('2019-06-01', '203317');
insert into stats_counts (date, count) values ('2020-05-01', '253027');",
)
.unwrap();
// 2020-05-10 would be the data for the current state of the last, incomplete month.
Expand All @@ -349,9 +306,8 @@ fn test_handle_daily_total() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-04-27", "251614"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2020-04-27', '251614');",
)
.unwrap();
}
Expand All @@ -378,9 +334,8 @@ fn test_handle_user_total() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_usercounts (date, count) values (?1, ?2)"#,
["2020-04-27", "43"],
conn.execute_batch(
"insert into stats_usercounts (date, count) values ('2020-04-27', '43');",
)
.unwrap();
}
Expand All @@ -407,9 +362,8 @@ fn test_handle_monthly_total() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2019-06-01", "203317"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2019-06-01', '203317');",
)
.unwrap();
}
Expand All @@ -436,14 +390,9 @@ fn test_handle_monthly_total_one_element_day_range() {
let ctx = context::tests::make_test_context().unwrap();
{
let conn = ctx.get_database_connection().unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-01", "253027"],
)
.unwrap();
conn.execute(
r#"insert into stats_counts (date, count) values (?1, ?2)"#,
["2020-05-10", "254651"],
conn.execute_batch(
"insert into stats_counts (date, count) values ('2020-05-01', '253027');
insert into stats_counts (date, count) values ('2020-05-10', '254651');",
)
.unwrap();
}
Expand Down

0 comments on commit 34a4b07

Please sign in to comment.