package database import ( "testing" "github.com/drone/drone/shared/model" "github.com/franela/goblin" ) func TestCommitstore(t *testing.T) { db := mustConnectTest() cs := NewCommitstore(db) //ps := NewPermstore(db) defer db.Close() g := goblin.Goblin(t) g.Describe("Commitstore", func() { // before each test be sure to purge the package // table data from the database. g.BeforeEach(func() { db.Exec("DELETE FROM perms") db.Exec("DELETE FROM commits") }) g.It("Should Put a Commit", func() { commit := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", } err := cs.PutCommit(&commit) g.Assert(err == nil).IsTrue() g.Assert(commit.ID != 0).IsTrue() }) g.It("Should Post a Commit", func() { commit := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", } err := cs.PostCommit(&commit) g.Assert(err == nil).IsTrue() g.Assert(commit.ID != 0).IsTrue() }) g.It("Should Get a Commit by ID", func() { commit := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusSuccess, Created: 1398065343, Updated: 1398065344, } cs.PostCommit(&commit) getcommit, err := cs.GetCommit(commit.ID) g.Assert(err == nil).IsTrue() g.Assert(commit.ID).Equal(getcommit.ID) g.Assert(commit.RepoID).Equal(getcommit.RepoID) g.Assert(commit.Branch).Equal(getcommit.Branch) g.Assert(commit.Sha).Equal(getcommit.Sha) g.Assert(commit.Status).Equal(getcommit.Status) g.Assert(commit.Created).Equal(getcommit.Created) g.Assert(commit.Updated).Equal(getcommit.Updated) }) g.It("Should Delete a Commit", func() { commit := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", } cs.PostCommit(&commit) _, err1 := cs.GetCommit(commit.ID) err2 := cs.DelCommit(&commit) _, err3 := cs.GetCommit(commit.ID) g.Assert(err1 == nil).IsTrue() g.Assert(err2 == nil).IsTrue() g.Assert(err3 == nil).IsFalse() }) g.It("Should Kill Pending or Started Commits", func() { commit1 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusEnqueue, } commit2 := model.Commit{ RepoID: 1, Branch: "bar", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusEnqueue, } cs.PutCommit(&commit1) cs.PutCommit(&commit2) err := cs.KillCommits() g.Assert(err == nil).IsTrue() getcommit1, _ := cs.GetCommit(commit1.ID) getcommit2, _ := cs.GetCommit(commit1.ID) g.Assert(getcommit1.Status).Equal(model.StatusKilled) g.Assert(getcommit2.Status).Equal(model.StatusKilled) }) g.It("Should Get a Commit by Sha", func() { commit := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", } cs.PostCommit(&commit) getcommit, err := cs.GetCommitSha(&model.Repo{ID: 1}, commit.Branch, commit.Sha) g.Assert(err == nil).IsTrue() g.Assert(commit.ID).Equal(getcommit.ID) g.Assert(commit.RepoID).Equal(getcommit.RepoID) g.Assert(commit.Branch).Equal(getcommit.Branch) g.Assert(commit.Sha).Equal(getcommit.Sha) }) g.It("Should get the last Commit by Branch", func() { commit1 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusFailure, } commit2 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "0a74b46d7d62b737b6906897f48dbeb72cfda222", Status: model.StatusSuccess, } cs.PutCommit(&commit1) cs.PutCommit(&commit2) lastcommit, err := cs.GetCommitLast(&model.Repo{ID: 1}, commit1.Branch) g.Assert(err == nil).IsTrue() g.Assert(commit2.ID).Equal(lastcommit.ID) g.Assert(commit2.RepoID).Equal(lastcommit.RepoID) g.Assert(commit2.Branch).Equal(lastcommit.Branch) g.Assert(commit2.Sha).Equal(lastcommit.Sha) }) g.It("Should get the recent Commit List for a Repo", func() { commit1 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusFailure, } commit2 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "0a74b46d7d62b737b6906897f48dbeb72cfda222", Status: model.StatusSuccess, } cs.PutCommit(&commit1) cs.PutCommit(&commit2) commits, err := cs.GetCommitList(&model.Repo{ID: 1}) g.Assert(err == nil).IsTrue() g.Assert(len(commits)).Equal(2) g.Assert(commits[0].ID).Equal(commit2.ID) g.Assert(commits[0].RepoID).Equal(commit2.RepoID) g.Assert(commits[0].Branch).Equal(commit2.Branch) g.Assert(commits[0].Sha).Equal(commit2.Sha) }) g.It("Should get the recent Commit List for a User") /* g.It("Should get the recent Commit List for a User", func() { perm1 := model.Perm{ RepoID: 1, UserID: 1, Read: true, Write: true, Admin: true, } commit1 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusFailure, } commit2 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "0a74b46d7d62b737b6906897f48dbeb72cfda222", Status: model.StatusSuccess, } commit3 := model.Commit{ RepoID: 2, Branch: "baz", Sha: "0a74b46d7d62b737b6906897f48dbeb72cfda222", Status: model.StatusSuccess, } cs.PutCommit(&commit1) cs.PutCommit(&commit2) cs.PutCommit(&commit3) ps.PutPerm(&perm1) commits, err := cs.GetCommitListUser(&model.User{ID: 1}) g.Assert(err == nil).IsTrue() g.Assert(len(commits)).Equal(2) g.Assert(commits[0].ID).Equal(commit2.ID) g.Assert(commits[0].RepoID).Equal(commit2.RepoID) g.Assert(commits[0].Branch).Equal(commit2.Branch) g.Assert(commits[0].Sha).Equal(commit2.Sha) }) */ g.It("Should enforce unique Sha + Branch", func() { commit1 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusEnqueue, } commit2 := model.Commit{ RepoID: 1, Branch: "foo", Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac", Status: model.StatusEnqueue, } err1 := cs.PutCommit(&commit1) err2 := cs.PutCommit(&commit2) g.Assert(err1 == nil).IsTrue() g.Assert(err2 == nil).IsFalse() }) }) }