fix tests

formancehq · Oct 2, 2023 · 3795cf4 · 3795cf4
1 parent 57fca75
commit 3795cf4
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 6 deletions.
diff --git a/cmd/internal/analytics_test.go b/cmd/internal/analytics_test.go
@@ -126,7 +126,7 @@ func TestAnalyticsModule(t *testing.T) {
 		module,
 		fx.Provide(func(lc fx.Lifecycle) (storage.Driver[ledger.Store], error) {
 			id := uuid.New()
-			driver := sqlstorage.NewDriver("sqlite", sqlstorage.NewSQLiteDB(os.TempDir(), id))
+			driver := sqlstorage.NewDriver("sqlite", sqlstorage.NewSQLiteDB(os.TempDir(), id), false)
 			lc.Append(fx.Hook{
 				OnStart: driver.Initialize,
 			})

diff --git a/pkg/analytics/segment_test.go b/pkg/analytics/segment_test.go
@@ -88,7 +88,7 @@ var (
 		}),
 		fx.Provide(func(lc fx.Lifecycle) (storage.Driver[ledger.Store], error) {
 			id := uuid.New()
-			driver := sqlstorage.NewDriver("sqlite", sqlstorage.NewSQLiteDB(os.TempDir(), id))
+			driver := sqlstorage.NewDriver("sqlite", sqlstorage.NewSQLiteDB(os.TempDir(), id), false)
 			lc.Append(fx.Hook{
 				OnStart: driver.Initialize,
 			})

diff --git a/pkg/ledgertesting/storage.go b/pkg/ledgertesting/storage.go
@@ -26,7 +26,7 @@ func StorageDriver() (*sqlstorage.Driver, func(), error) {
 	switch StorageDriverName() {
 	case "sqlite":
 		id := uuid.New()
-		return sqlstorage.NewDriver("sqlite", sqlstorage.NewSQLiteDB(os.TempDir(), id)), func() {}, nil
+		return sqlstorage.NewDriver("sqlite", sqlstorage.NewSQLiteDB(os.TempDir(), id), false), func() {}, nil
 	case "postgres":
 		pgServer, err := pgtesting.PostgresServer()
 		if err != nil {
@@ -39,6 +39,7 @@ func StorageDriver() (*sqlstorage.Driver, func(), error) {
 		return sqlstorage.NewDriver(
 				"postgres",
 				sqlstorage.NewPostgresDB(db),
+				false,
 			), func() {
 				_ = pgServer.Close()
 			}, nil

diff --git a/pkg/storage/sqlstorage/accounts_test.go b/pkg/storage/sqlstorage/accounts_test.go
@@ -14,7 +14,7 @@ func TestAccounts(t *testing.T) {
 	d := NewDriver("sqlite", &sqliteDB{
 		directory: os.TempDir(),
 		dbName:    uuid.New(),
-	})
+	}, false)
 
 	assert.NoError(t, d.Initialize(context.Background()))
 

diff --git a/pkg/storage/sqlstorage/driver_test.go b/pkg/storage/sqlstorage/driver_test.go
@@ -14,7 +14,7 @@ func TestNewDriver(t *testing.T) {
 	d := NewDriver("sqlite", &sqliteDB{
 		directory: os.TempDir(),
 		dbName:    uuid.New(),
-	})
+	}, false)
 
 	assert.NoError(t, d.Initialize(context.Background()))
 
@@ -39,7 +39,7 @@ func TestConfiguration(t *testing.T) {
 	d := NewDriver("sqlite", &sqliteDB{
 		directory: os.TempDir(),
 		dbName:    uuid.New(),
-	})
+	}, false)
 	require.NoError(t, d.Initialize(context.Background()))
 
 	require.NoError(t, d.GetSystemStore().InsertConfiguration(context.Background(), "foo", "bar"))