Influxdata
/
influxdb
mirror of https://github.com/influxdata/influxdb.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

test: more tests that inlcude the repro of 2546

pull/24376/head
Nga Tran 2021-09-28 17:42:23 -04:00
parent 619d95492a
commit 2837aae479
2 changed files with 179 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
query_tests/src/scenarios/delete.rs
View File

@ -18,6 +18,30 @@ use super::{DbScenario, DbSetup};
// =========================================================================================================================
// DELETE TEST SETUPS: chunk lp data, how many chunks, their types, how many delete predicates and when they happen
#[derive(Debug)]
/// Setup for delete query test with one table and one chunk moved from MUB to RUB to OS
pub struct OneDeleteSimpleExprOneChunkDeleteAll {}
#[async_trait]
impl DbSetup for OneDeleteSimpleExprOneChunkDeleteAll {
async fn make(&self) -> Vec<DbScenario> {
let partition_key = "1970-01-01T00";
let table_name = "cpu";
// chunk data
let lp_lines = vec!["cpu bar=1 10", "cpu bar=2 20"];
// delete predicate
let pred = PredicateBuilder::new()
.table(table_name)
.timestamp_range(0, 25)
.build();
// this returns 15 scenarios
all_delete_scenarios_for_one_chunk(vec![&pred], vec![], lp_lines, table_name, partition_key)
.await
}
}
#[derive(Debug)]
/// Setup for delete query test with one table and one chunk moved from MUB to RUB to OS
pub struct OneDeleteSimpleExprOneChunk {}

156
query_tests/src/sql.rs
View File

@ -844,6 +844,25 @@ async fn sql_select_all_different_tags_chunks() {
.await;
}
// ----------------------------------------------
// Delete tests
// reproduce of https://github.com/influxdata/influxdb_iox/issues/2546
// Not throw error when all of rows of a OS or RUB chunk are soft deleted
// ignore until we have a fix that I am actively working on
#[ignore]
#[tokio::test]
async fn sql_select_with_delete_from_one_expr_delete_all() {
let expected = vec!["++", "++"];
run_sql_test_case(
scenarios::delete::OneDeleteSimpleExprOneChunkDeleteAll {},
"SELECT * from cpu",
&expected,
)
.await;
}
#[tokio::test]
async fn sql_select_with_delete_from_one_expr() {
let expected = vec![
@ -862,6 +881,35 @@ async fn sql_select_with_delete_from_one_expr() {
.await;
}
#[tokio::test]
async fn sql_select_with_delete_from_one_expr_with_select_predicate() {
// Select predicate does not eliminate extra rows
let expected = vec![
"+-----+--------------------------------+",
"| bar | time |",
"+-----+--------------------------------+",
"| 2 | 1970-01-01T00:00:00.000000020Z |",
"+-----+--------------------------------+",
];
run_sql_test_case(
scenarios::delete::OneDeleteSimpleExprOneChunk {},
"SELECT * from cpu where bar = 2.0",
&expected,
)
.await;
// Select predicate eliminates rows
let expected = vec!["++", "++"];
run_sql_test_case(
scenarios::delete::OneDeleteSimpleExprOneChunk {},
"SELECT * from cpu where bar != 2.0",
&expected,
)
.await;
}
#[tokio::test]
async fn sql_select_with_delete_from_multi_exprs() {
let expected = vec![
@ -881,6 +929,51 @@ async fn sql_select_with_delete_from_multi_exprs() {
.await;
}
#[tokio::test]
async fn sql_select_with_delete_from_multi_exprs_with_select_predicate() {
// not eliminate extra row
let expected = vec![
"+-----+-----+--------------------------------+",
"| bar | foo | time |",
"+-----+-----+--------------------------------+",
"| 1 | me | 1970-01-01T00:00:00.000000040Z |",
"| 2 | you | 1970-01-01T00:00:00.000000020Z |",
"+-----+-----+--------------------------------+",
];
run_sql_test_case(
scenarios::delete::OneDeleteMultiExprsOneChunk {},
"SELECT * from cpu where bar >= 1.0",
&expected,
)
.await;
// eliminate something
let expected = vec![
"+-----+-----+--------------------------------+",
"| bar | foo | time |",
"+-----+-----+--------------------------------+",
"| 2 | you | 1970-01-01T00:00:00.000000020Z |",
"+-----+-----+--------------------------------+",
];
run_sql_test_case(
scenarios::delete::OneDeleteMultiExprsOneChunk {},
"SELECT * from cpu where foo = 'you'",
&expected,
)
.await;
// eliminate all
let expected = vec!["++", "++"];
run_sql_test_case(
scenarios::delete::OneDeleteMultiExprsOneChunk {},
"SELECT * from cpu where foo = 'you' and bar > 2.0",
&expected,
)
.await;
}
#[tokio::test]
async fn sql_select_with_two_deletes_from_multi_exprs() {
let expected = vec![
@ -891,7 +984,6 @@ async fn sql_select_with_two_deletes_from_multi_exprs() {
"+-----+-----+--------------------------------+",
];
// Data deleted when it is in MUB, and then moved to RUB and OS
run_sql_test_case(
scenarios::delete::TwoDeletesMultiExprsOneChunk {},
"SELECT * from cpu",
@ -900,6 +992,35 @@ async fn sql_select_with_two_deletes_from_multi_exprs() {
.await;
}
#[tokio::test]
async fn sql_select_with_two_deletes_from_multi_exprs_with_select_predicate() {
// no extra eliminated
let expected = vec![
"+-----+-----+--------------------------------+",
"| bar | foo | time |",
"+-----+-----+--------------------------------+",
"| 1 | me | 1970-01-01T00:00:00.000000040Z |",
"+-----+-----+--------------------------------+",
];
run_sql_test_case(
scenarios::delete::TwoDeletesMultiExprsOneChunk {},
"SELECT * from cpu where cast(time as bigint) > 30",
&expected,
)
.await;
// all eliminated
let expected = vec!["++", "++"];
run_sql_test_case(
scenarios::delete::TwoDeletesMultiExprsOneChunk {},
"SELECT * from cpu where cast(time as bigint) > 40",
&expected,
)
.await;
}
#[tokio::test]
async fn sql_select_with_three_deletes_from_three_chunks() {
let expected = vec![
@ -983,4 +1104,37 @@ async fn sql_select_with_three_deletes_from_three_chunks_with_select_predicate()
&expected,
)
.await;
//--------------------
// Multi select expressions and eliminate something
let expected = vec![
"+-----+-----+--------------------------------+",
"| bar | foo | time |",
"+-----+-----+--------------------------------+",
"| 1 | me | 1970-01-01T00:00:00.000000040Z |",
"| 1 | me | 1970-01-01T00:00:00.000000042Z |",
"| 1 | me | 1970-01-01T00:00:00.000000062Z |",
"| 4 | me | 1970-01-01T00:00:00.000000050Z |",
"| 5 | me | 1970-01-01T00:00:00.000000060Z |",
"| 7 | me | 1970-01-01T00:00:00.000000080Z |",
"+-----+-----+--------------------------------+",
];
run_sql_test_case(
scenarios::delete::ThreeDeleteThreeChunks {},
"SELECT * from cpu where foo = 'me' and (bar > 2.0 or bar = 1.0)",
&expected,
)
.await;
//--------------------
// Multi select expressions and eliminate everything
let expected = vec!["++", "++"];
run_sql_test_case(
scenarios::delete::ThreeDeleteThreeChunks {},
"SELECT * from cpu where foo = 'you' and (bar > 3.0 or bar = 1.0)",
&expected,
)
.await;
}