mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-10-24 19:13:59 +00:00
e8b993e7da
This allows you to restrict email notifications to only be sent after the build changes from success to failure or failure to success. It errs on the side of sending the notification; if the build is in another state (hung, for instance) or there was no previous build on the branch the email will also be sent. Since the notify plugin shouldn't really have any responsibility for querying the database to find the previous commit's status, we store it on the commit when we save it.
360 lines
9.7 KiB
Go
360 lines
9.7 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/drone/drone/shared/model"
|
|
"github.com/franela/goblin"
|
|
)
|
|
|
|
func TestCommitstore(t *testing.T) {
|
|
db := mustConnectTest()
|
|
cs := NewCommitstore(db)
|
|
rs := NewRepostore(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 repos")
|
|
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's prior status", func() {
|
|
commit := model.Commit{
|
|
RepoID: 1,
|
|
Branch: "foo",
|
|
Status: "Failure",
|
|
Created: 1419892236,
|
|
Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
}
|
|
err := cs.PostCommit(&commit)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(commit.ID != 0).IsTrue()
|
|
g.Assert(commit.PriorStatus == "").IsTrue()
|
|
|
|
commit = model.Commit{
|
|
RepoID: 1,
|
|
Branch: "foo",
|
|
Status: "Success",
|
|
Created: 1419893583,
|
|
Sha: "25f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
}
|
|
err = cs.PostCommit(&commit)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(commit.ID != 0).IsTrue()
|
|
g.Assert(commit.PriorStatus == "Failure").IsTrue()
|
|
})
|
|
|
|
g.It("Should not find prior status from commits on other branches", func() {
|
|
commit := model.Commit{
|
|
RepoID: 1,
|
|
Branch: "foo",
|
|
Status: "Failure",
|
|
Created: 1419892236,
|
|
Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
}
|
|
err := cs.PostCommit(&commit)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(commit.ID != 0).IsTrue()
|
|
g.Assert(commit.PriorStatus == "").IsTrue()
|
|
|
|
commit = model.Commit{
|
|
RepoID: 1,
|
|
Branch: "bar",
|
|
Status: "Success",
|
|
Created: 1419893583,
|
|
Sha: "25f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
}
|
|
err = cs.PostCommit(&commit)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(commit.ID != 0).IsTrue()
|
|
g.Assert(commit.PriorStatus == "").IsTrue()
|
|
})
|
|
|
|
g.It("Should not find prior status from commits that didn't succeed or fail", func() {
|
|
commit := model.Commit{
|
|
RepoID: 1,
|
|
Branch: "foo",
|
|
Status: "Pending",
|
|
Created: 1419892236,
|
|
Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
}
|
|
err := cs.PostCommit(&commit)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(commit.ID != 0).IsTrue()
|
|
g.Assert(commit.PriorStatus == "").IsTrue()
|
|
|
|
commit = model.Commit{
|
|
RepoID: 1,
|
|
Branch: "bar",
|
|
Status: "Success",
|
|
Created: 1419893583,
|
|
Sha: "25f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
}
|
|
err = cs.PostCommit(&commit)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(commit.ID != 0).IsTrue()
|
|
g.Assert(commit.PriorStatus == "").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", func() {
|
|
repo1 := model.Repo{
|
|
UserID: 1,
|
|
Remote: "enterprise.github.com",
|
|
Host: "github.drone.io",
|
|
Owner: "bradrydzewski",
|
|
Name: "drone",
|
|
}
|
|
repo2 := model.Repo{
|
|
UserID: 1,
|
|
Remote: "enterprise.github.com",
|
|
Host: "github.drone.io",
|
|
Owner: "drone",
|
|
Name: "drone",
|
|
}
|
|
repo3 := model.Repo{
|
|
UserID: 2,
|
|
Remote: "enterprise.github.com",
|
|
Host: "github.drone.io",
|
|
Owner: "droneio",
|
|
Name: "drone",
|
|
}
|
|
rs.PostRepo(&repo1)
|
|
rs.PostRepo(&repo2)
|
|
commit1 := model.Commit{
|
|
RepoID: repo1.ID,
|
|
Branch: "foo",
|
|
Sha: "85f8c029b902ed9400bc600bac301a0aadb144ac",
|
|
Status: model.StatusFailure,
|
|
}
|
|
commit2 := model.Commit{
|
|
RepoID: repo2.ID,
|
|
Branch: "bar",
|
|
Sha: "0a74b46d7d62b737b6906897f48dbeb72cfda222",
|
|
Status: model.StatusSuccess,
|
|
}
|
|
commit3 := model.Commit{
|
|
RepoID: 99999,
|
|
Branch: "baz",
|
|
Sha: "0a74b46d7d62b737b6906897f48dbeb72cfda222",
|
|
Status: model.StatusSuccess,
|
|
}
|
|
commit4 := model.Commit{
|
|
RepoID: repo2.ID,
|
|
Branch: "bar",
|
|
Sha: "d923a61d8ad3d8d02db4fef0bf40a726bad0fc03",
|
|
Status: model.StatusStarted,
|
|
}
|
|
commit5 := model.Commit{
|
|
RepoID: repo3.ID,
|
|
Branch: "bar",
|
|
Sha: "d923a61d8ad3d8d02db4fef0bf40a726bad0fc03",
|
|
Status: model.StatusStarted,
|
|
}
|
|
cs.PostCommit(&commit1)
|
|
cs.PostCommit(&commit2)
|
|
cs.PostCommit(&commit3)
|
|
cs.PostCommit(&commit4)
|
|
cs.PostCommit(&commit5)
|
|
perm1 := model.Perm{
|
|
RepoID: repo1.ID,
|
|
UserID: 1,
|
|
Read: true,
|
|
Write: true,
|
|
Admin: true,
|
|
}
|
|
perm2 := model.Perm{
|
|
RepoID: repo2.ID,
|
|
UserID: 1,
|
|
Read: true,
|
|
Write: true,
|
|
Admin: true,
|
|
}
|
|
ps.PostPerm(&perm1)
|
|
ps.PostPerm(&perm2)
|
|
commits, err := cs.GetCommitListUser(&model.User{ID: 1})
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(len(commits)).Equal(2)
|
|
g.Assert(commits[0].RepoID).Equal(commit1.RepoID)
|
|
g.Assert(commits[0].Branch).Equal(commit1.Branch)
|
|
g.Assert(commits[0].Sha).Equal(commit1.Sha)
|
|
g.Assert(commits[1].Sha).Equal(commit4.Sha)
|
|
g.Assert(commits[1].Status).Equal(commit4.Status)
|
|
|
|
commits, err = cs.GetCommitListActivity(&model.User{ID: 1})
|
|
fmt.Println(commits)
|
|
fmt.Println(err)
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(len(commits)).Equal(3)
|
|
})
|
|
|
|
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()
|
|
})
|
|
})
|
|
}
|