Transport users tests (#105)

2026-05-24 09:28:54 +00:00 · 2016-09-04 14:31:17 -04:00 · 2016-09-04 14:31:17 -04:00 · 1abc44790e
commit 1abc44790e
parent 1f6c8a82de
1 changed files with 147 additions and 0 deletions
--- a/kitserver/transport_users_test.go
+++ b/kitserver/transport_users_test.go
@ -0,0 +1,147 @@
+package kitserver
+
+import (
+	"bytes"
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gorilla/mux"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDecodeCreateUserRequest(t *testing.T) {
+	router := mux.NewRouter()
+	router.HandleFunc("/api/v1/kolide/users", func(writer http.ResponseWriter, request *http.Request) {
+		r, err := decodeCreateUserRequest(context.Background(), request)
+		assert.Nil(t, err)
+
+		params := r.(createUserRequest)
+		assert.Equal(t, "foo", *params.payload.Name)
+		assert.Equal(t, "foo@kolide.co", *params.payload.Email)
+	}).Methods("POST")
+
+	var body bytes.Buffer
+	body.Write([]byte(`{
+        "name": "foo",
+        "email": "foo@kolide.co"
+    }`))
+
+	router.ServeHTTP(
+		httptest.NewRecorder(),
+		httptest.NewRequest("POST", "/api/v1/kolide/users", &body),
+	)
+}
+
+func TestDecodeGetUserRequest(t *testing.T) {
+	router := mux.NewRouter()
+	router.HandleFunc("/api/v1/kolide/users/{id}", func(writer http.ResponseWriter, request *http.Request) {
+		r, err := decodeGetUserRequest(context.Background(), request)
+		assert.Nil(t, err)
+
+		params := r.(getUserRequest)
+		assert.Equal(t, uint(1), params.ID)
+	}).Methods("GET")
+
+	router.ServeHTTP(
+		httptest.NewRecorder(),
+		httptest.NewRequest("GET", "/api/v1/kolide/users/1", nil),
+	)
+}
+
+func TestDecodeChangePasswordRequest(t *testing.T) {
+	router := mux.NewRouter()
+	router.HandleFunc("/api/v1/kolide/users/{id}/password", func(writer http.ResponseWriter, request *http.Request) {
+		r, err := decodeChangePasswordRequest(context.Background(), request)
+		assert.Nil(t, err)
+
+		params := r.(changePasswordRequest)
+		assert.Equal(t, "foo", params.CurrentPassword)
+		assert.Equal(t, "bar", params.NewPassword)
+		assert.Equal(t, "baz", params.PasswordResetToken)
+		assert.Equal(t, uint(1), params.UserID)
+	}).Methods("POST")
+
+	var body bytes.Buffer
+	body.Write([]byte(`{
+        "current_password": "foo",
+        "new_password": "bar",
+        "password_reset_token": "baz"
+    }`))
+
+	router.ServeHTTP(
+		httptest.NewRecorder(),
+		httptest.NewRequest("POST", "/api/v1/kolide/users/1/password", &body),
+	)
+}
+
+func TestDecodeUpdateAdminRoleRequest(t *testing.T) {
+	router := mux.NewRouter()
+	router.HandleFunc("/api/v1/kolide/users/{id}/admin", func(writer http.ResponseWriter, request *http.Request) {
+		r, err := decodeUpdateAdminRoleRequest(context.Background(), request)
+		assert.Nil(t, err)
+
+		params := r.(updateAdminRoleRequest)
+		assert.Equal(t, true, params.Admin)
+		assert.Equal(t, uint(1), params.UserID)
+	}).Methods("POST")
+
+	var body bytes.Buffer
+	body.Write([]byte(`{
+        "admin": true
+    }`))
+
+	router.ServeHTTP(
+		httptest.NewRecorder(),
+		httptest.NewRequest("POST", "/api/v1/kolide/users/1/admin", &body),
+	)
+}
+
+func TestDecodeUpdateUserStatusRequest(t *testing.T) {
+	router := mux.NewRouter()
+	router.HandleFunc("/api/v1/kolide/users/{id}/status", func(writer http.ResponseWriter, request *http.Request) {
+		r, err := decodeUpdateUserStatusRequest(context.Background(), request)
+		assert.Nil(t, err)
+
+		params := r.(updateUserStatusRequest)
+		assert.Equal(t, true, params.Enabled)
+		assert.Equal(t, "foo", params.CurrentPassword)
+		assert.Equal(t, uint(1), params.UserID)
+	}).Methods("POST")
+
+	var body bytes.Buffer
+	body.Write([]byte(`{
+        "enabled": true,
+        "current_password": "foo"
+    }`))
+
+	router.ServeHTTP(
+		httptest.NewRecorder(),
+		httptest.NewRequest("POST", "/api/v1/kolide/users/1/status", &body),
+	)
+}
+
+func TestDecodeModifyUserRequest(t *testing.T) {
+	router := mux.NewRouter()
+	router.HandleFunc("/api/v1/kolide/users/{id}", func(writer http.ResponseWriter, request *http.Request) {
+		r, err := decodeModifyUserRequest(context.Background(), request)
+		assert.Nil(t, err)
+
+		params := r.(modifyUserRequest)
+		assert.Equal(t, "foo", *params.payload.Name)
+		assert.Equal(t, "foo@kolide.co", *params.payload.Email)
+		assert.Equal(t, uint(1), params.ID)
+	}).Methods("PATCH")
+
+	var body bytes.Buffer
+	body.Write([]byte(`{
+        "name": "foo",
+        "email": "foo@kolide.co"
+    }`))
+
+	router.ServeHTTP(
+		httptest.NewRecorder(),
+		httptest.NewRequest("PATCH", "/api/v1/kolide/users/1", &body),
+	)
+}