mirror of
https://github.com/harness/drone.git
synced 2025-05-09 01:11:38 +08:00
8c2f900c80
This change introduces the concept of a principal (abstraction of call identity), and adds a new service account type principal. Also adds support for different tokens (session, PAT, SAT, OAuth2) and adds auth.Session which is being used to capture information about the caller and call method.
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
// Copyright 2021 Harness Inc. All rights reserved.
|
|
// Use of this source code is governed by the Polyform Free Trial License
|
|
// that can be found in the LICENSE.md file for this repository.
|
|
|
|
package database
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
|
|
_ "github.com/lib/pq"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
// connect opens a new test database connection.
|
|
func connect() (*sqlx.DB, error) {
|
|
var (
|
|
driver = "sqlite3"
|
|
config = ":memory:?_foreign_keys=1"
|
|
)
|
|
if os.Getenv("DATABASE_CONFIG") != "" {
|
|
driver = os.Getenv("DATABASE_DRIVER")
|
|
config = os.Getenv("DATABASE_CONFIG")
|
|
}
|
|
return Connect(driver, config)
|
|
}
|
|
|
|
// seed seed the database state.
|
|
func seed(db *sqlx.DB) error {
|
|
_, err := db.Exec("DELETE FROM executions")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = db.Exec("DELETE FROM pipelines")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = db.Exec("DELETE FROM users")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = db.Exec("ALTER SEQUENCE users_user_id_seq RESTART WITH 1")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = db.Exec("ALTER SEQUENCE pipelines_pipeline_id_seq RESTART WITH 1")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = db.Exec("ALTER SEQUENCE executions_execution_id_seq RESTART WITH 1")
|
|
return err
|
|
}
|
|
|
|
// unmarshal a testdata file.
|
|
//nolint:unparam, goimports // expected to be called for other paths in the future.
|
|
func unmarshal(path string, v interface{}) error {
|
|
out, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return json.Unmarshal(out, v)
|
|
}
|