woodpecker/pkg/store/builtin/job_test.go

package builtin

import (
	"testing"

	"github.com/drone/drone/Godeps/_workspace/src/github.com/franela/goblin"
	"github.com/drone/drone/pkg/types"
)

func TestBuildstore(t *testing.T) {
	db := mustConnectTest()
	bs := NewJobstore(db)
	cs := NewBuildstore(db)
	defer db.Close()

	g := goblin.Goblin(t)
	g.Describe("Jobstore", 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 := &types.Job{
				BuildID:  1,
				Status:   "pending",
				ExitCode: 0,
				Number:   1,
			}
			err1 := bs.AddJob(job)
			g.Assert(err1 == nil).IsTrue()
			g.Assert(job.ID != 0).IsTrue()

			job.Status = "started"
			err2 := bs.SetJob(job)
			g.Assert(err2 == nil).IsTrue()

			getjob, err3 := bs.Job(job.ID)
			g.Assert(err3 == nil).IsTrue()
			g.Assert(getjob.Status).Equal(job.Status)
		})

		g.It("Should Get a Job by ID", func() {
			job := &types.Job{
				BuildID:     1,
				Status:      "pending",
				ExitCode:    1,
				Number:      1,
				Environment: map[string]string{"foo": "bar"},
			}
			err1 := bs.AddJob(job)
			g.Assert(err1 == nil).IsTrue()
			g.Assert(job.ID != 0).IsTrue()

			getjob, err2 := bs.Job(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 := &types.Job{
				BuildID:  1,
				Status:   "pending",
				ExitCode: 1,
				Number:   1,
			}
			err1 := bs.AddJob(job)
			g.Assert(err1 == nil).IsTrue()
			g.Assert(job.ID != 0).IsTrue()

			getjob, err2 := bs.JobNumber(&types.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 := types.Build{
				RepoID: 1,
				Status: types.StateSuccess,
				Jobs: []*types.Job{
					&types.Job{
						BuildID:  1,
						Status:   "success",
						ExitCode: 0,
						Number:   1,
					},
					&types.Job{
						BuildID:  3,
						Status:   "error",
						ExitCode: 1,
						Number:   2,
					},
					&types.Job{
						BuildID:  5,
						Status:   "pending",
						ExitCode: 0,
						Number:   3,
					},
				},
			}
			//
			err1 := cs.AddBuild(&build)
			g.Assert(err1 == nil).IsTrue()
			getjobs, err2 := bs.JobList(&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(types.StateSuccess)
		})
	})
}