services_test.go

package mackerel

import (
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"reflect"
	"testing"
)

func TestFindServices(t *testing.T) {
	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
		if req.URL.Path != "/api/v0/services" {
			t.Error("request URL should be /api/v0/services but: ", req.URL.Path)
		}

		if req.Method != "GET" {
			t.Error("request method should be GET but: ", req.Method)
		}

		respJSON, _ := json.Marshal(map[string][]map[string]interface{}{
			"services": {
				{
					"name":  "My-Service",
					"memo":  "hello",
					"roles": []string{"db-master", "db-slave"},
				},
			},
		})

		res.Header()["Content-Type"] = []string{"application/json"}
		fmt.Fprint(res, string(respJSON))
	}))
	defer ts.Close()

	client, _ := NewClientWithOptions("dummy-key", ts.URL, false)
	services, err := client.FindServices()

	if err != nil {
		t.Error("err should be nil but: ", err)
	}

	if services[0].Memo != "hello" {
		t.Error("request sends json including memo but: ", services[0])
	}

	if reflect.DeepEqual(services[0].Roles, []string{"db-master", "db-slave"}) != true {
		t.Errorf("Wrong data for roles: %v", services[0].Roles)
	}

}

func TestCreateService(t *testing.T) {
	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
		if req.URL.Path != "/api/v0/services" {
			t.Error("request URL should be /api/v0/services but: ", req.URL.Path)
		}

		if req.Method != "POST" {
			t.Error("request method should be POST but: ", req.Method)
		}

		respJSON, _ := json.Marshal(map[string]interface{}{
			"name":  "My-Service",
			"memo":  "hello",
			"roles": []string{},
		})

		res.Header()["Content-Type"] = []string{"application/json"}
		fmt.Fprint(res, string(respJSON))
	}))
	defer ts.Close()

	client, _ := NewClientWithOptions("dummy-key", ts.URL, false)

	service, err := client.CreateService(&CreateServiceParam{
		Name: "My-Service",
		Memo: "hello",
	})

	if err != nil {
		t.Error("err should be nil but: ", err)
	}

	if service.Name != "My-Service" {
		t.Error("request sends json including name but: ", service.Name)
	}

	if service.Memo != "hello" {
		t.Error("request sends json including name but: ", service.Memo)
	}

	if len(service.Roles) != 0 {
		t.Error("request sends json including name but: ", service.Roles)
	}
}

func TestDeleteService(t *testing.T) {

	testName := "My-Service"

	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
		if req.URL.Path != fmt.Sprintf("/api/v0/services/%s", testName) {
			t.Error("request URL should be /api/v0/services/<ID> but: ", req.URL.Path)
		}

		if req.Method != "DELETE" {
			t.Error("request method should be DELETE but: ", req.Method)
		}

		respJSON, _ := json.Marshal(map[string]interface{}{
			"name":  "My-Service",
			"memo":  "hello",
			"roles": []string{"ancient-role"},
		})

		res.Header()["Content-Type"] = []string{"application/json"}
		fmt.Fprint(res, string(respJSON))
	}))
	defer ts.Close()

	client, _ := NewClientWithOptions("dummy-key", ts.URL, false)

	service, err := client.DeleteService(testName)

	if err != nil {
		t.Error("err should be nil but: ", err)
	}

	if service.Name != "My-Service" {
		t.Error("request sends json including name but: ", service.Name)
	}

	if service.Memo != "hello" {
		t.Error("request sends json including name but: ", service.Memo)
	}

	if len(service.Roles) != 1 || service.Roles[0] != "ancient-role" {
		t.Error("request sends json including name but: ", service.Roles)
	}
}

func TestListServiceMetricNames(t *testing.T) {
	serviceName := "my-service"
	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
		if req.URL.Path != fmt.Sprintf("/api/v0/services/%s/metric-names", serviceName) {
			t.Error("request URL should be /api/v0/services/<ServiceName>/metric-names but: ", req.URL.Path)
		}

		if req.Method != "GET" {
			t.Error("request method should be GET but: ", req.Method)
		}

		respJSON, _ := json.Marshal(map[string][]string{
			"names": {"access.api", "access.web"},
		})

		res.Header()["Content-Type"] = []string{"application/json"}
		fmt.Fprint(res, string(respJSON))
	}))
	defer ts.Close()

	client, _ := NewClientWithOptions("dummy-key", ts.URL, false)
	names, err := client.ListServiceMetricNames(serviceName)

	if err != nil {
		t.Error("err should be nil but: ", err)
	}

	if reflect.DeepEqual(names, []string{"access.api", "access.web"}) != true {
		t.Errorf("Wrong data for metric names: %v", names)
	}
}