From dbe2ad64bbff2d1c4b0452e07043a668546e33b2 Mon Sep 17 00:00:00 2001 From: Kevin Biju Date: Wed, 20 Dec 2023 13:03:03 +0530 Subject: [PATCH] trying to fix pt.3 --- flow/e2e/bigquery/bigquery_helper.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flow/e2e/bigquery/bigquery_helper.go b/flow/e2e/bigquery/bigquery_helper.go index 591eba4f64..934435227b 100644 --- a/flow/e2e/bigquery/bigquery_helper.go +++ b/flow/e2e/bigquery/bigquery_helper.go @@ -453,7 +453,7 @@ func (b *BigQueryTestHelper) CreateTable(tableName string, schema *model.QRecord return nil } -func (b *BigQueryTestHelper) RunIntQuery(query string) (int64, error) { +func (b *BigQueryTestHelper) RunIntQuery(query string) (int, error) { recordBatch, err := b.ExecuteAndProcessQuery(query) if err != nil { return 0, fmt.Errorf("could not execute query: %w", err) @@ -462,5 +462,5 @@ func (b *BigQueryTestHelper) RunIntQuery(query string) (int64, error) { return 0, fmt.Errorf("expected only 1 record, got %d", recordBatch.NumRecords) } - return recordBatch.Records[0].Entries[0].Value.(int64), nil + return int(recordBatch.Records[0].Entries[0].Value.(int64)), nil }