mdiluz/rove
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Start to implement proper validation of HTTP interface

Browse source
This commit is contained in:
Marc Di Luzio 2020-06-13 12:35:37 +01:00
parent 42ee69b1a2
commit ba52458fd6
3 changed files with 32 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

40
cmd/rove-reverse-proxy/http_test.go
View file

@ -7,42 +7,23 @@ import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"net/url"
"os"
"testing"
"github.com/mdiluz/rove/pkg/rove"
)
// Server is a simple wrapper to a server path
type Server string
// Get performs a Get request
func (s Server) Get(path string, out interface{}) error {
// Request performs a HTTP
func (s Server) Request(method, path string, in, out interface{}) error {
u := url.URL{
Scheme: "http",
Host: string(s),
Path: path,
}
if resp, err := http.Get(u.String()); err != nil {
return err
} else if resp.StatusCode != http.StatusOK {
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("failed to read response body to code %d", resp.StatusCode)
}
return fmt.Errorf("http returned status %d: %s", resp.StatusCode, string(body))
} else {
return json.NewDecoder(resp.Body).Decode(out)
}
}
// Post performs a Post request
func (s Server) Post(path string, in, out interface{}) error {
u := url.URL{
Scheme: "http",
Host: string(s),
Host: fmt.Sprintf("%s:8080", string(s)),
Path: path,
}
client := &http.Client{}
@ -54,7 +35,7 @@ func (s Server) Post(path string, in, out interface{}) error {
}
// Set up the request
req, err := http.NewRequest("POST", u.String(), bytes.NewReader(marshalled))
req, err := http.NewRequest(method, u.String(), bytes.NewReader(marshalled))
if err != nil {
return err
}
@ -76,9 +57,14 @@ func (s Server) Post(path string, in, out interface{}) error {
}
}
var serv = Server(os.Getenv("ROVE_GRPC"))
var serv = Server(os.Getenv("ROVE_HTTP"))
func TestServer_Status(t *testing.T) {
req := &rove.StatusRequest{}
resp := &rove.StatusResponse{}
if err := serv.Request("GET", "status", req, resp); err != nil {
log.Fatal(err)
}
}
func TestServer_Register(t *testing.T) {