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)
+}