mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-27 14:41:59 +08:00
49 lines
1.6 KiB
Go
49 lines
1.6 KiB
Go
package sqlmigrator
|
|
|
|
import (
|
|
"context"
|
|
"database/sql/driver"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"github.com/stretchr/testify/require"
|
|
"go.signoz.io/signoz/pkg/factory"
|
|
"go.signoz.io/signoz/pkg/instrumentation/instrumentationtest"
|
|
"go.signoz.io/signoz/pkg/sqlstore"
|
|
"go.signoz.io/signoz/pkg/sqlstore/sqlstoretest"
|
|
)
|
|
|
|
func TestMigratorWithSqliteAndNoopMigration(t *testing.T) {
|
|
ctx := context.Background()
|
|
sqlstoreConfig := sqlstore.Config{
|
|
Provider: "sqlite",
|
|
}
|
|
|
|
migrationConfig := Config{
|
|
Lock: Lock{
|
|
Timeout: 10 * time.Second,
|
|
Interval: 1 * time.Second,
|
|
},
|
|
}
|
|
|
|
providerSettings := instrumentationtest.New().ToProviderSettings()
|
|
sqlstore := sqlstoretest.New(sqlstoreConfig, sqlmock.QueryMatcherRegexp)
|
|
migrator := New(
|
|
ctx,
|
|
providerSettings,
|
|
sqlstore,
|
|
MustNewMigrations(ctx, providerSettings, migrationConfig, factory.MustNewNamedMap(NoopMigrationFactory())),
|
|
migrationConfig,
|
|
)
|
|
|
|
sqlstore.Mock().ExpectExec("CREATE TABLE IF NOT EXISTS migration (.+)").WillReturnResult(driver.ResultNoRows)
|
|
sqlstore.Mock().ExpectExec("CREATE TABLE IF NOT EXISTS migration_lock (.+)").WillReturnResult(driver.ResultNoRows)
|
|
sqlstore.Mock().ExpectQuery("INSERT INTO migration_lock (.+)").WillReturnRows(sqlstore.Mock().NewRows([]string{"id"}).AddRow(1))
|
|
sqlstore.Mock().ExpectQuery("(.+) FROM migration").WillReturnRows(sqlstore.Mock().NewRows([]string{"id"}).AddRow(1))
|
|
sqlstore.Mock().ExpectQuery("INSERT INTO migration (.+)").WillReturnRows(sqlstore.Mock().NewRows([]string{"id", "migrated_at"}).AddRow(1, time.Now()))
|
|
|
|
err := migrator.Migrate(ctx)
|
|
require.NoError(t, err)
|
|
}
|