Rework directory structure

Co-authored-by: Héctor Hurtado <hector.hurtado@bbva.com>

internal/client/get.go

@@ -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
+}

internal/client/get_test.go

@@ -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")
+	}
+}

internal/client/route_add.go

@@ -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
+}

internal/client/route_add_test.go

@@ -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()
+	}
+}

internal/client/set.go

@@ -0,0 +1,36 @@
+package client
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+)
+
+const (
+	errMandatoryParam   = "Mandatory parameter %s missing"
+	errInvalidURL       = "kapowURL, handlerId or path has invalid format"
+	errNotFound         = "Resource Item Not Found"
+	errNotValidResource = "Invalid Resource Path"
+	serverURLTemplate   = "%s/%s%s"
+)
+
+func SetData(kapowURL, handlerId, path string, r io.Reader) error {
+
+	req, err := http.NewRequest("PUT", fmt.Sprintf(serverURLTemplate, kapowURL, handlerId, path), r)
+	if err != nil {
+		return err
+	}
+
+	kpowClient := &http.Client{}
+	if resp, err := kpowClient.Do(req); err != nil {
+		return err
+	} else if resp.StatusCode == http.StatusNoContent {
+		return fmt.Errorf(errNotFound)
+	} else if resp.StatusCode == http.StatusBadRequest {
+		return fmt.Errorf(errNotValidResource)
+	} else if resp.StatusCode >= http.StatusNotFound {
+		return fmt.Errorf(resp.Status[4:])
+	}
+
+	return nil
+}

internal/client/set_test.go

@@ -0,0 +1,123 @@
+package client_test
+
+import (
+	"net/http"
+	"strings"
+	"testing"
+
+	"github.com/BBVA/kapow/internal/client"
+
+	gock "gopkg.in/h2non/gock.v1"
+)
+
+// Test that no content errors are detected as non-existent resource
+func TestNoContent(t *testing.T) {
+	expectedErr := "Resource Item Not Found"
+	host := "http://localhost:8080"
+	hid := "xxxxxxxxxxxxxx"
+	path := "/unpath"
+	reader := strings.NewReader("Esto es un peacho de dato pa repartir")
+
+	defer gock.Off()
+
+	gock.New(host).Put("/" + hid + path).Reply(http.StatusNoContent)
+
+	if err := client.SetData(host, hid, path, reader); err == nil {
+		t.Error("Expected error not present")
+	} else if err.Error() != expectedErr {
+		t.Errorf("Error don't match: expected \"%s\", got \"%s\"", expectedErr, err.Error())
+	}
+
+	if !gock.IsDone() {
+		t.Errorf("No endpoint called")
+	}
+}
+
+// Test that bad request errors are detected as invalid resource
+func TestBadRequest(t *testing.T) {
+	expectedErr := "Invalid Resource Path"
+	host := "http://localhost:8080"
+	hid := "xxxxxxxxxxxxxx"
+	path := "/unpath"
+	reader := strings.NewReader("Esto es un peacho de dato pa repartir")
+
+	defer gock.Off()
+
+	gock.New(host).Put("/" + hid + path).Reply(http.StatusBadRequest)
+
+	if err := client.SetData(host, hid, path, reader); err == nil {
+		t.Error("Expected error not present")
+	} else if err.Error() != expectedErr {
+		t.Errorf("Error don't match: expected \"%s\", got \"%s\"", expectedErr, err.Error())
+	}
+
+	if !gock.IsDone() {
+		t.Errorf("No endpoint called")
+	}
+}
+
+// Test that not found errors are detected as invalid handler id
+func TestNotFound(t *testing.T) {
+	expectedErr := "Not Found"
+	host := "http://localhost:8080"
+	hid := "xxxxxxxxxxxxxx"
+	path := "/unpath"
+	reader := strings.NewReader("Esto es un peacho de dato pa repartir")
+
+	defer gock.Off()
+
+	gock.New(host).Put("/" + hid + path).Reply(http.StatusNotFound)
+
+	if err := client.SetData(host, hid, path, reader); err == nil {
+		t.Error("Expected error not present")
+	} else if err.Error() != expectedErr {
+		t.Errorf("Error don't match: expected \"%s\", got \"%s\"", expectedErr, err.Error())
+	}
+
+	if !gock.IsDone() {
+		t.Errorf("No endpoint called")
+	}
+}
+
+// Test that internal server errors are detected correctly
+func TestInternalServerError(t *testing.T) {
+	expectedErr := "Internal Server Error"
+	host := "http://localhost:8080"
+	hid := "xxxxxxxxxxxxxx"
+	path := "/unpath"
+	reader := strings.NewReader("Esto es un peacho de dato pa repartir")
+
+	defer gock.Off()
+
+	gock.New(host).Put("/" + hid + path).Reply(http.StatusInternalServerError)
+
+	if err := client.SetData(host, hid, path, reader); err == nil {
+		t.Error("Expected error not present")
+	} else if err.Error() != expectedErr {
+		t.Errorf("Error don't match: expected \"%s\", got \"%s\"", expectedErr, err.Error())
+	}
+
+	if !gock.IsDone() {
+		t.Errorf("No endpoint called")
+	}
+}
+
+// Test a http ok request
+func TestOkRequest(t *testing.T) {
+	host := "http://localhost:8080"
+	hid := "xxxxxxxxxxxxxx"
+	path := "/response/status/code"
+	reader := strings.NewReader("200")
+
+	defer gock.Off()
+
+	gock.New(host).Put("/" + hid + path).Reply(http.StatusOK)
+
+	if err := client.SetData(host, hid, path, reader); err != nil {
+		t.Error("Unexpected error")
+	}
+
+	if !gock.IsDone() {
+		t.Errorf("No endpoint called")
+	}
+}