mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-26 06:04:31 +08:00
3bb9e05681
* chore(dashboard): intial commit * chore(dashboard): bring all the code in module * chore(dashboard): remove lock unlock from ee codebase * chore(dashboard): go deps * chore(dashboard): fix lint * chore(dashboard): implement the store * chore(dashboard): add migration * chore(dashboard): fix lint * chore(dashboard): api and frontend changes * chore(dashboard): frontend changes for new dashboards * chore(dashboard): fix test cases * chore(dashboard): add lock unlock APIs * chore(dashboard): add lock unlock APIs * chore(dashboard): move integrations controller out from module * chore(dashboard): move integrations controller out from module * chore(dashboard): move integrations controller out from module * chore(dashboard): rename migration file * chore(dashboard): surface errors for lock/unlock dashboard * chore(dashboard): some testing cleanups * chore(dashboard): fix postgres migrations --------- Co-authored-by: Vibhu Pandey <vibhupandey28@gmail.com>
90 lines
3.3 KiB
Go
90 lines
3.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/SigNoz/signoz/pkg/factory"
|
|
"github.com/SigNoz/signoz/pkg/factory/factorytest"
|
|
"github.com/SigNoz/signoz/pkg/sqlmigration"
|
|
"github.com/SigNoz/signoz/pkg/sqlmigrator"
|
|
"github.com/SigNoz/signoz/pkg/sqlstore"
|
|
"github.com/SigNoz/signoz/pkg/sqlstore/sqlitesqlstore"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func NewTestSqliteDB(t *testing.T) (sqlStore sqlstore.SQLStore, testDBFilePath string) {
|
|
testDBFile, err := os.CreateTemp("", "test-signoz-db-*")
|
|
if err != nil {
|
|
t.Fatalf("could not create temp file for test db: %v", err)
|
|
}
|
|
testDBFilePath = testDBFile.Name()
|
|
t.Cleanup(func() { os.Remove(testDBFilePath) })
|
|
testDBFile.Close()
|
|
|
|
sqlStore, err = sqlitesqlstore.New(context.Background(), factorytest.NewSettings(), sqlstore.Config{Provider: "sqlite", Sqlite: sqlstore.SqliteConfig{Path: testDBFilePath}})
|
|
if err != nil {
|
|
t.Fatalf("could not create test db sqlite store: %v", err)
|
|
}
|
|
|
|
sqlmigrations, err := sqlmigration.New(
|
|
context.Background(),
|
|
factorytest.NewSettings(),
|
|
sqlmigration.Config{},
|
|
factory.MustNewNamedMap(
|
|
sqlmigration.NewAddDataMigrationsFactory(),
|
|
sqlmigration.NewAddOrganizationFactory(),
|
|
sqlmigration.NewAddPreferencesFactory(),
|
|
sqlmigration.NewAddDashboardsFactory(),
|
|
sqlmigration.NewAddSavedViewsFactory(),
|
|
sqlmigration.NewAddAgentsFactory(),
|
|
sqlmigration.NewAddPipelinesFactory(),
|
|
sqlmigration.NewAddIntegrationsFactory(),
|
|
sqlmigration.NewAddLicensesFactory(),
|
|
sqlmigration.NewAddPatsFactory(),
|
|
sqlmigration.NewModifyDatetimeFactory(),
|
|
sqlmigration.NewModifyOrgDomainFactory(),
|
|
sqlmigration.NewUpdateOrganizationFactory(sqlStore),
|
|
sqlmigration.NewAddAlertmanagerFactory(sqlStore),
|
|
sqlmigration.NewUpdateDashboardAndSavedViewsFactory(sqlStore),
|
|
sqlmigration.NewUpdatePatAndOrgDomainsFactory(sqlStore),
|
|
sqlmigration.NewUpdatePipelines(sqlStore),
|
|
sqlmigration.NewDropLicensesSitesFactory(sqlStore),
|
|
sqlmigration.NewUpdateInvitesFactory(sqlStore),
|
|
sqlmigration.NewUpdatePatFactory(sqlStore),
|
|
sqlmigration.NewUpdateAlertmanagerFactory(sqlStore),
|
|
sqlmigration.NewUpdatePreferencesFactory(sqlStore),
|
|
sqlmigration.NewUpdateApdexTtlFactory(sqlStore),
|
|
sqlmigration.NewUpdateResetPasswordFactory(sqlStore),
|
|
sqlmigration.NewUpdateRulesFactory(sqlStore),
|
|
sqlmigration.NewAddVirtualFieldsFactory(),
|
|
sqlmigration.NewUpdateIntegrationsFactory(sqlStore),
|
|
sqlmigration.NewUpdateOrganizationsFactory(sqlStore),
|
|
sqlmigration.NewDropGroupsFactory(sqlStore),
|
|
sqlmigration.NewCreateQuickFiltersFactory(sqlStore),
|
|
sqlmigration.NewUpdateQuickFiltersFactory(sqlStore),
|
|
sqlmigration.NewAuthRefactorFactory(sqlStore),
|
|
sqlmigration.NewMigratePATToFactorAPIKey(sqlStore),
|
|
sqlmigration.NewUpdateApiMonitoringFiltersFactory(sqlStore),
|
|
sqlmigration.NewAddKeyOrganizationFactory(sqlStore),
|
|
sqlmigration.NewUpdateDashboardFactory(sqlStore),
|
|
),
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("could not create test db sql migrations: %v", err)
|
|
}
|
|
|
|
err = sqlmigrator.New(context.Background(), factorytest.NewSettings(), sqlStore, sqlmigrations, sqlmigrator.Config{}).Migrate(context.Background())
|
|
if err != nil {
|
|
t.Fatalf("could not migrate test db sql migrations: %v", err)
|
|
}
|
|
|
|
return sqlStore, testDBFilePath
|
|
}
|
|
|
|
func NewQueryServiceDBForTests(t *testing.T) sqlstore.SQLStore {
|
|
sqlStore, _ := NewTestSqliteDB(t)
|
|
return sqlStore
|
|
}
|