From 0113cc8d0257d06ce3f8ac843a1223626d21a424 Mon Sep 17 00:00:00 2001
From: winebarrel <sugawara@winebarrel.jp>
Date: Fri, 14 Jun 2024 13:29:46 +0900
Subject: [PATCH] Add error test

---
 query_test.go | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/query_test.go b/query_test.go
index 5f145d1..f46a529 100644
--- a/query_test.go
+++ b/query_test.go
@@ -797,6 +797,20 @@ func Test_CreateFavoriteQuery_OK(t *testing.T) {
 	assert.NoError(err)
 }
 
+func Test_CreateFavoriteQuery_Err_5xx(t *testing.T) {
+	assert := assert.New(t)
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+
+	httpmock.RegisterResponder(http.MethodPost, "https://redash.example.com/api/queries/1/favorite", func(req *http.Request) (*http.Response, error) {
+		return httpmock.NewStringResponse(http.StatusInternalServerError, "error"), nil
+	})
+
+	client, _ := redash.NewClient("https://redash.example.com", testRedashAPIKey)
+	err := client.CreateFavoriteQuery(context.Background(), 1)
+	assert.ErrorContains(err, "POST api/queries/1/favorite failed: HTTP status code not OK: 500\nerror")
+}
+
 func Test_ForkQuery_OK(t *testing.T) {
 	assert := assert.New(t)
 	httpmock.Activate()
@@ -906,6 +920,34 @@ func Test_ForkQuery_OK(t *testing.T) {
 	}, res)
 }
 
+func Test_ForkQuery_Err_5xx(t *testing.T) {
+	assert := assert.New(t)
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+
+	httpmock.RegisterResponder(http.MethodPost, "https://redash.example.com/api/queries/1/fork", func(req *http.Request) (*http.Response, error) {
+		return httpmock.NewStringResponse(http.StatusServiceUnavailable, "error"), nil
+	})
+
+	client, _ := redash.NewClient("https://redash.example.com", testRedashAPIKey)
+	_, err := client.ForkQuery(context.Background(), 1)
+	assert.ErrorContains(err, "POST api/queries/1/fork failed: HTTP status code not OK: 503\nerror")
+}
+
+func Test_ForkQuery_IOErr(t *testing.T) {
+	assert := assert.New(t)
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+
+	httpmock.RegisterResponder(http.MethodPost, "https://redash.example.com/api/queries/1/fork", func(req *http.Request) (*http.Response, error) {
+		return testIOErrResp, nil
+	})
+
+	client, _ := redash.NewClient("https://redash.example.com", testRedashAPIKey)
+	_, err := client.ForkQuery(context.Background(), 1)
+	assert.ErrorContains(err, "Read response body failed: IO error")
+}
+
 func Test_GetQueryResultsJSON_OK(t *testing.T) {
 	assert := assert.New(t)
 	httpmock.Activate()