test: add tests for Unifi client

2025-02-09 01:14:45 +01:00
parent f79f21c4ad
commit 685572d5ce
4 changed files with 511 additions and 15 deletions
--- a/unifi/unifi_test.go
+++ b/unifi/unifi_test.go
@@ -0,0 +1,484 @@
+package unifi
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"github.com/stretchr/testify/assert"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"reflect"
+	"slices"
+	"strings"
+	"testing"
+)
+
+const (
+	localUrl = "http://127.0.0.1:64431"
+	testUrl  = "http://test.url"
+)
+
+func verifyContainsInterceptors(a *assert.Assertions, c *Client, interceptors ...interface{}) {
+	var (
+		expectedTypes []reflect.Type
+		matchingTypes []reflect.Type
+	)
+	for _, i := range interceptors {
+		expectedTypes = append(expectedTypes, reflect.TypeOf(i))
+	}
+	for _, i := range c.interceptors {
+		actualType := reflect.TypeOf(i)
+		if slices.Contains(expectedTypes, actualType) {
+			matchingTypes = append(matchingTypes, actualType)
+		}
+	}
+	if len(matchingTypes) != len(expectedTypes) {
+		a.Fail(fmt.Sprintf("interceptors not found; expected: %v, found: %v", expectedTypes, matchingTypes))
+	}
+}
+
+func verifyDoesNotContainInterceptors(a *assert.Assertions, c *Client, interceptors ...interface{}) {
+	var (
+		expectedTypes []reflect.Type
+		matchingTypes []reflect.Type
+	)
+	for _, i := range interceptors {
+		expectedTypes = append(expectedTypes, reflect.TypeOf(i))
+	}
+	for _, i := range c.interceptors {
+		actualType := reflect.TypeOf(i)
+		if slices.Contains(expectedTypes, actualType) {
+			matchingTypes = append(matchingTypes, actualType)
+		}
+	}
+	if len(matchingTypes) != 0 {
+		a.Fail(fmt.Sprintf("interceptors found; expected to be not present: %v, found: %v", expectedTypes, matchingTypes))
+	}
+}
+
+func TestNewClient(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	c, err := NewClient(&ClientConfig{
+		URL:       localUrl,
+		User:      "admin",
+		Pass:      "password",
+		VerifySSL: false,
+	})
+	a.NotNil(err)
+	a.EqualValues(localUrl, c.BaseURL.String())
+	a.Contains(err.Error(), "connection refused", "an invalid destination should produce a connection error.")
+	verifyContainsInterceptors(a, c, &CsrfInterceptor{}, &DefaultHeadersInterceptor{})
+	verifyDoesNotContainInterceptors(a, c, &ApiKeyAuthInterceptor{})
+}
+
+func TestNewClientWithApiKey(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// when
+	c, err := NewClient(&ClientConfig{
+		URL:       localUrl,
+		APIKey:    "test",
+		VerifySSL: false,
+	})
+
+	// then
+	a.NotNil(err)
+	a.EqualValues(localUrl, c.BaseURL.String())
+	a.Contains(err.Error(), "connection refused", "an invalid destination should produce a connection error.")
+	verifyContainsInterceptors(a, c, &ApiKeyAuthInterceptor{}, &DefaultHeadersInterceptor{})
+	verifyDoesNotContainInterceptors(a, c, &CsrfInterceptor{})
+}
+
+func TestCustomizeHttpClient(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	called := false
+
+	// when
+	NewClient(&ClientConfig{
+		URL: localUrl,
+		HttpCustomizer: func(transport *http.Transport) error {
+			called = true
+			return nil
+		},
+	})
+
+	// then
+	a.True(called, "http customizer not called")
+}
+
+type TestInterceptor struct {
+	request       *http.Request
+	response      *http.Response
+	failOnRequest bool
+}
+
+func (i *TestInterceptor) IsRequestIntercepted() bool {
+	return i.request != nil
+}
+
+func (i *TestInterceptor) IsResponseIntercepted() bool {
+	return i.response != nil
+}
+
+func (i *TestInterceptor) InterceptRequest(req *http.Request) error {
+	i.request = req
+	if i.failOnRequest {
+		return fmt.Errorf("request interceptor failed")
+	}
+	return nil
+}
+func (i *TestInterceptor) InterceptResponse(resp *http.Response) error {
+	i.response = resp
+	return nil
+}
+
+func (i *TestInterceptor) RequestHeader(key string) string {
+	return i.request.Header.Get(key)
+}
+
+func (i *TestInterceptor) ResponseHeader(key string) string {
+	return i.response.Header.Get(key)
+}
+
+func (i *TestInterceptor) Method() string {
+	return i.request.Method
+}
+
+func NewTestInterceptor() *TestInterceptor {
+	return &TestInterceptor{}
+}
+
+func (i *TestInterceptor) AsList() []ClientInterceptor {
+	return []ClientInterceptor{i}
+}
+
+func NewTestClientWithInterceptor() (*Client, *TestInterceptor) {
+	interceptor := NewTestInterceptor()
+	c, _ := NewClient(&ClientConfig{
+		URL:          testUrl,
+		APIKey:       "test-key",
+		Interceptors: interceptor.AsList(),
+	})
+	c.apiPaths = &NewStyleAPI
+	return c, interceptor
+}
+
+func TestInterceptors(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Get(context.Background(), "/", nil, nil)
+
+	// then
+	a.True(interceptor.IsRequestIntercepted(), "request interceptor not called")
+	a.False(interceptor.IsResponseIntercepted(), "response interceptor called, but should not because of failed request")
+}
+
+func TestNoSendRequestWhenRequestInterceptorReturnsError(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+	interceptor.failOnRequest = true
+
+	// when
+	err := c.Get(context.Background(), "/", nil, nil)
+
+	// then
+	a.NotNil(err)
+	a.Contains(err.Error(), "request interceptor failed")
+}
+
+func TestProperRequestUrl(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	testCases := []struct {
+		path     string
+		expected string
+	}{
+		{"", testUrl + NewStyleAPI.ApiPath},
+		{"test", testUrl + NewStyleAPI.ApiPath + "/test"},
+		{"test/", testUrl + NewStyleAPI.ApiPath + "/test"},
+		{"test/test", testUrl + NewStyleAPI.ApiPath + "/test/test"},
+		{"/test/", testUrl + "/test/"},
+		{"/test", testUrl + "/test"},
+		{"/test/test", testUrl + "/test/test"},
+	}
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	for _, tc := range testCases {
+		t.Run(tc.path, func(t *testing.T) {
+			// when
+			c.Get(context.Background(), tc.path, nil, nil)
+
+			// then
+			a.EqualValues(tc.expected, interceptor.request.URL.String())
+		})
+	}
+}
+
+func TestApiKeyAddedToRequest(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Get(context.Background(), "/", nil, nil)
+
+	// then
+	a.EqualValues("test-key", interceptor.RequestHeader(ApiKeyHeader))
+}
+
+func TestDefaultHeadersAddedToRequest(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+	
+	// when
+	c.Get(context.Background(), "/", nil, nil)
+
+	// then
+	a.EqualValues("application/json", interceptor.RequestHeader(AcceptHeader))
+	a.EqualValues("application/json; charset=utf-8", interceptor.RequestHeader(ContentTypeHeader))
+	a.EqualValues(defaultUserAgent, interceptor.RequestHeader(UserAgentHeader))
+}
+
+type TestData struct {
+	Data string `json:"data"`
+}
+
+func TestRequestSentWithJson(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+	data := &TestData{
+		Data: "test",
+	}
+
+	// when
+	c.Get(context.Background(), "/", data, nil)
+
+	// then
+	body := &TestData{}
+	err := json.NewDecoder(interceptor.request.Body).Decode(body)
+
+	a.Nil(err)
+	a.Equal(data, body)
+}
+
+func TestRequestMethod(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	testCases := []string{
+		http.MethodGet, http.MethodPost, http.MethodPut, http.MethodDelete, http.MethodPatch, http.MethodOptions, http.MethodHead, http.MethodTrace, http.MethodConnect,
+	}
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Post(context.Background(), "/", nil, nil)
+
+	// then
+	for _, tc := range testCases {
+		t.Run(tc, func(t *testing.T) {
+			// when
+			c.Do(context.Background(), tc, "", nil, nil)
+
+			// then
+			a.EqualValues(tc, interceptor.Method())
+		})
+	}
+}
+
+func TestGetRequest(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Get(context.Background(), "/", nil, nil)
+
+	// then
+	a.EqualValues(http.MethodGet, interceptor.Method())
+}
+
+func TestPostRequest(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Post(context.Background(), "/", nil, nil)
+
+	// then
+	a.EqualValues(http.MethodPost, interceptor.Method())
+}
+
+func TestPutRequest(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Put(context.Background(), "/", nil, nil)
+
+	// then
+	a.EqualValues(http.MethodPut, interceptor.Method())
+}
+
+func TestDeleteRequest(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	c, interceptor := NewTestClientWithInterceptor()
+
+	// when
+	c.Delete(context.Background(), "/", nil, nil)
+
+	// then
+	a.EqualValues(http.MethodDelete, interceptor.Method())
+}
+
+func RunTestServer(path string, requestBody interface{}) *httptest.Server {
+	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add(CsrfHeader, "csrf-token")
+		if !strings.EqualFold(r.URL.Path, path) {
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+		w.WriteHeader(http.StatusOK)
+		data, err := io.ReadAll(r.Body)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			fmt.Printf("error reading body:%v", err)
+			return
+		}
+		err = json.Unmarshal(data, &requestBody)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			fmt.Printf("error decoding body: %s: %s", string(data), err)
+			return
+		}
+		resp := TestData{
+			Data: "test",
+		}
+		respData, err := json.Marshal(resp)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			fmt.Printf("error encoding response: %s", err)
+			return
+		}
+		_, err = w.Write(respData)
+		if err != nil {
+			fmt.Printf("error writing response: %s", err)
+		}
+	}))
+}
+
+func TestUnifiIntegrationUserPassInjected(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	type userPass struct {
+		Username string `json:"username"`
+		Password string `json:"password"`
+	}
+	srv := RunTestServer(NewStyleAPI.LoginPath, userPass{})
+	interceptor := NewTestInterceptor()
+	c, _ := NewClient(&ClientConfig{
+		URL:          srv.URL,
+		User:         "test-user",
+		Pass:         "test-pass",
+		Interceptors: interceptor.AsList(),
+	})
+	c.apiPaths = &NewStyleAPI
+
+	// when
+	err := c.Login()
+
+	// then
+	a.Nil(err, "user/pass login must not produce an error")
+	a.EqualValues(http.MethodPost, interceptor.Method())
+	a.EqualValues(http.StatusOK, interceptor.response.StatusCode)
+}
+
+func TestResponseDataHandling(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	reqData := TestData{
+		Data: "request",
+	}
+	srv := RunTestServer(NewStyleAPI.ApiPath+"/test", TestData{})
+	c, _ := NewClient(&ClientConfig{
+		URL: srv.URL,
+	})
+	c.apiPaths = &NewStyleAPI
+	var data TestData
+
+	// when
+	err := c.Get(context.Background(), "test", reqData, &data)
+
+	// then
+	a.Nil(err)
+	a.EqualValues("test", data.Data)
+}
+
+func TestCsrfHandling(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	srv := RunTestServer("", struct{}{})
+	interceptor := NewTestInterceptor()
+	c, _ := NewClient(&ClientConfig{
+		URL:          srv.URL,
+		Interceptors: interceptor.AsList(),
+	})
+	c.apiPaths = &NewStyleAPI
+
+	// when
+	c.Get(context.Background(), "", nil, nil)
+
+	// then
+	a.EqualValues("", interceptor.RequestHeader(CsrfHeader))
+	a.EqualValues("csrf-token", interceptor.ResponseHeader(CsrfHeader))
+
+	// when
+	c.Get(context.Background(), "", nil, nil)
+
+	// then
+	a.EqualValues("csrf-token", interceptor.RequestHeader(CsrfHeader))
+}
+
+func TestOverrideUserAgent(t *testing.T) {
+	t.Parallel()
+	a := assert.New(t)
+	// given
+	interceptor := NewTestInterceptor()
+	c, _ := NewClient(&ClientConfig{
+		URL:          testUrl,
+		Interceptors: interceptor.AsList(),
+		UserAgent:    "test-agent",
+	})
+	c.apiPaths = &NewStyleAPI
+
+	// when
+	c.Get(context.Background(), "", nil, nil)
+
+	// then
+	a.EqualValues("test-agent", interceptor.RequestHeader(UserAgentHeader))
+}