118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package datastore
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/drone/drone/model"
|
|
"github.com/franela/goblin"
|
|
)
|
|
|
|
func TestJobs(t *testing.T) {
|
|
db := openTest()
|
|
defer db.Close()
|
|
|
|
s := From(db)
|
|
g := goblin.Goblin(t)
|
|
g.Describe("Job", func() {
|
|
|
|
// before each test we purge the package table data from the database.
|
|
g.BeforeEach(func() {
|
|
db.Exec("DELETE FROM jobs")
|
|
db.Exec("DELETE FROM builds")
|
|
})
|
|
|
|
g.It("Should Set a job", func() {
|
|
job := &model.Job{
|
|
BuildID: 1,
|
|
Status: "pending",
|
|
ExitCode: 0,
|
|
Number: 1,
|
|
}
|
|
err1 := s.CreateJob(job)
|
|
g.Assert(err1 == nil).IsTrue()
|
|
g.Assert(job.ID != 0).IsTrue()
|
|
|
|
job.Status = "started"
|
|
err2 := s.UpdateJob(job)
|
|
g.Assert(err2 == nil).IsTrue()
|
|
|
|
getjob, err3 := s.GetJob(job.ID)
|
|
g.Assert(err3 == nil).IsTrue()
|
|
g.Assert(getjob.Status).Equal(job.Status)
|
|
})
|
|
|
|
g.It("Should Get a Job by ID", func() {
|
|
job := &model.Job{
|
|
BuildID: 1,
|
|
Status: "pending",
|
|
ExitCode: 1,
|
|
Number: 1,
|
|
Environment: map[string]string{"foo": "bar"},
|
|
}
|
|
err1 := s.CreateJob(job)
|
|
g.Assert(err1 == nil).IsTrue()
|
|
g.Assert(job.ID != 0).IsTrue()
|
|
|
|
getjob, err2 := s.GetJob(job.ID)
|
|
g.Assert(err2 == nil).IsTrue()
|
|
g.Assert(getjob.ID).Equal(job.ID)
|
|
g.Assert(getjob.Status).Equal(job.Status)
|
|
g.Assert(getjob.ExitCode).Equal(job.ExitCode)
|
|
g.Assert(getjob.Environment).Equal(job.Environment)
|
|
g.Assert(getjob.Environment["foo"]).Equal("bar")
|
|
})
|
|
|
|
g.It("Should Get a Job by Number", func() {
|
|
job := &model.Job{
|
|
BuildID: 1,
|
|
Status: "pending",
|
|
ExitCode: 1,
|
|
Number: 1,
|
|
}
|
|
err1 := s.CreateJob(job)
|
|
g.Assert(err1 == nil).IsTrue()
|
|
g.Assert(job.ID != 0).IsTrue()
|
|
|
|
getjob, err2 := s.GetJobNumber(&model.Build{ID: 1}, 1)
|
|
g.Assert(err2 == nil).IsTrue()
|
|
g.Assert(getjob.ID).Equal(job.ID)
|
|
g.Assert(getjob.Status).Equal(job.Status)
|
|
})
|
|
|
|
g.It("Should Get a List of Jobs by Commit", func() {
|
|
|
|
build := model.Build{
|
|
RepoID: 1,
|
|
Status: model.StatusSuccess,
|
|
}
|
|
jobs := []*model.Job{
|
|
{
|
|
BuildID: 1,
|
|
Status: "success",
|
|
ExitCode: 0,
|
|
Number: 1,
|
|
},
|
|
{
|
|
BuildID: 3,
|
|
Status: "error",
|
|
ExitCode: 1,
|
|
Number: 2,
|
|
},
|
|
{
|
|
BuildID: 5,
|
|
Status: "pending",
|
|
ExitCode: 0,
|
|
Number: 3,
|
|
},
|
|
}
|
|
|
|
err1 := s.CreateBuild(&build, jobs...)
|
|
g.Assert(err1 == nil).IsTrue()
|
|
getjobs, err2 := s.GetJobList(&build)
|
|
g.Assert(err2 == nil).IsTrue()
|
|
g.Assert(len(getjobs)).Equal(3)
|
|
g.Assert(getjobs[0].Number).Equal(1)
|
|
g.Assert(getjobs[0].Status).Equal(model.StatusSuccess)
|
|
})
|
|
})
|
|
}
|