usergroup_test.go

package wavefront

import (
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"net/url"
	"os"
	"testing"
)

type MockUserGroupClient struct {
	Client
	T *testing.T
}

type MockCrudUserGroupClient struct {
	Client
	T      *testing.T
	method string
}

func (m *MockUserGroupClient) Do(req *http.Request) (io.ReadCloser, error) {
	return testDo(m.T, req, "./fixtures/search-usergroup-response.json", "POST", &SearchParams{})
}

func TestUserGroups_Find(t *testing.T) {
	baseurl, _ := url.Parse("http://testing.wavefront.com")
	g := &UserGroups{
		client: &MockUserGroupClient{
			Client: Client{
				Config:     &Config{Token: "1234-5678-9977"},
				BaseURL:    baseurl,
				httpClient: http.DefaultClient,
				debug:      true,
			},
			T: t,
		},
	}

	userGroups, err := g.Find(nil)
	if err != nil {
		t.Fatal(err)
	}

	assertEqual(t, 1, len(userGroups))
	assertEqual(t, "12345678-1234-5678-9977-123456789111", *userGroups[0].ID)
}

func (m *MockCrudUserGroupClient) Do(req *http.Request) (io.ReadCloser, error) {
	resp, err := os.ReadFile("./fixtures/crud-usergroup-response.json")
	if err != nil {
		m.T.Fatal(err)
	}

	if req.Method != m.method {
		m.T.Errorf("request expected %s method, got %s", m.method, req.Method)
	}

	if req.Body != nil {
		body, _ := io.ReadAll(req.Body)

		// The calls for adding/removing users only transmit an array of strings
		// Not an actual UserGroup object.
		var addRemoveBody []string
		if err := json.Unmarshal(body, &addRemoveBody); err == nil {
			return io.NopCloser(bytes.NewReader(resp)), nil
		}

		userGroup := UserGroup{}
		err = json.Unmarshal(body, &userGroup)
		if err != nil {
			m.T.Fatal(err)
		}
	}

	return io.NopCloser(bytes.NewReader(resp)), nil
}

func Test_CreatReadUpdateDelete(t *testing.T) {
	baseurl, _ := url.Parse("http://testing.wavefront.com")
	g := &UserGroups{
		client: &MockCrudUserGroupClient{
			Client: Client{
				Config:     &Config{Token: "1234-5678-9977"},
				BaseURL:    baseurl,
				httpClient: http.DefaultClient,
				debug:      true,
			},
			T: t,
		},
	}

	g.client.(*MockCrudUserGroupClient).method = "POST"

	userGroup := &UserGroup{}
	if err := g.Create(userGroup); err == nil {
		t.Errorf("expected to receive error for missing name")
	}

	userGroup.Name = "testing"
	if err := g.Create(userGroup); err != nil {
		t.Errorf("error creating user group. %s", err)
	}

	assertEqual(t, "12345678-1234-5678-9977-123456789111", *userGroup.ID)

	g.client.(*MockCrudUserGroupClient).method = "PUT"
	var _ = g.Update(userGroup)

	g.client.(*MockCrudUserGroupClient).method = "POST"
	modifyUser := []string{userGroup.Users[0]}

	var _ = g.RemoveUsers(userGroup.ID, &modifyUser)

	var _ = g.AddUsers(userGroup.ID, &modifyUser)

	g.client.(*MockCrudUserGroupClient).method = "DELETE"
	var _ = g.Delete(userGroup)
	assertEqual(t, "", *userGroup.ID)
}