add more tests for data.go
diff --git a/data_test.go b/data_test.go index 8a26c0d..d678828 100644 --- a/data_test.go +++ b/data_test.go
@@ -48,7 +48,6 @@ testDbMan.setDbVersion("test" + strconv.Itoa(testCount)) initTestDb(testDbMan.getDb()) testDbMan.initDb() - insertTestAvailableBlobTable(testDbMan.getDb()) time.Sleep(100 * time.Millisecond) }) @@ -69,7 +68,7 @@ for rows.Next() { rows.Scan(&count) } - Expect(count).Should(Equal(1)) + Expect(count).Should(Equal(0)) // metadata_runtime_entity_metadata rows, err = testDbMan.getDb().Query(` @@ -83,8 +82,38 @@ Expect(count).Should(Equal(3)) }) + It("should succefully update local FS location", func() { + + err := testDbMan.updateLocalFsLocation(readyBlobId, readyblobLocalFs) + Expect(err).Should(Succeed()) + // edgex_blob_available + rows, err := testDbMan.getDb().Query(` + SELECT count(*) from edgex_blob_available; + `) + Expect(err).Should(Succeed()) + defer rows.Close() + var count int + for rows.Next() { + rows.Scan(&count) + } + Expect(count).Should(Equal(1)) + }) + + It("should succefully get local FS location", func() { + + err := testDbMan.updateLocalFsLocation(readyBlobId, readyblobLocalFs) + Expect(err).Should(Succeed()) + + // edgex_blob_available + location, err := testDbMan.getLocalFSLocation(readyBlobId) + Expect(err).Should(Succeed()) + Expect(location).Should(Equal(readyblobLocalFs)) + }) + It("should succefully get ready deployments", func() { + err := testDbMan.updateLocalFsLocation(readyBlobId, readyblobLocalFs) + Expect(err).Should(Succeed()) deps, err := testDbMan.getReadyDeployments() Expect(err).Should(Succeed()) Expect(len(deps)).Should(Equal(1)) @@ -94,6 +123,8 @@ It("should succefully get unready blob ids", func() { + err := testDbMan.updateLocalFsLocation(readyBlobId, readyblobLocalFs) + Expect(err).Should(Succeed()) ids, err := testDbMan.getUnreadyBlobs() Expect(err).Should(Succeed()) Expect(len(ids)).Should(Equal(2)) @@ -188,12 +219,3 @@ `) Expect(err).Should(Succeed()) } - -func insertTestAvailableBlobTable(db apid.DB) { - stmt, err := db.Prepare(` - INSERT INTO "edgex_blob_available" VALUES(?, ?); - `) - Expect(err).Should(Succeed()) - _, err = stmt.Exec(readyBlobId, readyblobLocalFs) - Expect(err).Should(Succeed()) -}