SQL/Storage: Remove SkipDataMigration flag (#99404)

This commit is contained in:
Ryan McKinley
2025-01-23 12:34:48 +02:00
committed by GitHub
parent fc7db91bf1
commit c4c934e0bd
2 changed files with 16 additions and 22 deletions
+12 -15
View File
@@ -34,10 +34,9 @@ type Backend interface {
}
type BackendOptions struct {
DBProvider db.DBProvider
Tracer trace.Tracer
PollingInterval time.Duration
SkipDataMigration bool
DBProvider db.DBProvider
Tracer trace.Tracer
PollingInterval time.Duration
}
func NewBackend(opts BackendOptions) (Backend, error) {
@@ -54,13 +53,12 @@ func NewBackend(opts BackendOptions) (Backend, error) {
pollingInterval = defaultPollingInterval
}
return &backend{
done: ctx.Done(),
cancel: cancel,
log: log.New("sql-resource-server"),
tracer: opts.Tracer,
dbProvider: opts.DBProvider,
pollingInterval: pollingInterval,
skipDataMigration: opts.SkipDataMigration,
done: ctx.Done(),
cancel: cancel,
log: log.New("sql-resource-server"),
tracer: opts.Tracer,
dbProvider: opts.DBProvider,
pollingInterval: pollingInterval,
}, nil
}
@@ -76,10 +74,9 @@ type backend struct {
tracer trace.Tracer
// database
dbProvider db.DBProvider
db db.DB
dialect sqltemplate.Dialect
skipDataMigration bool
dbProvider db.DBProvider
db db.DB
dialect sqltemplate.Dialect
// watch streaming
//stream chan *resource.WatchEvent
+4 -7
View File
@@ -62,10 +62,7 @@ func setupBackendTest(t *testing.T) (testBackend, context.Context) {
ctx := testutil.NewDefaultTestContext(t)
dbp := test.NewDBProviderMatchWords(t)
b, err := NewBackend(BackendOptions{
DBProvider: dbp,
SkipDataMigration: true, // Calling migrations makes startup SQL calls (avoid the mock)
})
b, err := NewBackend(BackendOptions{DBProvider: dbp})
require.NoError(t, err)
require.NotNil(t, b)
@@ -112,7 +109,7 @@ func TestBackend_Init(t *testing.T) {
ctx := testutil.NewDefaultTestContext(t)
dbp := test.NewDBProviderWithPing(t)
b, err := NewBackend(BackendOptions{DBProvider: dbp, SkipDataMigration: true})
b, err := NewBackend(BackendOptions{DBProvider: dbp})
require.NoError(t, err)
require.NotNil(t, b)
@@ -169,7 +166,7 @@ func TestBackend_Init(t *testing.T) {
ctx := testutil.NewDefaultTestContext(t)
dbp := test.NewDBProviderWithPing(t)
b, err := NewBackend(BackendOptions{DBProvider: dbp, SkipDataMigration: true})
b, err := NewBackend(BackendOptions{DBProvider: dbp})
require.NoError(t, err)
require.NotNil(t, dbp.DB)
@@ -185,7 +182,7 @@ func TestBackend_IsHealthy(t *testing.T) {
ctx := testutil.NewDefaultTestContext(t)
dbp := test.NewDBProviderWithPing(t)
b, err := NewBackend(BackendOptions{DBProvider: dbp, SkipDataMigration: true})
b, err := NewBackend(BackendOptions{DBProvider: dbp})
require.NoError(t, err)
require.NotNil(t, dbp.DB)