func (cmd *ATCCommand) updateBasicAuthCredentials(sqlDB db.DB) error {
var team db.Team
if cmd.BasicAuth.Username != "" && cmd.BasicAuth.Password != "" {
team = db.Team{
Name: atc.DefaultTeamName,
BasicAuth: db.BasicAuth{
BasicAuthUsername: cmd.BasicAuth.Username,
BasicAuthPassword: cmd.BasicAuth.Password,
},
}
} else {
team = db.Team{Name: atc.DefaultTeamName}
}
_, err := sqlDB.UpdateTeamBasicAuth(team)
return err
}
func (cmd *ATCCommand) configureOAuthProviders(logger lager.Logger, sqlDB db.DB) error {
var err error
team := db.Team{
Name: atc.DefaultTeamName,
}
gitHubTeams := []db.GitHubTeam{}
for _, gitHubTeam := range cmd.GitHubAuth.Teams {
gitHubTeams = append(gitHubTeams, db.GitHubTeam{
TeamName: gitHubTeam.TeamName,
OrganizationName: gitHubTeam.OrganizationName,
})
}
if len(cmd.GitHubAuth.Organizations) > 0 ||
len(gitHubTeams) > 0 ||
len(cmd.GitHubAuth.Users) > 0 {
gitHubAuth := db.GitHubAuth{
ClientID: cmd.GitHubAuth.ClientID,
ClientSecret: cmd.GitHubAuth.ClientSecret,
Organizations: cmd.GitHubAuth.Organizations,
Teams: gitHubTeams,
Users: cmd.GitHubAuth.Users,
AuthURL: cmd.GitHubAuth.AuthURL,
TokenURL: cmd.GitHubAuth.TokenURL,
APIURL: cmd.GitHubAuth.APIURL,
}
team.GitHubAuth = gitHubAuth
} else {
team.GitHubAuth = db.GitHubAuth{}
}
_, err = sqlDB.UpdateTeamGitHubAuth(team)
if err != nil {
return err
}
return nil
}
"golang.org/x/crypto/bcrypt"
"github.com/lib/pq"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/pivotal-golang/lager/lagertest"
"github.com/concourse/atc"
"github.com/concourse/atc/db"
)
var _ = Describe("SQL DB Teams", func() {
var dbConn db.Conn
var listener *pq.Listener
var database db.DB
BeforeEach(func() {
postgresRunner.Truncate()
dbConn = db.Wrap(postgresRunner.Open())
listener = pq.NewListener(postgresRunner.DataSourceName(), time.Second, time.Minute, nil)
Eventually(listener.Ping, 5*time.Second).ShouldNot(HaveOccurred())
bus := db.NewNotificationsBus(listener, dbConn)
database = db.NewSQL(lagertest.NewTestLogger("test"), dbConn, bus)
database.DeleteTeamByName(atc.DefaultTeamName)
})
"time"
"github.com/lib/pq"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/pivotal-golang/lager/lagertest"
"github.com/concourse/atc"
"github.com/concourse/atc/db"
)
var _ = Describe("Keeping track of workers", func() {
var dbConn db.Conn
var listener *pq.Listener
var database db.DB
BeforeEach(func() {
postgresRunner.Truncate()
dbConn = db.Wrap(postgresRunner.Open())
listener = pq.NewListener(postgresRunner.DataSourceName(), time.Second, time.Minute, nil)
Eventually(listener.Ping, 5*time.Second).ShouldNot(HaveOccurred())
bus := db.NewNotificationsBus(listener, dbConn)
database = db.NewSQL(lagertest.NewTestLogger("test"), dbConn, bus)
})
AfterEach(func() {
err := dbConn.Close()
"github.com/lib/pq"
"github.com/nu7hatch/gouuid"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/pivotal-golang/lager/lagertest"
"github.com/concourse/atc"
"github.com/concourse/atc/db"
"github.com/concourse/atc/event"
)
var _ = Describe("SQL DB", func() {
var dbConn db.Conn
var listener *pq.Listener
var database db.DB
BeforeEach(func() {
postgresRunner.Truncate()
dbConn = db.Wrap(postgresRunner.Open())
listener = pq.NewListener(postgresRunner.DataSourceName(), time.Second, time.Minute, nil)
Eventually(listener.Ping, 5*time.Second).ShouldNot(HaveOccurred())
bus := db.NewNotificationsBus(listener, dbConn)
database = db.NewSQL(lagertest.NewTestLogger("test"), dbConn, bus)
})
AfterEach(func() {
err := dbConn.Close()
"time"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/concourse/atc"
"github.com/concourse/atc/db"
"github.com/lib/pq"
"github.com/pivotal-golang/lager/lagertest"
)
var _ = Describe("Keeping track of volumes", func() {
var dbConn db.Conn
var listener *pq.Listener
var database db.DB
var pipelineDB db.PipelineDB
BeforeEach(func() {
postgresRunner.Truncate()
dbConn = db.Wrap(postgresRunner.Open())
listener = pq.NewListener(postgresRunner.DataSourceName(), time.Second, time.Minute, nil)
Eventually(listener.Ping, 5*time.Second).ShouldNot(HaveOccurred())
bus := db.NewNotificationsBus(listener, dbConn)
sqlDB := db.NewSQL(lagertest.NewTestLogger("test"), dbConn, bus)
database = sqlDB
pipelineDBFactory := db.NewPipelineDBFactory(lagertest.NewTestLogger("test"), dbConn, bus, sqlDB)
"time"
"github.com/lib/pq"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/pivotal-golang/lager/lagertest"
"github.com/concourse/atc"
"github.com/concourse/atc/db"
)
var _ = Describe("Keeping track of builds", func() {
var dbConn db.Conn
var listener *pq.Listener
var database db.DB
var pipelineDB db.PipelineDB
BeforeEach(func() {
postgresRunner.Truncate()
dbConn = db.Wrap(postgresRunner.Open())
listener = pq.NewListener(postgresRunner.DataSourceName(), time.Second, time.Minute, nil)
Eventually(listener.Ping, 5*time.Second).ShouldNot(HaveOccurred())
bus := db.NewNotificationsBus(listener, dbConn)
sqlDB := db.NewSQL(lagertest.NewTestLogger("test"), dbConn, bus)
pipelineDBFactory := db.NewPipelineDBFactory(lagertest.NewTestLogger("test"), dbConn, bus, sqlDB)