Skip to content

Commit 2cb856c

Browse files
feat: added schema evolution test cases
1 parent b335e57 commit 2cb856c

File tree

2 files changed

+11
-10
lines changed

2 files changed

+11
-10
lines changed

drivers/mongodb/internal/mon_test.go

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,15 @@ import (
1010
func TestMongodbIntegration(t *testing.T) {
1111
t.Parallel()
1212
testConfig := &testutils.IntegrationTest{
13-
TestConfig: testutils.GetTestConfig(string(constants.MongoDB)),
14-
Namespace: "olake_mongodb_test",
15-
ExpectedData: ExpectedMongoData,
16-
ExpectedIcebergUpdateData: ExpectedIcebergUpdatedData,
17-
ExpectedParquetUpdateData: ExpectedParquetUpdatedData,
18-
DestinationDataTypeSchema: MongoToDestinationSchema, // Use same schema for parquet
19-
ExecuteQuery: ExecuteQuery,
20-
DestinationDB: "mongodb_olake_mongodb_test",
13+
TestConfig: testutils.GetTestConfig(string(constants.MongoDB)),
14+
Namespace: "olake_mongodb_test",
15+
ExpectedData: ExpectedMongoData,
16+
ExpectedIcebergUpdateData: ExpectedIcebergUpdatedData,
17+
ExpectedParquetUpdateData: ExpectedParquetUpdatedData,
18+
DestinationDataTypeSchema: MongoToDestinationSchema,
19+
UpdatedDestinationDataTypeSchema: MongoToDestinationSchema,
20+
ExecuteQuery: ExecuteQuery,
21+
DestinationDB: "mongodb_olake_mongodb_test",
2122
}
2223
testConfig.TestIntegration(t)
2324
}

drivers/mongodb/internal/mon_test_util.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ func ExecuteQuery(ctx context.Context, t *testing.T, streams []string, operation
114114
update := bson.M{
115115
"$set": bson.M{
116116
"id_bigint": int64(987654321098765),
117-
"id_int": int32(200),
117+
"id_int": int64(200),
118118
"id_timestamp": time.Date(2024, 7, 1, 15, 30, 0, 0, time.UTC),
119119
"id_double": float64(202.456),
120120
"id_bool": false,
@@ -250,7 +250,7 @@ var ExpectedMongoData = map[string]interface{}{
250250

251251
var ExpectedIcebergUpdatedData = map[string]interface{}{
252252
"id_bigint": int64(987654321098765),
253-
"id_int": int32(200),
253+
"id_int": int64(200),
254254
"id_timestamp": arrow.Timestamp(time.Date(2024, 7, 1, 15, 30, 0, 0, time.UTC).UnixNano() / int64(time.Microsecond)),
255255
"id_double": float64(202.456),
256256
"id_bool": false,

0 commit comments

Comments
 (0)