From 3aaf5e6b8153c65c69625cc1aacf5816738d51a8 Mon Sep 17 00:00:00 2001
From: Marc Di Luzio <marc.diluzio@gmail.com>
Date: Sun, 31 May 2020 11:22:20 +0100
Subject: [PATCH] Add test for HandleRegister function

---
 pkg/server/router_test.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/pkg/server/router_test.go b/pkg/server/router_test.go
index 8bbd892..1983770 100644
--- a/pkg/server/router_test.go
+++ b/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")
+	}
+}