gsadmin/pulse
1
0
Fork 0
mirror of https://github.com/rcourtman/Pulse.git synced 2026-09-11 14:00:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make API update test resilient to stale cookies

Browse Source
This commit is contained in:
rcourtman
2025-11-12 21:32:33 +00:00
parent e3952a073c
commit 4f950de29d
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/api/update_flow_test.go
+12
View File
@@ -107,6 +107,9 @@ func login(t *testing.T, client *http.Client, baseURL, username, password string
resp := doJSONRequest(t, client, "POST", baseURL+"/api/login", payload)
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
if client != nil && client.Jar != nil {
clearCookies(client.Jar, resp.Request.URL)
}
t.Fatalf("login failed with status %s", resp.Status)
}
}
@@ -203,6 +206,8 @@ func doRequest(t *testing.T, client *http.Client, method, endpoint string, body
if client != nil && client.Jar != nil && methodRequiresCSRF(method) {
if token := csrfTokenForURL(client.Jar, req.URL); token != "" {
req.Header.Set("X-CSRF-Token", token)
} else {
req.Header.Del("X-CSRF-Token")
}
}
resp, err := client.Do(req)
@@ -246,3 +251,10 @@ func csrfTokenForURL(jar http.CookieJar, target *url.URL) string {
}
return ""
}
func clearCookies(jar http.CookieJar, target *url.URL) {
if jar == nil || target == nil {
return
}
jar.SetCookies(target, []*http.Cookie{})
}