Merge pull request #3 from gravitl/add-tests

basic user api tests

user_test.go

@@ -0,0 +1,203 @@
+package main
+
+import (
+	"bytes"
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"sync"
+	"testing"
+	"time"
+
+	controller "github.com/gravitl/netmaker/controllers"
+	"github.com/gravitl/netmaker/models"
+	"github.com/gravitl/netmaker/mongoconn"
+	"github.com/stretchr/testify/assert"
+)
+
+type AuthorizationResponse struct {
+	Username  string
+	AuthToken string
+}
+
+type goodResponse struct {
+	Code     int
+	Message  string
+	Response AuthorizationResponse
+}
+
+type badResponse struct {
+	Code    int
+	Message string
+}
+
+//assumption:  starting with empty database
+func TestMain(m *testing.M) {
+	mongoconn.ConnectDatabase()
+	var waitgroup sync.WaitGroup
+	waitgroup.Add(1)
+	go controller.HandleRESTRequests(&waitgroup)
+	//wait for http server to start
+	time.Sleep(time.Second * 1)
+	os.Exit(m.Run())
+}
+
+func TestUsers(t *testing.T) {
+
+	t.Run("check that admin user does not exist", func(t *testing.T) {
+		response := checkAdminExists(t)
+		assert.Equal(t, false, response)
+	})
+
+	t.Run("add admin user", func(t *testing.T) {
+		var admin, user models.User
+		admin.UserName = "admin"
+		admin.Password = "password"
+
+		//payload := map[string]string{"username": "admin", "password": "admin"}
+		payload, _ := json.Marshal(admin)
+		request, err := http.NewRequest(http.MethodPost, "http://localhost:8081/users/createadmin", bytes.NewBuffer(payload))
+		if err != nil {
+			t.Error(err)
+		}
+		request.Header.Set("Authorization", "Bearer secretkey")
+		request.Header.Set("Content-Type", "application/json")
+		client := &http.Client{}
+		response, err := client.Do(request)
+		if err != nil {
+			t.Error("error calling createadmin", err)
+		}
+		defer response.Body.Close()
+		body, _ := ioutil.ReadAll(response.Body)
+		_ = json.Unmarshal(body, &user)
+		assert.Equal(t, admin.UserName, user.UserName)
+		assert.Equal(t, true, user.IsAdmin)
+		assert.Equal(t, http.StatusOK, response.StatusCode)
+		adminExists := checkAdminExists(t)
+		assert.Equal(t, true, adminExists)
+	})
+
+	t.Run("GetUser", func(t *testing.T) {
+		t.Skip()
+		//ensure admin exists
+		if !checkAdminExists(t) {
+			t.Error("admin account does not exist")
+			return
+		}
+		//authenticate
+		var admin models.User
+		admin.UserName = "admin"
+		admin.Password = "admin"
+
+		payload, _ := json.Marshal(admin)
+		request, err := http.NewRequest(http.MethodPut, "http://localhost:8081/users/authenticate", bytes.NewBuffer(payload))
+		if err != nil {
+			t.Error(err)
+		}
+		request.Header.Set("Content-Type", "application/json")
+		client := &http.Client{}
+
+		response, err := client.Do(request)
+		if err != nil {
+			t.Error("error calling authenticate", err)
+		}
+		defer response.Body.Close()
+		body := models.User{}
+		json.NewDecoder(response.Body).Decode(&body)
+		t.Log(body)
+		assert.Equal(t, http.StatusOK, response.StatusCode)
+		assert.Equal(t, "admin", body.UserName)
+
+		request, err = http.NewRequest(http.MethodGet, "http://localhost:8081/users/admim", nil)
+		if err != nil {
+			t.Error(err)
+		}
+		request.Header.Set("Authorization", "Bearer secretkey")
+		client = &http.Client{}
+		response, err = client.Do(request)
+		if err != nil {
+			t.Error(err)
+		}
+		defer response.Body.Close()
+		body = models.User{}
+		json.NewDecoder(response.Body).Decode(&body)
+		t.Log(body)
+		assert.Equal(t, http.StatusOK, response.StatusCode)
+		assert.Equal(t, "admin", body.UserName)
+	})
+
+	t.Run("Update User", func(t *testing.T) {
+		t.Skip()
+		var admin, user models.User
+		admin.UserName = "admin"
+		admin.Password = "admin"
+
+		//payload := map[string]string{"username": "admin", "password": "admin"}
+		payload, _ := json.Marshal(admin)
+		request, err := http.NewRequest(http.MethodPut, "http://localhost:8081/users/admin", bytes.NewBuffer(payload))
+		if err != nil {
+			t.Error(err)
+		}
+		request.Header.Set("Authorization", "Bearer secretkey")
+		request.Header.Set("Content-Type", "application/json")
+		client := &http.Client{}
+		response, err := client.Do(request)
+		if err != nil {
+			t.Error("error calling createadmin", err)
+		}
+		defer response.Body.Close()
+		body, _ := ioutil.ReadAll(response.Body)
+		t.Log(string(body))
+		_ = json.Unmarshal(body, &user)
+		assert.Equal(t, admin.UserName, user.UserName)
+		assert.Equal(t, true, user.IsAdmin)
+		assert.Equal(t, http.StatusOK, response.StatusCode)
+	})
+
+	t.Run("Authenticate User", func(t *testing.T) {
+		var admin models.User
+
+		admin.UserName = "admin"
+		admin.Password = "password"
+
+		payload, _ := json.Marshal(admin)
+		request, err := http.NewRequest(http.MethodPost, "http://localhost:8081/users/authenticate", bytes.NewBuffer(payload))
+		if err != nil {
+			t.Error(err)
+		}
+		request.Header.Set("Authorization", "Bearer secretkey")
+		request.Header.Set("Content-Type", "application/json")
+		client := &http.Client{}
+		response, err := client.Do(request)
+		if err != nil {
+			t.Error(err)
+		}
+		defer response.Body.Close()
+		message := goodResponse{}
+		json.NewDecoder(response.Body).Decode(&message)
+		assert.Equal(t, http.StatusOK, response.StatusCode)
+		assert.Equal(t, "W1R3: Device admin Authorized", message.Message)
+	})
+
+	t.Run("empty test", func(t *testing.T) {
+		assert.Equal(t, true, true)
+	})
+
+}
+
+func checkAdminExists(t *testing.T) bool {
+	response, err := http.Get("http://localhost:8081/users/hasadmin")
+	if err != nil {
+		t.Fatal("error calling users/hasadmin", err)
+	}
+	defer response.Body.Close()
+	var body bool
+	json.NewDecoder(response.Body).Decode(&body)
+	return body
+	//body, _ := ioutil.ReadAll(response.Body)
+	//vif body {
+	//	return true
+	//}
+	//return false
+}