Update store.go

This commit is contained in:
İsmail
2024-11-04 15:46:10 +03:00
committed by GitHub
parent ee49368f98
commit 0c1cac3d89

View File

@@ -305,6 +305,23 @@ func NewTestStoreFromSQL(ctx context.Context, filename string, dataDir string) (
}
}
if kind == MysqlStoreEngine {
cleanUp, err = testutil.CreateMyDB()
if err != nil {
return nil, nil, err
}
dsn, ok := os.LookupEnv(mysqlDsnEnv)
if !ok {
return nil, nil, fmt.Errorf("%s is not set", mysqlDsnEnv)
}
store, err = NewMysqlStoreFromSqlStore(ctx, store, dsn, nil)
if err != nil {
return nil, nil, err
}
}
return store, cleanUp, nil
}