db manager

tests/service_test.go

@@ -12,6 +12,18 @@ import (
 	"gorm.io/gorm"
 )
 
+type dbManager struct {
+	db *gorm.DB
+}
+
+func (s *dbManager) GetMaster() *gorm.DB {
+	return s.db
+}
+
+func (s *dbManager) GetReplica() *gorm.DB {
+	return s.db
+}
+
 // helper: инициализация базы и сервиса
 func newTestService(t *testing.T) (*gorm.DB, *service.VersionService, *store.GormStorage) {
 	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
@@ -22,10 +34,10 @@ func newTestService(t *testing.T) (*gorm.DB, *service.VersionService, *store.Gor
 		t.Fatalf("failed to migrate: %v", err)
 	}
 
+	manager := &dbManager{db}
+
 	storage := store.NewGormStorage(
-		func() *gorm.DB {
-			return db
-		},
+		manager,
 	)
 	service := service.NewVersionService(storage, 5)
 	return db, service, storage

tests/snapshot_test.go

@@ -22,10 +22,10 @@ func TestSnapshotInterval(t *testing.T) {
 		t.Fatalf("failed to migrate: %v", err)
 	}
 
+	manager := dbManager{db}
+
 	storage := store.NewGormStorage(
-		func() *gorm.DB {
-			return db
-		},
+		&manager,
 	)
 
 	// --- Service with snapshot interval = 3 ---