Stub out client tests

bryphe-coder · bryphe-coder · commit 4dcd1fd0a441 · 2022-01-24T18:33:17.000Z
diff --git a/codersdk/client.go b/codersdk/client.go
@@ -47,6 +47,24 @@ func (c *Client) SetSessionToken(token string) error {
 	return nil
 }
 
+// ClearSessionToken clears tokens and authentication context in the current cline.t
+func (c *Client) ClearSessionToken() error {
+	//  If there is no cookie jar defined, just leave it be
+	if c.httpClient.Jar == nil {
+		return nil
+	}
+
+	// Otherwise, just clear it out
+	var err error
+	c.httpClient.Jar, err = cookiejar.New(nil)
+
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // request performs an HTTP request with the body provided.
 // The caller is responsible for closing the response body.
 func (c *Client) request(ctx context.Context, method, path string, body interface{}) (*http.Response, error) {
diff --git a/codersdk/client_test.go b/codersdk/client_test.go
@@ -0,0 +1,30 @@
+package codersdk_test
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/codersdk"
+)
+
+var (
+	serverURL = &url.URL{
+		Scheme: "https",
+		Host:   "coder.com",
+	}
+)
+
+func TestClient(t *testing.T) {
+	t.Parallel()
+	t.Run("SetClearSessionToken", func(t *testing.T) {
+		t.Parallel()
+		client := codersdk.New(serverURL)
+		err := client.SetSessionToken("test-session")
+		require.NoError(t, err, "Setting a session token should be successful")
+
+		err = client.ClearSessionToken()
+		require.NoError(t, err, "Clearing a session token should be successful")
+	})
+}
diff --git a/codersdk/users.go b/codersdk/users.go
@@ -44,6 +44,19 @@ func (c *Client) LoginWithPassword(ctx context.Context, req coderd.LoginWithPass
 	return resp, nil
 }
 
+// Logout calls the /logout API
+// Call `ClearSessionToken()` to clear the session token of the client.
+func (c *Client) Logout(ctx context.Context) error {
+	// Since `LoginWithPassword` doesn't actually set a SessionToken
+	// (it requires a call to SetSessionToken), this is essentially a no-op
+	res, err := c.request(ctx, http.MethodPost, "/api/v2/logout", nil)
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+	return nil
+}
+
 // User returns a user for the ID provided.
 // If the ID string is empty, the current user will be returned.
 func (c *Client) User(ctx context.Context, id string) (coderd.User, error) {
diff --git a/codersdk/users_test.go b/codersdk/users_test.go
@@ -47,4 +47,12 @@ func TestUsers(t *testing.T) {
 		require.NoError(t, err)
 		require.Len(t, orgs, 1)
 	})
+
+	t.Run("LogoutIsSuccessful", func(t *testing.T) {
+		t.Parallel()
+		server := coderdtest.New(t)
+		_ = server.RandomInitialUser(t)
+		err := server.Client.Logout(context.Background())
+		require.NoError(t, err)
+	})
 }
