func before() {
expectedProbe = models.NewProbe("Is this test ok ?", []string{})
var err error
session, err = mgo.Dial("localhost")
if err != nil {
panic(err)
}
collection = session.DB("test").C("probe")
collection.RemoveAll(bson.M{})
}
func TestGetPollsTemplate(testing *testing.T) {
templatePath = "../static/polls.html"
probe := models.NewProbe("Is this test ok ?", []string{"Oui", "Non"})
probe.Save()
writer := httptest.NewRecorder()
PollTemplateHandler(writer, nil)
assert.Contains(testing, writer.Body.String(), "Is this test ok ?")
assert.Contains(testing, writer.Body.String(), "Oui")
assert.Contains(testing, writer.Body.String(), "Non")
}
func TestRespondProbe(testing *testing.T) {
collection.RemoveAll(bson.M{})
probe := models.NewProbe("Aimez-vous golang ?", []string{})
probe.Id = bson.NewObjectId()
collection.Insert(probe)
request := createRequest(models.ProbeResponse{probe.Id, "Oui"})
RespondProbe(nil, request)
actualProbe := findOneProbeAndRemoveIt()
assert.Equal(testing, 1, len(actualProbe.Responses))
assert.Equal(testing, "Oui", actualProbe.Responses[0])
}
