(MAINT) Add classes endpoint. (#94)

Author: eoinmcq

pkg/classifier/classes.go

@@ -0,0 +1,19 @@
+package classifier
+
+const (
+	classes = "/classifier-api/v1/classes"
+)
+
+func (c *Client) Classes(pagination *Pagination) ([]Class, error) {
+	payload := []Class{}
+	err := getRequest(c, classes, nil, &payload)
+	return payload, err
+}
+
+// Class represents a group returned by the classes endpoint.
+// See https://www.puppet.com/docs/pe/2019.8/classes_endpoint#get_v1_classes
+type Class struct {
+	Name        string
+	Environment string
+	Parameters  map[string]interface{}
+}

pkg/classifier/classes_test.go

@@ -0,0 +1,39 @@
+package classifier
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/jarcoal/httpmock"
+	"github.com/stretchr/testify/require"
+)
+
+var (
+	client        *Client
+	classifierURL = "https://test-host:4433"
+)
+
+func init() {
+	client = NewClient(classifierURL, "xxxx", nil)
+	httpmock.Activate()
+	httpmock.ActivateNonDefault(client.resty.GetClient())
+}
+
+func TestClassesSuccess(t *testing.T) {
+	setupGetResponder(t, classes, "", "classes-response.json")
+	actual, err := client.Classes(nil)
+
+	expected := []Class{{Name: "classinproduction", Environment: "production"},
+		{Name: "classindev", Environment: "development"}}
+
+	require.Nil(t, err)
+	require.True(t, len(actual) == len(expected))
+	require.Equal(t, expected, actual)
+}
+
+func TestClassesFailure(t *testing.T) {
+	setupResponderWithStatusCodeAndBody(t, classes, http.StatusInternalServerError, nil)
+	_, err := client.Classes(nil)
+
+	require.NotNil(t, err)
+}

pkg/classifier/groups_test.go

@@ -47,6 +47,14 @@ func setupGetResponder(t *testing.T, url, query, responseFilename string) {
 	response.Body.Close()
 }
 
+func setupResponderWithStatusCodeAndBody(t *testing.T, url string, statusCode int, response interface{}) {
+	httpmock.Reset()
+	responder, err := httpmock.NewJsonResponder(statusCode, response)
+	require.Nil(t, err)
+	httpmock.RegisterResponder(http.MethodGet, hostURL+url, responder)
+	httpmock.RegisterResponder(http.MethodPost, hostURL+url, responder)
+}
+
 var (
 	pdbClient *Client
 	hostURL   = "https://test-host:4433"

pkg/classifier/testdata/classes-response.json

@@ -0,0 +1,10 @@
+[
+  {
+    "name": "classinproduction",
+    "environment": "production"
+  },
+  {
+    "name": "classindev",
+    "environment": "development"
+  }
+]