Finishing the test/edit route

web/forms/test.go

@@ -4,5 +4,15 @@ type TestForm struct {
   Name        string    `form:"test_name" validate:"required"`
   Description string    `form:"test_description"`
   Hardware    int       `form:"test_hardware" validate:"required"`
-  Benchmarks  []string  `form:"test_benchmarks" validate:"required"`
+  Benchmarks  []uint  `form:"test_benchmarks" validate:"required"`
+}
+
+func (t *TestForm) IsBenchmarkSelected(checkID uint) bool {
+  for _, selectedID := range t.Benchmarks {
+    if checkID == selectedID {
+      return true
+    }
+  }
+
+  return false
 }

web/routes.go

@@ -54,6 +54,7 @@ func RegisterRoutes(f *flamego.Flame) {
     f.Group("/{test_id}", func() {
       f.Get("", routes.TestGetView)
       f.Get("/edit", routes.TestGetEdit)
+      f.Post("/edit", binding.Form(forms.TestForm{}), routes.TestPostEdit)
     })
   })
 

web/routes/test.go

@@ -141,7 +141,11 @@ func TestPostEdit(c flamego.Context, form forms.TestForm, errs binding.Errors) {
 
   // removed associated benchmarks that weren't in the form
   for _, b := range test.Benchmarks {
-
+    if ! form.IsBenchmarkSelected(b.ID) {
+      var benchmark models.Benchmark
+      models.DB.First(&benchmark, b) // find benchmark
+      models.DB.Model(&test).Association("Benchmarks").Delete(&benchmark)
+    }
   }
 
   c.Redirect(fmt.Sprintf("/test/%d", test.ID))