// RemoveFlag removes a given flag from a FormSubmission based on the provided // route params. // 200 Success, 400 Bad Request, 404 Not Found, 500 Internal func (formSubmissionHandle) RemoveFlag(c *web.Context) error { id := c.Params["id"] flag := c.Params["flag"] s, err := submission.RemoveFlag(c.SessionID, c.Ctx["DB"].(*db.DB), id, flag) if err != nil { return err } c.Respond(s, http.StatusOK) return nil }
func Test_Flags(t *testing.T) { subs, db := setup(t, "submission") defer teardown(t, db) t.Log("Given the need to add and remove flags.") { t.Log("\tWhen starting from an empty submissions collection") { //---------------------------------------------------------------------- // Create the submission. if err := submission.Create(tests.Context, db, subs[0].FormID.Hex(), &subs[0]); err != nil { t.Fatalf("\t%s\tShould be able to create a submission : %s", tests.Failed, err) } t.Logf("\t%s\tShould be able to create a submission.", tests.Success) //---------------------------------------------------------------------- // Ensure the submission has been added to the database. sub, err := submission.Retrieve(tests.Context, db, subs[0].ID.Hex()) if err != nil { t.Fatalf("\t%s\tShould be able to retrieve the submission : %s", tests.Failed, err.Error()) } t.Logf("\t%s\tShould be able to retrieve the submission", tests.Success) newFlag := time.Now().String() //---------------------------------------------------------------------- // Ensure that the new flag is not already on the submission. // Ensure that the new flag is not in the current flags. for _, flag := range sub.Flags { if flag == newFlag { t.Fatalf("\t%s\tShould not have test flag already : Flag already exists", tests.Failed) } } t.Logf("\t%s\tShould not have test flag already.", tests.Success) //---------------------------------------------------------------------- // Add the flag to the submission. nsub, err := submission.AddFlag(tests.Context, db, sub.ID.Hex(), newFlag) if err != nil { t.Fatalf("\t%s\tShould be able to add the flag : %s", tests.Failed, err.Error()) } t.Logf("\t%s\tShould be able to add the flag", tests.Success) //---------------------------------------------------------------------- // Ensure that the new flag was added to the submission returned. var found bool for _, flag := range nsub.Flags { if flag == newFlag { found = true break } } if !found { t.Fatalf("\t%s\tShould have test flag on returned object : Flag does not exist", tests.Failed) } t.Logf("\t%s\tShould have test flag on returned object.", tests.Success) //---------------------------------------------------------------------- // Ensure that the new flag was added to the store's submission. rsub, err := submission.Retrieve(tests.Context, db, sub.ID.Hex()) if err != nil { t.Fatalf("\t%s\tShould be able to retrieve the submission : %s", tests.Failed, err.Error()) } t.Logf("\t%s\tShould be able to retrieve the submission", tests.Success) found = false for _, flag := range rsub.Flags { if flag == newFlag { found = true break } } if !found { t.Fatalf("\t%s\tShould have test flag on database object : Flag does not exist", tests.Failed) } t.Logf("\t%s\tShould have test flag on database object.", tests.Success) //---------------------------------------------------------------------- // Remove the new flag from the submission. nsub, err = submission.RemoveFlag(tests.Context, db, sub.ID.Hex(), newFlag) if err != nil { t.Fatalf("\t%s\tShould be able to remove the flag : %s", tests.Failed, err.Error()) } t.Logf("\t%s\tShould be able to remove the flag", tests.Success) //---------------------------------------------------------------------- // Ensure that the new flag was removed from the submission returned. found = false for _, flag := range nsub.Flags { if flag == newFlag { found = true break } } if found { t.Fatalf("\t%s\tShould not have test flag on returned object : Flag found", tests.Failed) } t.Logf("\t%s\tShould not have test flag on returned object.", tests.Success) //---------------------------------------------------------------------- // Ensure that the new flag was removed from the submission in the store. rsub, err = submission.Retrieve(tests.Context, db, sub.ID.Hex()) if err != nil { t.Fatalf("\t%s\tShould be able to retrieve the submission : %s", tests.Failed, err.Error()) } t.Logf("\t%s\tShould be able to retrieve the submission", tests.Success) found = false for _, flag := range rsub.Flags { if flag == newFlag { found = true break } } if found { t.Fatalf("\t%s\tShould not have test flag on database object : Flag found", tests.Failed) } t.Logf("\t%s\tShould not have test flag on database object.", tests.Success) } } }