Commit c1b8214

Richard Luby <richluby@gmail.com>
2017-01-26 07:37:50
fixed db test
1 parent b946aea
Changed files (2)
db.go
@@ -49,7 +49,7 @@ func returnForQuery(selectedCategories []string, numQuestions int) ([]Record, er
 	var collector []Record
 	var err error
 	for _, category := range selectedCategories {
-		err = collection.Find(bson.M{"categorystring": category}).All(&collector)
+		err = collection.Find(bson.M{"path": category}).All(&collector)
 		if err != nil {
 			return nil, err
 		} else {
db_test.go
@@ -20,19 +20,20 @@ func initDBForTest(t *testing.T) *mgo.Session {
 }
 
 // Test_AddAndRetriveForDB
-func Test_AddAndRetriveForDB(t *testing.T) {
+func TestAddAndRetriveForDB(t *testing.T) {
 	t.SkipNow()
 	DBsession = initDBForTest(t)
 	var recordTest = Record{Question: "Question",
 		Reference: "Reference",
 		Answer:    "Answer",
-		Path:      ""}
+		Path:      "linux:file:thing"}
 	if err := addRecordToDB(&recordTest); err != nil {
 		if !strings.Contains(err.Error(), "duplicate key error") {
 			t.Errorf("Error for adding record: %+v", err)
+			t.FailNow()
 		}
 	}
-	results, err := returnForQuery([]string{"CategoryString"}, 1)
+	results, err := returnForQuery([]string{"linux:file:thing"}, 1)
 	if err != nil {
 		t.Errorf("Error for returning initial query: %+v", err)
 	}