Add test for HandleRegister function

pkg/server/router_test.go

@@ -1,6 +1,7 @@
 package server
 
 import (
+	"bytes"
 	"encoding/json"
 	"net/http"
 	"net/http/httptest"
@@ -23,3 +24,26 @@ func TestHandleStatus(t *testing.T) {
 		t.Errorf("got false for /status")
 	}
 }
+
+func TestHandleRegister(t *testing.T) {
+	data := RegisterData{Name: "one"}
+	b, err := json.Marshal(data)
+	if err != nil {
+		t.Error(err)
+	}
+
+	request, _ := http.NewRequest(http.MethodPost, "/register", bytes.NewReader(b))
+	response := httptest.NewRecorder()
+
+	s := NewServer(8080)
+	s.Initialise()
+
+	s.HandleRegister(response, request)
+
+	var status RegisterResponse
+	json.NewDecoder(response.Body).Decode(&status)
+
+	if status.Success != true {
+		t.Errorf("got false for /register")
+	}
+}