Added addons service

Author: ess

Diff for: addon.go

@@ -0,0 +1,67 @@
+package eygo
+
+import (
+	"encoding/json"
+)
+
+// Addon is a data structure that models a platform addon on the Engine
+// Yard API.
+type Addon struct {
+	ID     int               `json:"id,omitempty"`
+	Name   string            `json:"name,omitempty"`
+	SSOURL string            `json:"sso_url,omitempty"`
+	Vars   map[string]string `json:"vars,omitempty"`
+}
+
+// AddonService is a repository one can use to retrieve, enable, and disable
+// Addon records on the API.
+type AddonService struct {
+	Driver Driver
+}
+
+// NewAddonService returns a AddonService configured with the provided
+// Driver.
+func NewAddonService(driver Driver) *AddonService {
+	return &AddonService{Driver: driver}
+}
+
+// ForAccount returns an array of Addons that are both associated with the
+// given Account and matching the given Params.
+func (service *AddonService) ForAccount(account *Account, params Params) []*Addon {
+	return service.collection("accounts/"+account.ID+"/addons", params)
+}
+
+func (service *AddonService) collection(path string, params Params) []*Addon {
+	addons := make([]*Addon, 0)
+	response := service.Driver.Get(path, params)
+
+	if response.Okay() {
+		for _, page := range response.Pages {
+			wrapper := struct {
+				Addons []*Addon `json:"addons,omitempty"`
+			}{}
+
+			if err := json.Unmarshal(page, &wrapper); err == nil {
+				addons = append(addons, wrapper.Addons...)
+			}
+		}
+	}
+
+	return addons
+}
+
+/*
+Copyright 2018 Dennis Walters
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/

Diff for: addon_test.go

@@ -0,0 +1,96 @@
+package eygo
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestNewAddonService(t *testing.T) {
+	driver := NewMockDriver()
+	service := NewAddonService(driver)
+
+	t.Run("it is configured with the given driver", func(t *testing.T) {
+		if service.Driver != driver {
+			t.Errorf("Expected the service to use the given driver")
+		}
+	})
+}
+
+func TestAddonService_ForAccount(t *testing.T) {
+	account := &Account{ID: "1", Name: "Account 1"}
+	driver := NewMockDriver()
+	service := NewAddonService(driver)
+
+	t.Run("when there are matching addons", func(t *testing.T) {
+		addon1 := &Addon{ID: 1}
+		addon2 := &Addon{ID: 2}
+		addon3 := &Addon{ID: 3}
+
+		stubAccountAddons(driver, account, addon1, addon2, addon3)
+
+		all := service.ForAccount(account, nil)
+
+		t.Run("it contains all matching addons", func(t *testing.T) {
+			addons := []*Addon{addon1, addon2, addon3}
+
+			if len(all) != len(addons) {
+				t.Errorf("Expected %d addons, got %d", len(addons), len(all))
+			}
+
+			for _, addon := range addons {
+				found := false
+
+				for _, other := range all {
+					if addon.ID == other.ID {
+						found = true
+					}
+				}
+
+				if !found {
+					t.Errorf("Addon %d was not present", addon.ID)
+				}
+			}
+		})
+
+	})
+
+	t.Run("when there are no matching addons", func(t *testing.T) {
+		driver.Reset()
+
+		t.Run("it is empty", func(t *testing.T) {
+			all := service.ForAccount(account, nil)
+
+			if len(all) != 0 {
+				t.Errorf("Expected 0 addons, got")
+			}
+		})
+
+	})
+
+}
+
+func stubAddons(driver *MockDriver, addons ...*Addon) {
+	pages := make([][]byte, 0)
+
+	wrapper := struct {
+		Addons []*Addon `json:"addons,omitempty"`
+	}{Addons: addons}
+
+	if encoded, err := json.Marshal(&wrapper); err == nil {
+		pages = append(pages, encoded)
+		driver.AddResponse("get", "addons", Response{Pages: pages})
+	}
+}
+
+func stubAccountAddons(driver *MockDriver, account *Account, addons ...*Addon) {
+	pages := make([][]byte, 0)
+
+	wrapper := struct {
+		Addons []*Addon `json:"addons,omitempty"`
+	}{Addons: addons}
+
+	if encoded, err := json.Marshal(&wrapper); err == nil {
+		pages = append(pages, encoded)
+		driver.AddResponse("get", "accounts/"+account.ID+"/addons", Response{Pages: pages})
+	}
+}