go-mockid/vendor/github.com/mailgun/mailgun-go/v3/mock_exports.go

49 lines
1.1 KiB
Go

package mailgun
import (
"net/http"
"strconv"
"github.com/go-chi/chi"
)
func (ms *MockServer) addExportRoutes(r chi.Router) {
r.Post("/exports", ms.postExports)
r.Get("/exports", ms.listExports)
r.Get("/exports/{id}", ms.getExport)
r.Get("/exports/{id}/download_url", ms.getExportLink)
}
func (ms *MockServer) postExports(w http.ResponseWriter, r *http.Request) {
e := Export{
ID: strconv.Itoa(len(ms.exportList)),
URL: r.FormValue("url"),
Status: "complete",
}
ms.exportList = append(ms.exportList, e)
toJSON(w, okResp{Message: "success"})
}
func (ms *MockServer) listExports(w http.ResponseWriter, _ *http.Request) {
toJSON(w, ExportList{
Items: ms.exportList,
})
}
func (ms *MockServer) getExport(w http.ResponseWriter, r *http.Request) {
for _, export := range ms.exportList {
if export.ID == chi.URLParam(r, "id") {
toJSON(w, export)
return
}
}
w.WriteHeader(http.StatusNotFound)
toJSON(w, okResp{Message: "export not found"})
}
func (ms *MockServer) getExportLink(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Location", "/some/s3/url")
w.WriteHeader(http.StatusFound)
}