A simple HTTP server to mock responses

mockserver.go

package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"
)

func initFlags(addr *string, port *int, retCode *int, body *string) {
	flag.StringVar(addr, "addr", "0.0.0.0", "addr where http server will listen")
	flag.IntVar(port, "port", 9876, "port where http server will listen")
	flag.IntVar(retCode, "return-code", http.StatusOK, "return code that will be on the response of every request")
	flag.StringVar(body, "body", "mocked response", "return body that will be on the response of every request")
	flag.Parse()
}

func mightyHandler(retCode int, retBody string) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(retCode)
		w.Write([]byte(retBody))
		log.Printf("Request to '%s' with method '%s' received", r.URL.Path, r.Method)
	}
}

func main() {
	var addr, rbody string
	var port, rcode int

	initFlags(&addr, &port, &rcode, &rbody)
	log.Printf("Starting mock HTTP server with %d return code", rcode)

	laddr := fmt.Sprintf("%s:%d", addr, port)
	log.Printf("Listening on %s", laddr)

	if err := http.ListenAndServe(laddr, mightyHandler(rcode, rbody)); err != nil {
		log.Fatal(err)
	}
}