Skip to content

Commit

Permalink
refactor: test
Browse files Browse the repository at this point in the history
  • Loading branch information
@shuiyisong
shuiyisong committed Jul 15, 2024
1 parent 7674c4b commit ea89866
Show file tree
Hide file tree
Showing 3 changed files with 62 additions and 86 deletions.
14 changes: 6 additions & 8 deletions src/pipeline/tests/gsub.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

use greptime_proto::v1::value::ValueData::TimestampMillisecondValue;
use greptime_proto::v1::{ColumnDataType, ColumnSchema, SemanticType};
use greptime_proto::v1::{ColumnDataType, SemanticType};

mod common;

Expand Down Expand Up @@ -49,13 +49,11 @@ transform:

let output = common::parse_and_exec(input_value_str, pipeline_yaml);

let expected_schema = vec![ColumnSchema {
column_name: "reqTimeSec".to_string(),
datatype: ColumnDataType::TimestampMillisecond.into(),
semantic_type: SemanticType::Timestamp.into(),
datatype_extension: None,
options: None,
}];
let expected_schema = vec![common::make_column_schema(
"reqTimeSec".to_string(),
ColumnDataType::TimestampMillisecond,
SemanticType::Timestamp,
)];

assert_eq!(output.schema, expected_schema);
assert_eq!(
Expand Down
24 changes: 10 additions & 14 deletions src/pipeline/tests/join.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -32,20 +32,16 @@ transform:

lazy_static! {
pub static ref EXPECTED_SCHEMA: Vec<ColumnSchema> = vec![
ColumnSchema {
column_name: "join_test".to_string(),
datatype: ColumnDataType::String.into(),
semantic_type: SemanticType::Field.into(),
datatype_extension: None,
options: None,
},
ColumnSchema {
column_name: "greptime_timestamp".to_string(),
datatype: ColumnDataType::TimestampNanosecond.into(),
semantic_type: SemanticType::Timestamp.into(),
datatype_extension: None,
options: None,
},
common::make_column_schema(
"join_test".to_string(),
ColumnDataType::String,
SemanticType::Field,
),
common::make_column_schema(
"greptime_timestamp".to_string(),
ColumnDataType::TimestampNanosecond,
SemanticType::Timestamp,
),
];
}

Expand Down
110 changes: 46 additions & 64 deletions src/pipeline/tests/on_failure.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

use greptime_proto::v1::value::ValueData::{U16Value, U8Value};
use greptime_proto::v1::{ColumnDataType, ColumnSchema, SemanticType};
use greptime_proto::v1::{ColumnDataType, SemanticType};

mod common;

Expand All @@ -40,20 +40,16 @@ transform:
let output = common::parse_and_exec(input_value_str, pipeline_yaml);

let expected_schema = vec![
ColumnSchema {
column_name: "version".to_string(),
datatype: ColumnDataType::Uint8.into(),
semantic_type: SemanticType::Field.into(),
datatype_extension: None,
options: None,
},
ColumnSchema {
column_name: "greptime_timestamp".to_string(),
datatype: ColumnDataType::TimestampNanosecond.into(),
semantic_type: SemanticType::Timestamp.into(),
datatype_extension: None,
options: None,
},
common::make_column_schema(
"version".to_string(),
ColumnDataType::Uint8,
SemanticType::Field,
),
common::make_column_schema(
"greptime_timestamp".to_string(),
ColumnDataType::TimestampNanosecond,
SemanticType::Timestamp,
),
];

assert_eq!(output.schema, expected_schema);
Expand Down Expand Up @@ -85,20 +81,16 @@ transform:
let output = common::parse_and_exec(input_value_str, pipeline_yaml);

let expected_schema = vec![
ColumnSchema {
column_name: "version".to_string(),
datatype: ColumnDataType::Uint8.into(),
semantic_type: SemanticType::Field.into(),
datatype_extension: None,
options: None,
},
ColumnSchema {
column_name: "greptime_timestamp".to_string(),
datatype: ColumnDataType::TimestampNanosecond.into(),
semantic_type: SemanticType::Timestamp.into(),
datatype_extension: None,
options: None,
},
common::make_column_schema(
"version".to_string(),
ColumnDataType::Uint8,
SemanticType::Field,
),
common::make_column_schema(
"greptime_timestamp".to_string(),
ColumnDataType::TimestampNanosecond,
SemanticType::Timestamp,
),
];

assert_eq!(output.schema, expected_schema);
Expand All @@ -125,20 +117,16 @@ transform:
let output = common::parse_and_exec(input_value_str, pipeline_yaml);

let expected_schema = vec![
ColumnSchema {
column_name: "version".to_string(),
datatype: ColumnDataType::Uint8.into(),
semantic_type: SemanticType::Field.into(),
datatype_extension: None,
options: None,
},
ColumnSchema {
column_name: "greptime_timestamp".to_string(),
datatype: ColumnDataType::TimestampNanosecond.into(),
semantic_type: SemanticType::Timestamp.into(),
datatype_extension: None,
options: None,
},
common::make_column_schema(
"version".to_string(),
ColumnDataType::Uint8,
SemanticType::Field,
),
common::make_column_schema(
"greptime_timestamp".to_string(),
ColumnDataType::TimestampNanosecond,
SemanticType::Timestamp,
),
];

assert_eq!(output.schema, expected_schema);
Expand Down Expand Up @@ -176,27 +164,21 @@ transform:
let output = common::parse_and_exec(input_value_str, pipeline_yaml);

let expected_schema = vec![
ColumnSchema {
column_name: "version".to_string(),
datatype: ColumnDataType::Uint8.into(),
semantic_type: SemanticType::Field.into(),
datatype_extension: None,
options: None,
},
ColumnSchema {
column_name: "spec_version".to_string(),
datatype: ColumnDataType::Uint16.into(),
semantic_type: SemanticType::Field.into(),
datatype_extension: None,
options: None,
},
ColumnSchema {
column_name: "greptime_timestamp".to_string(),
datatype: ColumnDataType::TimestampNanosecond.into(),
semantic_type: SemanticType::Timestamp.into(),
datatype_extension: None,
options: None,
},
common::make_column_schema(
"version".to_string(),
ColumnDataType::Uint8,
SemanticType::Field,
),
common::make_column_schema(
"spec_version".to_string(),
ColumnDataType::Uint16,
SemanticType::Field,
),
common::make_column_schema(
"greptime_timestamp".to_string(),
ColumnDataType::TimestampNanosecond,
SemanticType::Timestamp,
),
];

assert_eq!(output.schema, expected_schema);
Expand Down

0 comments on commit ea89866

Please sign in to comment.