Adjust expected query

google · Sep 5, 2023 · 709e2cf · 709e2cf
1 parent 1f3edd4
commit 709e2cf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/exporters/postgres/postgres.go b/exporters/postgres/postgres.go
@@ -366,11 +366,12 @@ func tableExists(db *sql.DB, tableName string) (bool, error) {
         SELECT EXISTS (
             SELECT 1
             FROM   information_schema.tables
-            WHERE  table_name = $1
-        )
+            WHERE  table_name=$1
+        );
     `
 
 	var exists bool
+
 	err := db.QueryRow(query, tableName).Scan(&exists)
 	if err != nil {
 		return false, err

diff --git a/exporters/postgres/postgres_test.go b/exporters/postgres/postgres_test.go
@@ -33,10 +33,10 @@ func TestExport(t *testing.T) {
 	}
 	defer db.Close()
 
-	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name = $1 )`).WithArgs("samples").WillReturnRows(mock.NewRows([]string{"t"}))
-	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name = $1 )`).WithArgs("payloads").WillReturnRows(mock.NewRows([]string{"t"}))
-	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name = $1 )`).WithArgs("sources").WillReturnRows(mock.NewRows([]string{"t"}))
-	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name = $1 )`).WithArgs("samples_sources").WillReturnRows(mock.NewRows([]string{"t"}))
+	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name=$1 );`).WithArgs("samples").WillReturnRows(mock.NewRows([]string{"t"}).AddRow("t"))
+	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name=$1 );`).WithArgs("payloads").WillReturnRows(mock.NewRows([]string{"t"}).AddRow("t"))
+	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name=$1 );`).WithArgs("sources").WillReturnRows(mock.NewRows([]string{"t"}).AddRow("t"))
+	mock.ExpectQuery(`SELECT EXISTS ( SELECT 1 FROM information_schema.tables WHERE table_name=$1 );`).WithArgs("samples_sources").WillReturnRows(mock.NewRows([]string{"t"}).AddRow("t"))
 
 	postgresExporter, err := NewExporter(db, false)
 	if err != nil {