Need to review tests in data api

internal/server/data/server.go

@@ -1,9 +1,9 @@
 package data
 
 import (
-	"log"
 	"net/http"
 
+	"github.com/BBVA/kapow/internal/server/model"
 	"github.com/gorilla/mux"
 )
 
@@ -14,21 +14,34 @@ import (
 func configRouter() *mux.Router {
 	r := mux.NewRouter()
 
-	r.HandleFunc("/handlers/{handler_id}/{root}/{resource:.*$}", readResource).Methods("GET")
-	r.HandleFunc("/handlers/{handler_id}/{root}/{resource:.*$}", updateResource).Methods("PUT")
+	r.HandleFunc("/handlers/{handler_id}/response/headers/", updateResource).Methods("PUT")
+	r.HandleFunc("/handlers/{handler_id}/response/headers/{key}", updateResource).Methods("PUT")
 	return r
 }
 
-func Run(bindAddr string) {
-	r := configRouter()
+var getHandlerId func(string) (*model.Handler, bool) = Handlers.Get
 
-	log.Fatal(http.ListenAndServe(bindAddr, r))
-}
+//func Run(bindAddr string) {
+//	r := configRouter()
+//
+//	log.Fatal(http.ListenAndServe(bindAddr, r))
+//}
 
-func readResource(res http.ResponseWriter, req *http.Request) {
-
-}
+//func readResource(res http.ResponseWriter, req *http.Request) {
+//
+//}
 
 func updateResource(res http.ResponseWriter, req *http.Request) {
+	vars := mux.Vars(req)
+	hID := vars["handler_id"]
 
+	if _, ok := getHandlerId(hID); !ok {
+		res.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	if _, ok := vars["key"]; !ok {
+		res.WriteHeader(http.StatusBadRequest)
+		return
+	}
 }