test: fix some integration tests (#2432)

* test: fix some integration tests

Signed-off-by: Zhenchi <zhongzc_arch@outlook.com>

* test: add timezone setting

Signed-off-by: Zhenchi <zhongzc_arch@outlook.com>

---------

Signed-off-by: Zhenchi <zhongzc_arch@outlook.com>
This commit is contained in:
Zhenchi
2023-09-18 18:52:14 +08:00
committed by GitHub
parent fbbf3978d9
commit d6d46378a1
3 changed files with 38 additions and 9 deletions

View File

@@ -347,7 +347,7 @@ CREATE TABLE {table_name} (
.collect(),
..Default::default()
}),
semantic_type: SemanticType::Field as i32,
semantic_type: SemanticType::Tag as i32,
datatype: ColumnDataType::String as i32,
..Default::default()
},
@@ -421,7 +421,7 @@ CREATE TABLE {table_name} (
},
Column {
column_name: "b".to_string(),
semantic_type: SemanticType::Field as i32,
semantic_type: SemanticType::Tag as i32,
values: Some(Values {
string_values: b,
..Default::default()

View File

@@ -23,6 +23,7 @@ use common_test_util::temp_dir;
use datatypes::vectors::{StringVector, TimestampMillisecondVector, UInt64Vector, VectorRef};
use frontend::error::{Error, Result};
use frontend::instance::Instance;
use operator::error::Error as OperatorError;
use rstest::rstest;
use rstest_reuse::apply;
use servers::query_handler::sql::SqlQueryHandler;
@@ -375,20 +376,28 @@ async fn test_execute_insert_by_select(instance: Arc<dyn MockInstance>) {
try_execute_sql(&instance, "insert into demo2(host) select * from demo1")
.await
.unwrap_err(),
Error::PlanStatement { .. }
Error::TableOperation {
source: OperatorError::PlanStatement { .. },
..
}
));
assert!(matches!(
try_execute_sql(&instance, "insert into demo2 select cpu,memory from demo1")
.await
.unwrap_err(),
Error::PlanStatement { .. }
Error::TableOperation {
source: OperatorError::PlanStatement { .. },
..
}
));
assert!(matches!(
try_execute_sql(&instance, "insert into demo2(ts) select memory from demo1")
.await
.unwrap_err(),
Error::PlanStatement { .. }
Error::TableOperation {
source: OperatorError::PlanStatement { .. },
..
}
));
let output = execute_sql(&instance, "insert into demo2 select * from demo1").await;
@@ -705,6 +714,8 @@ async fn test_execute_query_external_table_parquet(instance: Arc<dyn MockInstanc
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_orc(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "orc";
let location = find_testing_resource("/src/common/datasource/tests/orc/test.orc");
@@ -781,6 +792,8 @@ async fn test_execute_query_external_table_orc(instance: Arc<dyn MockInstance>)
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_orc_with_schema(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "orc";
let location = find_testing_resource("/src/common/datasource/tests/orc/test.orc");
@@ -830,6 +843,8 @@ async fn test_execute_query_external_table_orc_with_schema(instance: Arc<dyn Moc
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_csv(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "csv";
let location = find_testing_resource("/tests/data/csv/various_type.csv");
@@ -877,6 +892,8 @@ async fn test_execute_query_external_table_csv(instance: Arc<dyn MockInstance>)
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_json(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "json";
let location = find_testing_resource("/tests/data/json/various_type.json");
@@ -931,6 +948,8 @@ async fn test_execute_query_external_table_json(instance: Arc<dyn MockInstance>)
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_json_with_schema(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "json";
let location = find_testing_resource("/tests/data/json/various_type.json");
@@ -993,6 +1012,8 @@ async fn test_execute_query_external_table_json_with_schema(instance: Arc<dyn Mo
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_json_type_cast(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "json";
let location = find_testing_resource("/tests/data/json/type_cast.json");
@@ -1059,6 +1080,8 @@ async fn test_execute_query_external_table_json_type_cast(instance: Arc<dyn Mock
#[apply(both_instances_cases)]
async fn test_execute_query_external_table_json_default_ts_column(instance: Arc<dyn MockInstance>) {
std::env::set_var("TZ", "UTC");
let instance = instance.frontend();
let format = "json";
let location = find_testing_resource("/tests/data/json/default_ts_column.json");

View File

@@ -605,15 +605,17 @@ pub async fn test_config_api(store_type: StorageType) {
assert_eq!(res_get.status(), StatusCode::OK);
let expected_toml_str = format!(
r#"mode = "standalone"
node_id = 0
coordination = false
rpc_addr = "127.0.0.1:3001"
rpc_runtime_size = 8
enable_telemetry = true
[heartbeat]
interval_millis = 5000
retry_interval_millis = 5000
interval_millis = 3000
retry_interval_millis = 3000
[http_opts]
[http]
addr = "127.0.0.1:4000"
timeout = "30s"
body_limit = "64MiB"
@@ -658,6 +660,10 @@ auto_flush_interval = "30m"
global_write_buffer_size = "1GiB"
global_write_buffer_reject_size = "2GiB"
[[region_engine]]
[region_engine.file]
[logging]
enable_jaeger_tracing = false"#,
store_type