2015-10-21 23:14:02 +00:00
|
|
|
package datastore
|
2015-09-30 01:21:17 +00:00
|
|
|
|
2017-04-02 14:13:26 +00:00
|
|
|
//
|
|
|
|
|
// 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)
|
|
|
|
|
// })
|
|
|
|
|
// })
|
|
|
|
|
// }
|
