Initial commit

test/api_worker_test.go

@@ -0,0 +1,83 @@
+package test
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"src/task_tracker/api"
+	"testing"
+)
+
+func TestCreateGetWorker(t *testing.T) {
+
+	resp, r := createWorker(api.CreateWorkerRequest{})
+
+	if r.StatusCode != 200 {
+		t.Fail()
+	}
+
+	if resp.Ok != true {
+		t.Fail()
+	}
+
+	getResp, r := getWorker(resp.WorkerId)
+
+	if r.StatusCode != 200 {
+		t.Fail()
+	}
+
+	if resp.WorkerId != getResp.Worker.Id.String() {
+		t.Fail()
+	}
+}
+
+func TestGetWorkerNotFound(t *testing.T) {
+
+	resp, r := getWorker("8bfc0ccd-d5ce-4dc5-a235-3a7ae760d9c6")
+
+	if r.StatusCode != 404 {
+		t.Fail()
+	}
+	if resp.Ok != false {
+		t.Fail()
+	}
+}
+
+func TestGetWorkerInvalid(t *testing.T) {
+
+	resp, r := getWorker("invalid-uuid")
+
+	if r.StatusCode != 400 {
+		t.Fail()
+	}
+	if resp.Ok != false {
+		t.Fail()
+	}
+	if len(resp.Message) <= 0 {
+		t.Fail()
+	}
+}
+
+func createWorker(req api.CreateWorkerRequest) (*api.CreateWorkerResponse, *http.Response) {
+	r := Post("/worker/create", req)
+
+	var resp *api.CreateWorkerResponse
+	data, _ := ioutil.ReadAll(r.Body)
+	err := json.Unmarshal(data, &resp)
+	handleErr(err)
+
+	return resp, r
+}
+
+func getWorker(id string) (*api.GetWorkerResponse, *http.Response) {
+
+	r := Get(fmt.Sprintf("/worker/get/%s", id))
+
+	var resp *api.GetWorkerResponse
+	data, _ := ioutil.ReadAll(r.Body)
+	err := json.Unmarshal(data, &resp)
+	handleErr(err)
+
+	return resp, r
+}