51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
|
package unifi
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNew(t *testing.T) {
|
||
|
|
// Mock UniFi controller response for login/initialization
|
||
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.Header().Set("Content-Type", "application/json")
|
||
|
|
fmt.Fprintln(w, `{"meta":{"rc":"ok"}, "data": []}`)
|
||
|
|
}))
|
||
|
|
defer ts.Close()
|
||
|
|
|
||
|
|
// Test basic client creation
|
||
|
|
client, err := New(ts.URL, "user", "pass", "", true)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NotNil(t, client)
|
||
|
|
assert.Equal(t, ts.URL, client.URL())
|
||
|
|
assert.NotNil(t, client.API())
|
||
|
|
|
||
|
|
if client.API().Client != nil && client.API().Client.Transport != nil {
|
||
|
|
if tr, ok := client.API().Client.Transport.(*http.Transport); ok {
|
||
|
|
assert.True(t, tr.TLSClientConfig.InsecureSkipVerify)
|
||
|
|
} else {
|
||
|
|
t.Errorf("expected *http.Transport, got %T", client.API().Client.Transport)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
t.Errorf("client or transport is nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Test with insecure false
|
||
|
|
client, err = New(ts.URL, "user", "pass", "", false)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
if tr, ok := client.API().Client.Transport.(*http.Transport); ok {
|
||
|
|
assert.False(t, tr.TLSClientConfig.InsecureSkipVerify)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNew_Error(t *testing.T) {
|
||
|
|
// uf.NewUnifi fails if URL is invalid (e.g. missing scheme)
|
||
|
|
client, err := New("localhost:8443", "user", "pass", "", false)
|
||
|
|
assert.Error(t, err)
|
||
|
|
assert.Nil(t, client)
|
||
|
|
}
|