includes columns in tests

PeerDB-io · Dec 18, 2023 · 3a5e2ec · 3a5e2ec
1 parent 6252da8
commit 3a5e2ec
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git 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
@@ -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"},
 		}