Function TestCredentialsAndEndpointMigration

database/sql/github_test.go:913–1052

(t *testing.T)

}

func TestCredentialsAndEndpointMigration(t *testing.T) {
	cfg := garmTesting.GetTestSqliteDBConfig(t)

	// Copy the sample DB
	data, err := os.ReadFile("../../testdata/db/v0.1.4/garm.db")
	if err != nil {
		t.Fatalf("failed to read test data: %s", err)
	}

	if cfg.SQLite.DBFile == "" {
		t.Fatalf("DB file not set")
	}
	if err := os.WriteFile(cfg.SQLite.DBFile, data, 0o600); err != nil {
		t.Fatalf("failed to write test data: %s", err)
	}

	// define some credentials
	credentials := []config.Github{
		{
			Name:        "test-creds",
			Description: "test creds",
			AuthType:    config.GithubAuthTypePAT,
			PAT: config.GithubPAT{
				OAuth2Token: "test",
			},
		},
		{
			Name:          "ghes-test",
			Description:   "ghes creds",
			APIBaseURL:    testAPIBaseURL,
			UploadBaseURL: testUploadBaseURL,
			BaseURL:       testBaseURL,
			AuthType:      config.GithubAuthTypeApp,
			App: config.GithubApp{
				AppID:          1,
				InstallationID: 99,
				PrivateKeyPath: "../../testdata/certs/srv-key.pem",
			},
		},
	}
	// Set the config credentials in the cfg. This is what happens in the main function.
	// of GARM as well.
	cfg.MigrateCredentials = credentials

	ctx := context.Background()
	watcher.InitWatcher(ctx)
	defer watcher.CloseWatcher()

	db, err := NewSQLDatabase(ctx, cfg)
	if err != nil {
		t.Fatalf("failed to create db connection: %s", err)
	}

	// We expect that 2 endpoints will exist in the migrated DB and 2 credentials.
	ctx = garmTesting.ImpersonateAdminContext(ctx, db, t)

	endpoints, err := db.ListGithubEndpoints(ctx)
	if err != nil {