(db dbAdapter)
| 21 | } |
| 22 | |
| 23 | func DBAdapterTest(db dbAdapter) { |
| 24 | var err error |
| 25 | testWorkerIDs := []string{"test_worker1", "test_worker2"} |
| 26 | Convey("create worker", func() { |
| 27 | for _, id := range testWorkerIDs { |
| 28 | w := WorkerStatus{ |
| 29 | ID: id, |
| 30 | Token: "token_" + id, |
| 31 | LastOnline: time.Now(), |
| 32 | LastRegister: time.Now(), |
| 33 | } |
| 34 | _, err = db.CreateWorker(w) |
| 35 | So(err, ShouldBeNil) |
| 36 | } |
| 37 | |
| 38 | Convey("get existent worker", func() { |
| 39 | _, err := db.GetWorker(testWorkerIDs[0]) |
| 40 | So(err, ShouldBeNil) |
| 41 | }) |
| 42 | |
| 43 | Convey("list existent workers", func() { |
| 44 | ws, err := db.ListWorkers() |
| 45 | So(err, ShouldBeNil) |
| 46 | So(len(ws), ShouldEqual, 2) |
| 47 | }) |
| 48 | |
| 49 | Convey("get non-existent worker", func() { |
| 50 | _, err := db.GetWorker("invalid workerID") |
| 51 | So(err, ShouldNotBeNil) |
| 52 | }) |
| 53 | |
| 54 | Convey("delete existent worker", func() { |
| 55 | err := db.DeleteWorker(testWorkerIDs[0]) |
| 56 | So(err, ShouldBeNil) |
| 57 | _, err = db.GetWorker(testWorkerIDs[0]) |
| 58 | So(err, ShouldNotBeNil) |
| 59 | ws, err := db.ListWorkers() |
| 60 | So(err, ShouldBeNil) |
| 61 | So(len(ws), ShouldEqual, 1) |
| 62 | }) |
| 63 | |
| 64 | Convey("delete non-existent worker", func() { |
| 65 | err := db.DeleteWorker("invalid workerID") |
| 66 | So(err, ShouldNotBeNil) |
| 67 | ws, err := db.ListWorkers() |
| 68 | So(err, ShouldBeNil) |
| 69 | So(len(ws), ShouldEqual, 2) |
| 70 | }) |
| 71 | }) |
| 72 | |
| 73 | Convey("update mirror status", func() { |
| 74 | status := []MirrorStatus{ |
| 75 | { |
| 76 | Name: "arch-sync1", |
| 77 | Worker: testWorkerIDs[0], |
| 78 | IsMaster: true, |
| 79 | Status: Success, |
| 80 | LastUpdate: time.Now(), |
no test coverage detected