test: add skip segment overrides table check
This commit is contained in:
@@ -3,6 +3,7 @@ package db
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"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;`
|
const query = `SELECT name FROM sqlite_master WHERE type='table' AND name='skip_segment_override' LIMIT 1;`
|
||||||
var name sql.NullString
|
var name sql.NullString
|
||||||
if err := q.db.QueryRowContext(ctx, query).Scan(&name); err != nil {
|
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 false, fmt.Errorf("check skip segment override table: %w", err)
|
||||||
}
|
}
|
||||||
return name.Valid && name.String != "", nil
|
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