test: add skip segment overrides table check
This commit is contained in:
@@ -3,6 +3,7 @@ package db
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
@@ -67,6 +68,9 @@ func (q *Queries) HasSkipSegmentOverrideTable(ctx context.Context) (bool, error)
|
||||
const query = `SELECT name FROM sqlite_master WHERE type='table' AND name='skip_segment_override' LIMIT 1;`
|
||||
var name sql.NullString
|
||||
if err := q.db.QueryRowContext(ctx, query).Scan(&name); err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return false, nil
|
||||
}
|
||||
return false, fmt.Errorf("check skip segment override table: %w", err)
|
||||
}
|
||||
return name.Valid && name.String != "", nil
|
||||
|
||||
25
internal/db/skip_segment_overrides_test.go
Normal file
25
internal/db/skip_segment_overrides_test.go
Normal file
@@ -0,0 +1,25 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"testing"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
func TestHasSkipSegmentOverrideTableReturnsFalseWhenMissing(t *testing.T) {
|
||||
sqlDB, err := sql.Open("sqlite3", ":memory:")
|
||||
if err != nil {
|
||||
t.Fatalf("open sqlite: %v", err)
|
||||
}
|
||||
defer func() { _ = sqlDB.Close() }()
|
||||
|
||||
ok, err := New(sqlDB).HasSkipSegmentOverrideTable(context.Background())
|
||||
if err != nil {
|
||||
t.Fatalf("HasSkipSegmentOverrideTable: %v", err)
|
||||
}
|
||||
if ok {
|
||||
t.Fatalf("HasSkipSegmentOverrideTable returned true for missing table")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user