API tests

mtojek · mtojek · commit 36b8c1c209d6 · 2023-11-28T12:34:29.000+01:00
diff --git a/coderd/debug.go b/coderd/debug.go
@@ -7,13 +7,14 @@ import (
 	"net/http"
 	"time"
 
+	"golang.org/x/exp/slices"
+	"golang.org/x/xerrors"
+
 	"github.com/coder/coder/v2/coderd/healthcheck"
 	"github.com/coder/coder/v2/coderd/httpapi"
 	"github.com/coder/coder/v2/coderd/httpmw"
 	"github.com/coder/coder/v2/coderd/rbac"
 	"github.com/coder/coder/v2/codersdk"
-	"golang.org/x/exp/slices"
-	"golang.org/x/xerrors"
 )
 
 // @Summary Debug Info Wireguard Coordinator
diff --git a/coderd/debug_test.go b/coderd/debug_test.go
@@ -14,6 +14,7 @@ import (
 	"github.com/coder/coder/v2/coderd/coderdtest"
 	"github.com/coder/coder/v2/coderd/healthcheck"
 	"github.com/coder/coder/v2/coderd/healthcheck/derphealth"
+	"github.com/coder/coder/v2/codersdk"
 	"github.com/coder/coder/v2/testutil"
 )
 
@@ -232,6 +233,52 @@ func TestDebugHealth(t *testing.T) {
 	})
 }
 
+func TestHealthSettings(t *testing.T) {
+	t.Parallel()
+
+	t.Run("InitialState", func(t *testing.T) {
+		t.Parallel()
+
+		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
+		defer cancel()
+
+		// given
+		adminClient := coderdtest.New(t, nil)
+		_ = coderdtest.CreateFirstUser(t, adminClient)
+
+		// when
+		settings, err := adminClient.HealthSettings(ctx)
+		require.NoError(t, err)
+
+		// then
+		require.Equal(t, codersdk.HealthSettings{DismissedHealthchecks: []string{}}, settings)
+	})
+
+	t.Run("Updated", func(t *testing.T) {
+		t.Parallel()
+
+		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
+		defer cancel()
+
+		// given
+		adminClient := coderdtest.New(t, nil)
+		_ = coderdtest.CreateFirstUser(t, adminClient)
+
+		expected := codersdk.HealthSettings{
+			DismissedHealthchecks: []string{healthcheck.SectionDERP, healthcheck.SectionWebsocket},
+		}
+
+		// when
+		err := adminClient.UpdateHealthSettings(ctx, expected)
+		require.NoError(t, err)
+
+		// then
+		settings, err := adminClient.HealthSettings(ctx)
+		require.NoError(t, err)
+		require.Equal(t, expected, settings)
+	})
+}
+
 func TestDebugWebsocket(t *testing.T) {
 	t.Parallel()
 
diff --git a/codersdk/health.go b/codersdk/health.go
@@ -1,9 +1,40 @@
 package codersdk
 
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+)
+
 type HealthSettings struct {
 	DismissedHealthchecks []string `json:"dismissed_healthchecks"`
 }
 
 type UpdateHealthSettings struct {
 	DismissedHealthchecks []string `json:"dismissed_healthchecks"`
 }
+
+func (c *Client) HealthSettings(ctx context.Context) (HealthSettings, error) {
+	res, err := c.Request(ctx, http.MethodGet, "/api/v2/debug/health/settings", nil)
+	if err != nil {
+		return HealthSettings{}, err
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		return HealthSettings{}, ReadBodyAsError(res)
+	}
+	var settings HealthSettings
+	return settings, json.NewDecoder(res.Body).Decode(&settings)
+}
+
+func (c *Client) UpdateHealthSettings(ctx context.Context, settings HealthSettings) error {
+	res, err := c.Request(ctx, http.MethodPut, "/api/v2/debug/health/settings", settings)
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		return ReadBodyAsError(res)
+	}
+	return nil
+}
