1 ae81184d 2022-01-11 o package icinga
 2 ae81184d 2022-01-11 o 
 3 ae81184d 2022-01-11 o import (
 4 ae81184d 2022-01-11 o 	"crypto/tls"
 5 ae81184d 2022-01-11 o 	"encoding/json"
 6 ae81184d 2022-01-11 o 	"errors"
 7 ae81184d 2022-01-11 o 	"net/http"
 8 ae81184d 2022-01-11 o 	"os"
 9 ae81184d 2022-01-11 o 	"reflect"
10 ae81184d 2022-01-11 o 	"testing"
11 ae81184d 2022-01-11 o )
12 ae81184d 2022-01-11 o 
13 ae81184d 2022-01-11 o func TestUser(t *testing.T) {
14 ae81184d 2022-01-11 o 	want := User{Name: "test", Email: "test@example.com", Groups: []string{}}
15 ae81184d 2022-01-11 o 	f, err := os.Open("testdata/users.json")
16 ae81184d 2022-01-11 o 	if err != nil {
17 ae81184d 2022-01-11 o 		t.Fatal(err)
18 ae81184d 2022-01-11 o 	}
19 ae81184d 2022-01-11 o 	defer f.Close()
20 ae81184d 2022-01-11 o 	resp, err := parseResponse(f)
21 ae81184d 2022-01-11 o 	if err != nil {
22 ae81184d 2022-01-11 o 		t.Fatal(err)
23 ae81184d 2022-01-11 o 	}
24 ae81184d 2022-01-11 o 	obj, err := objectFromLookup(resp)
25 ae81184d 2022-01-11 o 	if err != nil {
26 ae81184d 2022-01-11 o 		t.Fatal(err)
27 ae81184d 2022-01-11 o 	}
28 ae81184d 2022-01-11 o 	got, ok := obj.(User)
29 ae81184d 2022-01-11 o 	if !ok {
30 ae81184d 2022-01-11 o 		t.Fatalf("want %T, got %T", want, got)
31 ae81184d 2022-01-11 o 	}
32 ae81184d 2022-01-11 o 	if !reflect.DeepEqual(want, got) {
33 ae81184d 2022-01-11 o 		t.Errorf("want: %+v, got %+v", want, got)
34 ae81184d 2022-01-11 o 	}
35 ae81184d 2022-01-11 o }
36 ae81184d 2022-01-11 o 
37 ae81184d 2022-01-11 o func TestUserMarshal(t *testing.T) {
38 ae81184d 2022-01-11 o 	user := &User{Name: "test", Email: "test@example.com", Groups: []string{}}
39 ae81184d 2022-01-11 o 	want := `{"attrs":{"email":"test@example.com"}}`
40 ae81184d 2022-01-11 o 	got, err := json.Marshal(user)
41 ae81184d 2022-01-11 o 	if err != nil {
42 ae81184d 2022-01-11 o 		t.Fatal(err)
43 ae81184d 2022-01-11 o 	}
44 ae81184d 2022-01-11 o 	if string(got) != want {
45 ae81184d 2022-01-11 o 		t.Fail()
46 ae81184d 2022-01-11 o 	}
47 ae81184d 2022-01-11 o 	t.Logf("want %s, got %s", want, got)
48 ae81184d 2022-01-11 o }
49 ae81184d 2022-01-11 o 
50 ae81184d 2022-01-11 o func TestUserRoundTrip(t *testing.T) {
51 ae81184d 2022-01-11 o 	tp := http.DefaultTransport.(*http.Transport)
52 ae81184d 2022-01-11 o 	tp.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
53 ae81184d 2022-01-11 o 	c := http.DefaultClient
54 ae81184d 2022-01-11 o 	c.Transport = tp
55 ae81184d 2022-01-11 o 	client, err := Dial("127.0.0.1:5665", "root", "8eec5ede1673b757", c)
56 ae81184d 2022-01-11 o 	if err != nil {
57 ae81184d 2022-01-11 o 		t.Skipf("no local test icinga? got: %v", err)
58 ae81184d 2022-01-11 o 	}
59 ae81184d 2022-01-11 o 
60 ae81184d 2022-01-11 o 	want := User{Name: "olly", Email: "olly@example.com", Groups: []string{}}
61 ae81184d 2022-01-11 o 	if err := client.CreateUser(want); err != nil && !errors.Is(err, ErrExist) {
62 ae81184d 2022-01-11 o 		t.Fatal(err)
63 ae81184d 2022-01-11 o 	}
64 ae81184d 2022-01-11 o 	defer func() {
65 ae81184d 2022-01-11 o 		if err := client.DeleteUser(want.Name); err != nil {
66 ae81184d 2022-01-11 o 			t.Error(err)
67 ae81184d 2022-01-11 o 		}
68 ae81184d 2022-01-11 o 	}()
69 ae81184d 2022-01-11 o 	got, err := client.LookupUser(want.Name)
70 ae81184d 2022-01-11 o 	if err != nil {
71 ae81184d 2022-01-11 o 		t.Fatal(err)
72 ae81184d 2022-01-11 o 	}
73 ae81184d 2022-01-11 o 	if !reflect.DeepEqual(want, got) {
74 ae81184d 2022-01-11 o 		t.Errorf("want %+v, got %+v", want, got)
75 ae81184d 2022-01-11 o 	}
76 ae81184d 2022-01-11 o }