test server put, tree, get

spike/review/reinvent/ezmded/server/server.go

@@ -171,6 +171,19 @@ func (server *Server) apiV0FilesHandler(w http.ResponseWriter, r *http.Request)
 	if err := server.postContentHandler(filePath, w, r); err != nil {
 		return err
 	}
+	tree := server.tree()
+	if pid := r.Header.Get("PID"); pid == "" {
+	} else if branches, err := tree.Get(); err != nil {
+		return err
+	} else if _, ok := branches[pid]; !ok {
+		return errors.New("bad pid")
+	}
+	if err := tree.Put(id, Branch{
+		Title: r.Header.Get("Title"),
+		PID:   r.Header.Get("PID"),
+	}); err != nil {
+		return err
+	}
 	return json.NewEncoder(w).Encode(map[string]map[string]string{
 		"data": map[string]string{
 			"filePath": path.Join("/api/v0/files", id),
@@ -207,6 +220,9 @@ func (server *Server) apiV0FilesIDPutHandler(w http.ResponseWriter, r *http.Requ
 	if title := r.Header.Get("Title"); title != "" {
 		branch.Title = title
 	}
+	if pid := r.Header.Get("PID"); pid != "" {
+		branch.PID = pid
+	}
 	branch.Deleted = false
 	if err := server.putContentHandler(server.diskFilePath(id), w, r); err != nil {
 		return err
@@ -223,6 +239,7 @@ func (server *Server) apiV0FilesIDGetHandler(w http.ResponseWriter, r *http.Requ
 	id := path.Base(r.URL.Path)
 	branch, _ := branches[id]
 	w.Header().Set("Title", branch.Title)
+	w.Header().Set("PID", branch.PID)
 	return server.getContentHandler(server.diskFilePath(id), w, r)
 }