Other: Fix flaky cookies test

internal/bridge/mocks.go

@@ -1,6 +1,8 @@
 package bridge
 
 import (
+	"net/http"
+	"net/url"
 	"os"
 	"testing"
 
@@ -41,6 +43,24 @@ func (mocks *Mocks) Close() {
 	close(mocks.TLSIssueCh)
 }
 
+type TestCookieJar struct {
+	cookies map[string][]*http.Cookie
+}
+
+func NewTestCookieJar() *TestCookieJar {
+	return &TestCookieJar{
+		cookies: make(map[string][]*http.Cookie),
+	}
+}
+
+func (j *TestCookieJar) SetCookies(u *url.URL, cookies []*http.Cookie) {
+	j.cookies[u.Host] = cookies
+}
+
+func (j *TestCookieJar) Cookies(u *url.URL) []*http.Cookie {
+	return j.cookies[u.Host]
+}
+
 type TestLocationsProvider struct {
 	config, cache string
 }