From 4d431954791347d086a2c816633e5e48506ff7aa Mon Sep 17 00:00:00 2001 From: Thomas Boerger Date: Mon, 1 Aug 2016 23:01:19 +0200 Subject: [PATCH] Moved merged secret list into store --- store/datastore/secret.go | 33 ------------------------ store/datastore/secret_test.go | 46 ---------------------------------- store/store.go | 29 ++++++++++++++++++--- 3 files changed, 25 insertions(+), 83 deletions(-) delete mode 100644 store/datastore/secret.go delete mode 100644 store/datastore/secret_test.go diff --git a/store/datastore/secret.go b/store/datastore/secret.go deleted file mode 100644 index 3f3399c6e..000000000 --- a/store/datastore/secret.go +++ /dev/null @@ -1,33 +0,0 @@ -package datastore - -import ( - "github.com/drone/drone/model" -) - -func (db *datastore) GetMergedSecretList(repo *model.Repo) ([]*model.Secret, error) { - var ( - secrets []*model.Secret - ) - - repoSecs, err := db.GetSecretList(repo) - - if err != nil { - return nil, err - } - - for _, secret := range repoSecs { - secrets = append(secrets, secret.Secret()) - } - - teamSecs, err := db.GetTeamSecretList(repo.Owner) - - if err != nil { - return nil, err - } - - for _, secret := range teamSecs { - secrets = append(secrets, secret.Secret()) - } - - return secrets, nil -} diff --git a/store/datastore/secret_test.go b/store/datastore/secret_test.go deleted file mode 100644 index c955ac0df..000000000 --- a/store/datastore/secret_test.go +++ /dev/null @@ -1,46 +0,0 @@ -package datastore - -import ( - "testing" - - "github.com/drone/drone/model" - "github.com/franela/goblin" -) - -func TestSecrets(t *testing.T) { - db := openTest() - defer db.Close() - - s := From(db) - g := goblin.Goblin(t) - g.Describe("Secrets", func() { - - // before each test be sure to purge the package - // table data from the database. - g.BeforeEach(func() { - db.Exec(rebind("DELETE FROM secrets")) - db.Exec(rebind("DELETE FROM team_secrets")) - }) - - g.It("Should list all secrets", func() { - teamSec := &model.TeamSecret{ - Key: "octocat", - Name: "foo", - Value: "team", - } - - repoSec := &model.RepoSecret{ - RepoID: 1, - Name: "foo", - Value: "repo", - } - - s.SetSecret(repoSec) - s.SetTeamSecret(teamSec) - - secrets, err := s.GetMergedSecretList(&model.Repo{ID: 1, Owner: "octocat"}) - g.Assert(err == nil).IsTrue() - g.Assert(len(secrets)).Equal(2) - }) - }) -} diff --git a/store/store.go b/store/store.go index 624ae58d0..e531054e5 100644 --- a/store/store.go +++ b/store/store.go @@ -82,9 +82,6 @@ type Store interface { // DeleteTeamSecret deletes the named team secret. DeleteTeamSecret(*model.TeamSecret) error - // GetMergedSecretList gets a list of repo and team secrets - GetMergedSecretList(*model.Repo) ([]*model.Secret, error) - // GetBuild gets a build by unique ID. GetBuild(int64) (*model.Build, error) @@ -250,7 +247,31 @@ func DeleteTeamSecret(c context.Context, s *model.TeamSecret) error { } func GetMergedSecretList(c context.Context, r *model.Repo) ([]*model.Secret, error) { - return FromContext(c).GetMergedSecretList(r) + var ( + secrets []*model.Secret + ) + + repoSecs, err := FromContext(c).GetSecretList(r) + + if err != nil { + return nil, err + } + + for _, secret := range repoSecs { + secrets = append(secrets, secret.Secret()) + } + + teamSecs, err := FromContext(c).GetTeamSecretList(r.Owner) + + if err != nil { + return nil, err + } + + for _, secret := range teamSecs { + secrets = append(secrets, secret.Secret()) + } + + return secrets, nil } func GetBuild(c context.Context, id int64) (*model.Build, error) {