Add Logout test

bryphe-coder · bryphe-coder · commit 52d0136b89a0 · 2022-01-24T17:04:43.000Z
diff --git a/coderd/users_test.go b/coderd/users_test.go
@@ -2,12 +2,14 @@ package coderd_test
 
 import (
 	"context"
+	"net/http"
 	"testing"
 
 	"github.com/stretchr/testify/require"
 
 	"github.com/coder/coder/coderd"
 	"github.com/coder/coder/coderd/coderdtest"
+	"github.com/coder/coder/httpmw"
 )
 
 func TestUsers(t *testing.T) {
@@ -74,3 +76,30 @@ func TestUsers(t *testing.T) {
 		require.Len(t, orgs, 1)
 	})
 }
+
+func TestLogout(t *testing.T) {
+	t.Parallel()
+
+	t.Run("LogoutShouldClearCookie", func(t *testing.T) {
+		t.Parallel()
+
+		server := coderdtest.New(t)
+		fullURL, err := server.URL.Parse("/api/v2/logout")
+		require.NoError(t, err, "Server URL should parse successfully")
+
+		req, err := http.NewRequestWithContext(context.Background(), http.MethodPost, fullURL.String(), nil)
+		require.NoError(t, err, "/logout request construction should succeed")
+
+		httpClient := &http.Client{}
+
+		response, err := httpClient.Do(req)
+		require.NoError(t, err, "/logout request should succeed")
+		response.Body.Close()
+
+		cookies := response.Cookies()
+		require.Len(t, cookies, 1, "Exactly one cookie should be returned")
+
+		require.Equal(t, cookies[0].Name, httpmw.AuthCookie, "Cookie should be the auth cookie")
+		require.Equal(t, cookies[0].MaxAge, -1, "Cookie should be set to delete")
+	})
+}
