From a3d6899698af996597b30d1097a97b6000583bb0 Mon Sep 17 00:00:00 2001
From: tosuke <tasukeprg@gmail.com>
Date: Wed, 7 Sep 2022 17:30:53 +0900
Subject: [PATCH] feat: implement bulk retire API

---
 hosts.go      |  9 +++++
 hosts_test.go | 93 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+)

diff --git a/hosts.go b/hosts.go
index 68bb275..e23a67f 100644
--- a/hosts.go
+++ b/hosts.go
@@ -295,6 +295,15 @@ func (c *Client) RetireHost(id string) error {
 	return err
 }
 
+// BulkRetireHosts retires the hosts
+func (c *Client) BulkRetireHosts(ids []string) error {
+	resp, err := c.PostJSON("/api/v0/hosts/bulk-retire", map[string][]string{
+		"ids": ids,
+	})
+	defer closeResponse(resp)
+	return err
+}
+
 // ListHostMetricNames lists metric names of a host
 func (c *Client) ListHostMetricNames(id string) ([]string, error) {
 	req, err := http.NewRequest("GET", c.urlFor(fmt.Sprintf("/api/v0/hosts/%s/metric-names", id)).String(), nil)
diff --git a/hosts_test.go b/hosts_test.go
index 79bf7a7..7de0c3f 100644
--- a/hosts_test.go
+++ b/hosts_test.go
@@ -3,6 +3,7 @@ package mackerel
 import (
 	"encoding/json"
 	"fmt"
+	"io"
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
@@ -433,6 +434,98 @@ func TestRetireHost_NotFound(t *testing.T) {
 	}
 }
 
+func TestBulkRetireHosts(t *testing.T) {
+	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
+		if req.URL.Path != "/api/v0/hosts/bulk-retire" {
+			t.Error("request URL should be /api/v0/hosts/bulk-retire but: ", req.URL.Path)
+		}
+
+		if req.Method != "POST" {
+			t.Error("request method should be POST but: ", req.Method)
+		}
+
+		body, _ := io.ReadAll(req.Body)
+		var data struct {
+			IDs []string `json:"ids"`
+		}
+		if err := json.Unmarshal(body, &data); err != nil {
+			t.Error("request body should be decoded as json: ", string(body))
+		}
+
+		if reflect.DeepEqual(data.IDs, []string{"123456ABCD", "789012EFGH"}) != true {
+			t.Errorf("request IDs should be []string{\"123456ABCD\", \"789012EFGH\"} but: %+v", data.IDs)
+		}
+
+		respJSON, _ := json.Marshal(map[string]bool{
+			"success": true,
+		})
+
+		res.Header()["Content-Type"] = []string{"application/json"}
+		res.WriteHeader(http.StatusOK)
+		fmt.Fprint(res, string(respJSON))
+	}))
+	defer ts.Close()
+
+	client, _ := NewClientWithOptions("dummy-key", ts.URL, false)
+
+	ids := []string{"123456ABCD", "789012EFGH"}
+	if err := client.BulkRetireHosts(ids); err != nil {
+		t.Error("error should be nil but: ", err)
+	}
+}
+
+func TestBulkRetireHosts_NotFound(t *testing.T) {
+	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
+		if req.URL.Path != "/api/v0/hosts/bulk-retire" {
+			t.Error("request URL should be /api/v0/hosts/bulk-retire but: ", req.URL.Path)
+		}
+
+		if req.Method != "POST" {
+			t.Error("request method should be POST but: ", req.Method)
+		}
+
+		body, _ := io.ReadAll(req.Body)
+		var data struct {
+			IDs []string `json:"ids"`
+		}
+		if err := json.Unmarshal(body, &data); err != nil {
+			t.Error("request body should be decoded as json: ", string(body))
+		}
+
+		if expectIDs := []string{"123456ABCD", "789012EFGH"}; reflect.DeepEqual(data.IDs, expectIDs) != true {
+			t.Errorf("request IDs should be %+v but: %+v", expectIDs, data.IDs)
+		}
+
+		respJSON, _ := json.Marshal(map[string]string{
+			"error": "Hosts not found.",
+		})
+
+		res.Header()["Content-Type"] = []string{"application/json"}
+		res.WriteHeader(http.StatusNotFound)
+		fmt.Fprint(res, string(respJSON))
+	}))
+	defer ts.Close()
+
+	client, _ := NewClientWithOptions("dummy-key", ts.URL, false)
+
+	ids := []string{"123456ABCD", "789012EFGH"}
+	err := client.BulkRetireHosts(ids)
+	if err == nil {
+		t.Error("error should not be nil")
+	}
+
+	apiErr, ok := err.(*APIError)
+	if !ok {
+		t.Error("error should be APIError")
+	}
+	if expectStatus := http.StatusNotFound; apiErr.StatusCode != expectStatus {
+		t.Errorf("api error StatusCode should be %d but got %d", expectStatus, apiErr.StatusCode)
+	}
+	if expect := "API request failed: Hosts not found."; apiErr.Error() != expect {
+		t.Errorf("api error string should be \"%s\" but got \"%s\"", expect, apiErr.Error())
+	}
+}
+
 func TestListHostMetricNames(t *testing.T) {
 	ts := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
 		if req.URL.Path != "/api/v0/hosts/9rxGOHfVF8F/metric-names" {