Code imported from @CesarGallego private repo
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func getReason(r *http.Response) string {
|
||||
return strings.Join(strings.Split(r.Status, " ")[1:], " ")
|
||||
}
|
||||
|
||||
//GetData will perform the request and write the results on the provided writer
|
||||
func GetData(host, id, path string, wr io.Writer) error {
|
||||
url := host + "/handlers/" + id + path
|
||||
|
||||
res, err := http.Get(url)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer res.Body.Close()
|
||||
|
||||
if res.StatusCode < 200 || res.StatusCode >= 300 {
|
||||
return errors.New(getReason(res))
|
||||
}
|
||||
|
||||
_, err = io.Copy(wr, res.Body)
|
||||
return err
|
||||
}
|
||||
@@ -0,0 +1,66 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net/http"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
gock "gopkg.in/h2non/gock.v1"
|
||||
)
|
||||
|
||||
func TestGetInvalidUrl(t *testing.T) {
|
||||
err := GetData("", "", "", os.Stdout)
|
||||
if err == nil {
|
||||
t.Error("Expected error with invalid url ''")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetInvalidWriter(t *testing.T) {
|
||||
err := GetData("http://localhost:8081", "0000", "/", nil)
|
||||
if err == nil {
|
||||
t.Error("Expected error with no writer")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetURLNotFoundWithUnknownID(t *testing.T) {
|
||||
defer gock.Off()
|
||||
|
||||
gock.New("http://localhost:8081").
|
||||
Get("/handlers/000/").Reply(http.StatusNotFound)
|
||||
|
||||
err := GetData("http://localhost:8081", "000", "/", os.Stdout)
|
||||
|
||||
if err == nil {
|
||||
t.Errorf("Expect not found error but get no error")
|
||||
}
|
||||
|
||||
if gock.IsDone() == false {
|
||||
t.Error("No expected endpoint called")
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetRetrieveRequestMethod(t *testing.T) {
|
||||
defer gock.Off()
|
||||
|
||||
gock.New("http://localhost:8081").
|
||||
Get("/handlers/000/request/method").
|
||||
Reply(http.StatusAccepted).
|
||||
BodyString("POST")
|
||||
|
||||
rw := new(bytes.Buffer)
|
||||
|
||||
err := GetData("http://localhost:8081", "000", "/request/method", rw)
|
||||
if err != nil {
|
||||
t.Errorf("Unexpected error %v", err)
|
||||
}
|
||||
strRes := rw.String()
|
||||
|
||||
if strRes != "POST" {
|
||||
t.Errorf("POST string expected but found: '%v'", strRes)
|
||||
}
|
||||
|
||||
if gock.IsDone() == false {
|
||||
t.Error("No expected endpoint called")
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// AddRoute will add a new route in kapow
|
||||
func AddRoute(host, path, method, entrypoint, command string) error {
|
||||
reqData, err := http.NewRequest(
|
||||
"PUT",
|
||||
host+"/routes",
|
||||
strings.NewReader(command),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var client = new(http.Client)
|
||||
res, err := client.Do(reqData)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if res.StatusCode < 200 || res.StatusCode >= 300 {
|
||||
return errors.New(res.Status)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
package client
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestInvalidURL(t *testing.T) {
|
||||
err := AddRoute("http://localhost;8080", "/hi", "GET", "bash -c", "echo 'Hi' | kapow set /response/body")
|
||||
if err == nil {
|
||||
t.Error("expect to fail due invalid url")
|
||||
t.Fail()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user