Fix: table exists query

baking-bad · Oct 18, 2023 · e77e0a9 · e77e0a9
1 parent f8502c4
commit e77e0a9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/internal/postgres/core/table.go b/internal/postgres/core/table.go
@@ -120,7 +120,7 @@ type existsResponse struct {
 func (p *Postgres) TablesExist(ctx context.Context) bool {
 	for _, table := range models.AllDocuments() {
 		var exists existsResponse
-		err := p.DB.QueryRow(tableExistsQuery, p.schema, table).Scan(&exists)
+		err := p.DB.NewRaw(tableExistsQuery, p.schema, table).Scan(ctx, &exists)
 		if !exists.Flag || err != nil {
 			return false
 		}

diff --git a/internal/postgres/tests/general_test.go b/internal/postgres/tests/general_test.go
@@ -0,0 +1,14 @@
+package tests
+
+import (
+	"context"
+	"time"
+)
+
+func (s *StorageTestSuite) TestTableExists() {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+
+	exists := s.storage.TablesExist(ctx)
+	s.Require().True(exists)
+}