danvics/Pulse
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/integration/api/update_flow_test.go
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) resp := doJSONRequest(t, client, "POST", baseURL+"/api/login", payload)
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { 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) 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 client != nil && client.Jar != nil && methodRequiresCSRF(method) {
if token := csrfTokenForURL(client.Jar, req.URL); token != "" { if token := csrfTokenForURL(client.Jar, req.URL); token != "" {
req.Header.Set("X-CSRF-Token", token) req.Header.Set("X-CSRF-Token", token)
} else {
req.Header.Del("X-CSRF-Token")
} }
} }
resp, err := client.Do(req) resp, err := client.Do(req)
@ -246,3 +251,10 @@ func csrfTokenForURL(jar http.CookieJar, target *url.URL) string {
} }
return "" return ""
} }
func clearCookies(jar http.CookieJar, target *url.URL) {
if jar == nil || target == nil {
return
}
jar.SetCookies(target, []*http.Cookie{})
}