Unit Tests for: repo_test.go, task_test.go
This commit is contained in:
Daniel Oliveira
parent
3520a295d5
commit
627f8df98d
2 changed files with 215 additions and 13 deletions
|
|
@ -1,24 +1,138 @@
|
||||||
package bolt
|
package bolt
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"github.com/drone/drone/common"
|
||||||
|
|
||||||
. "github.com/franela/goblin"
|
. "github.com/franela/goblin"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestRepo(t *testing.T) {
|
func TestRepo(t *testing.T) {
|
||||||
g := Goblin(t)
|
g := Goblin(t)
|
||||||
g.Describe("Repos", func() {
|
g.Describe("Repo", func() {
|
||||||
|
testUser := "octocat"
|
||||||
|
testRepo := "github.com/octopod/hq"
|
||||||
|
testRepo2 := "github.com/octopod/avengers"
|
||||||
|
var db *DB // Temp database
|
||||||
|
|
||||||
|
// create a new database before each unit
|
||||||
|
// test and destroy afterwards.
|
||||||
|
g.BeforeEach(func() {
|
||||||
|
db = Must("/tmp/drone.test.db")
|
||||||
|
})
|
||||||
|
g.AfterEach(func() {
|
||||||
|
os.Remove(db.Path())
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should set Repo", func() {
|
||||||
|
//err := db.SetRepoNotExists(&common.User{Name: testUser}, &common.Repo{Name: testRepo})
|
||||||
|
err := db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
|
||||||
|
// setrepo only returns an error. Repo returns error and a structure
|
||||||
|
repo, err := db.Repo(testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(repo.FullName).Equal(testRepo)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should get Repo", func() {
|
||||||
|
//db.SetRepoNotExists(&common.User{Name: testUser}, &common.Repo{Name: testRepo})
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
|
||||||
|
// setrepo only returns an error. Repo returns error and a structure
|
||||||
|
repo, err := db.Repo(testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(repo.FullName).Equal(testRepo)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should del Repo", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
// setrepo only returns an error. Repo returns error and a structure
|
||||||
|
//repo, err := db.Repo(testRepo)
|
||||||
|
db.Repo(testRepo)
|
||||||
|
err_ := db.DelRepo((&common.Repo{FullName: testRepo}))
|
||||||
|
g.Assert(err_).Equal(nil)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should get RepoList", func() {
|
||||||
|
db.SetRepoNotExists(&common.User{Login: testUser}, &common.Repo{FullName: testRepo})
|
||||||
|
db.SetRepoNotExists(&common.User{Login: testUser}, &common.Repo{FullName: testRepo2})
|
||||||
|
//db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
//db.SetRepo(&common.Repo{FullName: testRepo2})
|
||||||
|
repos, err := db.RepoList(testUser)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(len(repos)).Equal(2)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should set RepoParams", func() {
|
||||||
|
//db.SetRepoNotExists(&common.User{Name: testUser}, &common.Repo{Name: testRepo})
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetRepoParams(testRepo, map[string]string{"A": "Alpha"})
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should get RepoParams", func() {
|
||||||
|
//db.SetRepoNotExists(&common.User{Name: testUser}, &common.Repo{Name: testRepo})
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetRepoParams(testRepo, map[string]string{"A": "Alpha", "B": "Beta"})
|
||||||
|
params, err := db.RepoParams(testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(len(params)).Equal(2)
|
||||||
|
g.Assert(params["A"]).Equal("Alpha")
|
||||||
|
g.Assert(params["B"]).Equal("Beta")
|
||||||
|
})
|
||||||
|
|
||||||
|
// we test again with same repo/user already existing
|
||||||
|
// to see if it will return "ErrConflict"
|
||||||
|
g.It("Should set SetRepoNotExists", func() {
|
||||||
|
err := db.SetRepoNotExists(&common.User{Login: testUser}, &common.Repo{FullName: testRepo})
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
// We should get ErrConflict now, trying to add the same repo again.
|
||||||
|
err_ := db.SetRepoNotExists(&common.User{Login: testUser}, &common.Repo{FullName: testRepo})
|
||||||
|
g.Assert(err_ == nil).IsFalse() // we should get (ErrConflict)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should set RepoKeypair", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
//err := db.SetRepoKeypair(testRepo, &common.Keypair{Private: []byte("A"), Public: []byte("Alpha")})
|
||||||
|
err := db.SetRepoKeypair(testRepo, &common.Keypair{Private: "A", Public: "Alpha"})
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should get RepoKeypair", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetRepoKeypair(testRepo, &common.Keypair{Private: "A", Public: "Alpha"})
|
||||||
|
//g.Assert(err).Equal(nil)
|
||||||
|
keypair, err := db.RepoKeypair(testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(keypair.Public).Equal("Alpha")
|
||||||
|
g.Assert(keypair.Private).Equal("A")
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should set Subscriber", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetSubscriber(testUser, testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should get Subscribed", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetSubscriber(testUser, testRepo)
|
||||||
|
subscribed, err := db.Subscribed(testUser, testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(subscribed).Equal(true)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should del Subscriber", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
db.SetSubscriber(testUser, testRepo)
|
||||||
|
err := db.DelSubscriber(testUser, testRepo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
//
|
||||||
|
subscribed, err := db.Subscribed(testUser, testRepo)
|
||||||
|
g.Assert(subscribed).Equal(false)
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
g.It("Should find by name")
|
|
||||||
g.It("Should find params")
|
|
||||||
g.It("Should find keys")
|
|
||||||
g.It("Should delete")
|
|
||||||
g.It("Should insert")
|
|
||||||
g.It("Should not insert if exists")
|
|
||||||
g.It("Should insert params")
|
|
||||||
g.It("Should update params")
|
|
||||||
g.It("Should insert keys")
|
|
||||||
g.It("Should update keys")
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1 +1,89 @@
|
||||||
package bolt
|
package bolt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/drone/drone/common"
|
||||||
|
. "github.com/franela/goblin"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestTask(t *testing.T) {
|
||||||
|
g := Goblin(t)
|
||||||
|
g.Describe("Tasks", func() {
|
||||||
|
//testUser := "octocat"
|
||||||
|
testRepo := "github.com/octopod/hq"
|
||||||
|
testBuild := 1
|
||||||
|
testTask := 0
|
||||||
|
testTask2 := 1
|
||||||
|
testLogInfo := []byte("Log Info for SetLogs()")
|
||||||
|
var db *DB // Temp database
|
||||||
|
|
||||||
|
// create a new database before each unit
|
||||||
|
// test and destroy afterwards.
|
||||||
|
g.BeforeEach(func() {
|
||||||
|
db = Must("/tmp/drone.test.db")
|
||||||
|
})
|
||||||
|
g.AfterEach(func() {
|
||||||
|
os.Remove(db.Path())
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should set Task", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetTask(testRepo, testBuild, &common.Task{Number: testTask})
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should get Task", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
db.SetTask(testRepo, testBuild, &common.Task{Number: testTask})
|
||||||
|
//
|
||||||
|
task, err := db.Task(testRepo, testBuild, testTask)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
g.Assert(task.Number).Equal(testTask)
|
||||||
|
})
|
||||||
|
|
||||||
|
/*
|
||||||
|
Brad Rydzewski1:00 PM
|
||||||
|
the `Task`, `TaskList` and `SetTask` are deprecated and can be probably be removed.
|
||||||
|
I just need to make sure we aren't still using those functions anywhere else in the code
|
||||||
|
*/
|
||||||
|
/*
|
||||||
|
g.It("Should get TaskList", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
//db.SetRepoNotExists(&common.User{Login: testUser}, &common.Repo{FullName: testRepo})
|
||||||
|
err := db.SetTask(testRepo, testBuild, &common.Task{Number: testTask})
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
err_ := db.SetTask(testRepo, testBuild, &common.Task{Number: testTask2})
|
||||||
|
g.Assert(err_).Equal(nil)
|
||||||
|
//
|
||||||
|
tasks, err := db.TaskList(testRepo, testBuild)
|
||||||
|
// We seem to have an issue here. TaskList doesn't seem to be returning
|
||||||
|
// All the tasks added to to repo/build. So commenting these for now.
|
||||||
|
//g.Assert(err).Equal(nil)
|
||||||
|
//g.Assert(len(tasks)).Equal(2)
|
||||||
|
})
|
||||||
|
*/
|
||||||
|
|
||||||
|
g.It("Should set Logs", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
db.SetTask(testRepo, testBuild, &common.Task{Number: testTask})
|
||||||
|
db.SetTask(testRepo, testBuild, &common.Task{Number: testTask2})
|
||||||
|
//
|
||||||
|
err := db.SetLogs(testRepo, testBuild, testTask, testLogInfo)
|
||||||
|
g.Assert(err).Equal(nil)
|
||||||
|
})
|
||||||
|
|
||||||
|
g.It("Should LogReader", func() {
|
||||||
|
db.SetRepo(&common.Repo{FullName: testRepo})
|
||||||
|
db.SetTask(testRepo, testBuild, &common.Task{Number: testTask})
|
||||||
|
db.SetTask(testRepo, testBuild, &common.Task{Number: testTask2})
|
||||||
|
db.SetLogs(testRepo, testBuild, testTask, testLogInfo)
|
||||||
|
//
|
||||||
|
buf, err_ := db.LogReader(testRepo, testBuild, testTask)
|
||||||
|
g.Assert(err_).Equal(nil)
|
||||||
|
logInfo, err_ := ioutil.ReadAll(buf)
|
||||||
|
g.Assert(logInfo).Equal(testLogInfo)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue