diff --git a/datastore/bolt/repo_test.go b/datastore/bolt/repo_test.go index e38685dfd..ab45babc7 100644 --- a/datastore/bolt/repo_test.go +++ b/datastore/bolt/repo_test.go @@ -25,21 +25,17 @@ func TestRepo(t *testing.T) { }) 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) @@ -47,8 +43,7 @@ func TestRepo(t *testing.T) { 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) @@ -57,22 +52,19 @@ func TestRepo(t *testing.T) { 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) @@ -89,12 +81,12 @@ func TestRepo(t *testing.T) { 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.Assert(err_).Equal(ErrKeyExists) }) 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) }) @@ -102,7 +94,7 @@ func TestRepo(t *testing.T) { 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") @@ -128,7 +120,7 @@ func TestRepo(t *testing.T) { 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) diff --git a/datastore/bolt/task.go b/datastore/bolt/task.go index 637161dd4..ff09fcf13 100644 --- a/datastore/bolt/task.go +++ b/datastore/bolt/task.go @@ -2,13 +2,18 @@ package bolt import ( "bytes" + "github.com/boltdb/bolt" + //"github.com/drone/drone/common" "io" "strconv" - - "github.com/boltdb/bolt" - "github.com/drone/drone/common" ) +/* + 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 +*/ +/* // GetTask gets the task at index N for the named // repository and build number. func (db *DB) Task(repo string, build int, task int) (*common.Task, error) { @@ -63,6 +68,7 @@ func (db *DB) SetTask(repo string, build int, task *common.Task) error { return update(t, bucketBuildTasks, key, task) }) } +*/ // SetLogs inserts or updates a task logs for the // named repository and build number. diff --git a/datastore/bolt/task_test.go b/datastore/bolt/task_test.go index d2997819f..573c75570 100644 --- a/datastore/bolt/task_test.go +++ b/datastore/bolt/task_test.go @@ -15,7 +15,7 @@ func TestTask(t *testing.T) { testRepo := "github.com/octopod/hq" testBuild := 1 testTask := 0 - testTask2 := 1 + //testTask2 := 1 testLogInfo := []byte("Log Info for SetLogs()") var db *DB // Temp database @@ -28,21 +28,6 @@ func TestTask(t *testing.T) { 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. @@ -63,12 +48,27 @@ func TestTask(t *testing.T) { //g.Assert(err).Equal(nil) //g.Assert(len(tasks)).Equal(2) }) + + 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) + }) */ 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}) + //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) @@ -76,8 +76,8 @@ func TestTask(t *testing.T) { 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.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) diff --git a/datastore/datastore.go b/datastore/datastore.go index 3bab2c5dc..19631da0f 100644 --- a/datastore/datastore.go +++ b/datastore/datastore.go @@ -121,15 +121,15 @@ type Datastore interface { // GetTask gets the task at index N for the named // repository and build number. - Task(string, int, int) (*common.Task, error) + //Task(string, int, int) (*common.Task, error) // TaskList gets all tasks for the named repository // and build number. - TaskList(string, int) ([]*common.Task, error) + //TaskList(string, int) ([]*common.Task, error) // SetTask inserts or updates a task for the named // repository and build number. - SetTask(string, int, *common.Task) error + //SetTask(string, int, *common.Task) error // LogReader gets the task logs at index N for // the named repository and build number.