github.com/go-chef/chef@v0.30.1/role_test.go

package chef

import (
	"encoding/json"
	"fmt"
	"github.com/r3labs/diff"
	"github.com/stretchr/testify/assert"
	"io"
	"log"
	"net/http"
	"os"
	"reflect"
	"testing"
)

var (
	testRoleJSON = "test/role.json"
	// FML
	testRole = &Role{
		Name:               "test",
		ChefType:           "role",
		Description:        "Test Role",
		RunList:            []string{"recipe[foo]", "recipe[baz]", "role[banana]"},
		JsonClass:          "Chef::Role",
		DefaultAttributes:  struct{}{},
		OverrideAttributes: struct{}{},
	}
)

func TestRoleName(t *testing.T) {
	n1 := testRole
	name := n1.Name
	assert.Equal(t, "test", name, "Role name")
}

// BUG(fujin): re-do with goconvey
func TestRoleFromJSONDecoder(t *testing.T) {
	if file, err := os.Open(testRoleJSON); err != nil {
		t.Error("unexpected error", err, "during os.Open on", testRoleJSON)
	} else {
		dec := json.NewDecoder(file)
		var n Role
		if err := dec.Decode(&n); err == io.EOF {
			log.Println(n)
		} else if err != nil {
			log.Fatal(err)
		}
	}
}

func TestRolesService_List(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `{"foo":"http://localhost:4000/roles/foo", "webserver":"http://localhost:4000/roles/webserver"}`)
	})

	roles, err := client.Roles.List()
	if err != nil {
		t.Errorf("Roles.List returned error: %v", err)
	}

	want := &RoleListResult{"foo": "http://localhost:4000/roles/foo", "webserver": "http://localhost:4000/roles/webserver"}

	if !reflect.DeepEqual(roles, want) {
		t.Errorf("Roles.List returned %+v, want %+v", roles, want)
	}
}

func TestRolesService_Get(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles/webserver", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `{
		  "name": "webserver",
		  "chef_type": "role",
		  "json_class": "Chef::Role",
		  "default_attributes": "",
		  "description": "A webserver",
		  "run_list": [
		    "recipe[unicorn]",
		    "recipe[apache2]"
		  ],
		  "override_attributes": ""
		}
		`)
	})

	role, err := client.Roles.Get("webserver")
	if err != nil {
		t.Errorf("Roles.Get returned error: %v", err)
	}

	want := &Role{
		Name:               "webserver",
		ChefType:           "role",
		JsonClass:          "Chef::Role",
		DefaultAttributes:  "",
		Description:        "A webserver",
		RunList:            []string{"recipe[unicorn]", "recipe[apache2]"},
		OverrideAttributes: "",
	}

	if !reflect.DeepEqual(role, want) {
		t.Errorf("Roles.Get returned %+v, want %+v", role, want)
	}
}

func TestRolesService_Create(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `{ "uri": "http://localhost:4000/roles/webserver" }`)
	})

	role := &Role{
		Name:               "webserver",
		ChefType:           "role",
		JsonClass:          "Chef::Role",
		DefaultAttributes:  "",
		Description:        "A webserver",
		RunList:            []string{"recipe[unicorn]", "recipe[apache2]"},
		OverrideAttributes: "",
	}

	uri, err := client.Roles.Create(role)
	if err != nil {
		t.Errorf("Roles.Create returned error: %v", err)
	}

	want := &RoleCreateResult{"uri": "http://localhost:4000/roles/webserver"}

	if !reflect.DeepEqual(uri, want) {
		t.Errorf("Roles.Create returned %+v, want %+v", uri, want)
	}
}

func TestRolesService_Put(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles/webserver", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `{
		  "name": "webserver",
		  "chef_type": "role",
		  "json_class": "Chef::Role",
		  "description": "A webserver",
		  "run_list": [
		    "recipe[apache2]"
		  ]
		}`)
	})

	role := &Role{
		Name:        "webserver",
		ChefType:    "role",
		JsonClass:   "Chef::Role",
		Description: "A webserver",
		RunList:     []string{"recipe[apache2]"},
	}

	updatedRole, err := client.Roles.Put(role)
	if err != nil {
		t.Errorf("Roles.Put returned error: %v", err)
	}

	if !reflect.DeepEqual(updatedRole, role) {
		t.Errorf("Roles.Put returned %+v, want %+v", updatedRole, role)
	}
}

func TestRolesService_GetEnvironments(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles/webserver/environments", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `[ "_default", "env1"]`)
	})

	want := RoleEnvironmentsResult{
		"_default",
		"env1",
	}

	updatedRole, err := client.Roles.GetEnvironments("webserver")
	if err != nil {
		t.Errorf("Roles.GetEnvironments returned error: %v", err)
	}

	if !reflect.DeepEqual(updatedRole, want) {
		t.Errorf("Roles.GetEnvironments returned %+v, want %+v", updatedRole, want)
	}
}

func TestRolesService_GetEnvironmentRunlist(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles/webserver/environments/env1", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `{"run_list": ["recipe[foo1]", "recipe[foo2]"]}`)
	})

	list := []string{"recipe[foo1]", "recipe[foo2]"}
	want := map[string][]string{"run_list": list}

	updatedRole, err := client.Roles.GetEnvironmentRunlist("webserver", "env1")
	if err != nil {
		t.Errorf("Roles.GetEnvironmentRunlist returned error: %v", err)
	}

	diff, err := diff.Diff(updatedRole, want)
	if err != nil {
		t.Errorf("Roles.GetEnvironmentRunlist returned %+v, want %+v", updatedRole, want)
		t.Errorf("Diff  comparison %+v err %+v\n", diff, err)
	}
}

func TestRolesService_RoleListResultString(t *testing.T) {
	r := &RoleListResult{"foo": "http://localhost:4000/roles/foo"}
	rstr := r.String()
	want := "foo => http://localhost:4000/roles/foo\n"
	if rstr != want {
		t.Errorf("RoleListResult.String returned %+v, want %+v", rstr, want)
	}
}

func TestRolesService_RoleCreateResultString(t *testing.T) {
	r := &RoleCreateResult{"uri": "http://localhost:4000/roles/webserver"}
	rstr := r.String()
	want := "uri => http://localhost:4000/roles/webserver\n"
	if rstr != want {
		t.Errorf("RoleCreateResult.String returned %+v, want %+v", rstr, want)
	}
}

func TestRolesService_Delete(t *testing.T) {
	setup()
	defer teardown()

	mux.HandleFunc("/roles/webserver", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, `{
		  "name": "webserver",
		  "chef_type": "role",
		  "json_class": "Chef::Role",
		  "description": "A webserver",
		  "run_list": [
		    "recipe[apache2]"
		  ]
		}`)
	})
	err := client.Roles.Delete("webserver")
	if err != nil {
		t.Errorf("Roles.Delete returned error: %v", err)
	}
}