From 3a5e2ecf0c5e7b5d0e30ee968cc37d4ea96b0a69 Mon Sep 17 00:00:00 2001 From: Amogh-Bharadwaj Date: Mon, 18 Dec 2023 22:10:18 +0530 Subject: [PATCH] includes columns in tests --- flow/e2e/bigquery/peer_flow_bq_test.go | 2 +- flow/e2e/postgres/peer_flow_pg_test.go | 10 ++++++---- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/flow/e2e/bigquery/peer_flow_bq_test.go b/flow/e2e/bigquery/peer_flow_bq_test.go index de3ddae7e5..a67ca58fe6 100644 --- a/flow/e2e/bigquery/peer_flow_bq_test.go +++ b/flow/e2e/bigquery/peer_flow_bq_test.go @@ -419,7 +419,7 @@ func (s PeerFlowE2ETestSuiteBQ) Test_Toast_Nochanges_BQ() { // allow only continue as new error require.Contains(s.t, err.Error(), "continue as new") - s.compareTableContentsBQ(dstTableName, "id,t1,t2,k") + s.compareTableContentsBQ(dstTableName, "id,t1,t2,k,_PEERDB_IS_DELETED, _PEERDB_SYNCED_AT") env.AssertExpectations(s.t) <-done } diff --git a/flow/e2e/postgres/peer_flow_pg_test.go b/flow/e2e/postgres/peer_flow_pg_test.go index 2720891fb6..d3996c843c 100644 --- a/flow/e2e/postgres/peer_flow_pg_test.go +++ b/flow/e2e/postgres/peer_flow_pg_test.go @@ -213,10 +213,12 @@ func (s *PeerFlowE2ETestSuitePG) Test_Simple_Schema_Changes_PG() { expectedTableSchema = &protos.TableSchema{ TableIdentifier: dstTableName, Columns: map[string]string{ - "id": string(qvalue.QValueKindInt64), - "c1": string(qvalue.QValueKindInt64), - "c2": string(qvalue.QValueKindInt64), - "c3": string(qvalue.QValueKindInt64), + "id": string(qvalue.QValueKindInt64), + "c1": string(qvalue.QValueKindInt64), + "c2": string(qvalue.QValueKindInt64), + "c3": string(qvalue.QValueKindInt64), + "_PEERDB_IS_DELETED": string(qvalue.QValueKindBoolean), + "_PEERDB_SYNCED_AT": string(qvalue.QValueKindTimestamp), }, PrimaryKeyColumns: []string{"id"}, }