go/sliding-sync
go/sliding-sync
1
0
Fork 0
mirror of https://github.com/matrix-org/sliding-sync.git synced 2025-03-10 13:37:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Call checkRequest earlier

Browse Source
This commit is contained in:
David Robertson 2023-09-05 18:16:27 +01:00
parent e152ce9ae7
commit cb3b588d3b
No known key found for this signature in database
GPG Key ID: 903ECE108A39DEDD
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests-integration/v3_test.go
View File

@ -54,6 +54,9 @@ type testV2Server struct {
// received from pollers, but before the response is generated. This allows us to
// confirm that the proxy is polling the homeserver's v2 sync endpoint in the
// manner that we expect.
//
// checkRequest is called before we lookup a user for the given token. Tests can
// use this to invalidate the token right before a poll is made.
checkRequest func(token string, req *http.Request)
mu *sync.Mutex
tokenToUser map[string]string
@ -253,6 +256,12 @@ func runTestV2Server(t testutils.TestBenchInterface) *testV2Server {
})
r.HandleFunc("/_matrix/client/r0/sync", func(w http.ResponseWriter, req *http.Request) {
token := strings.TrimPrefix(req.Header.Get("Authorization"), "Bearer ")
server.mu.Lock()
check := server.checkRequest
server.mu.Unlock()
if check != nil {
check(token, req)
}
userID := server.userID(token)
if userID == "" {
w.WriteHeader(401)
@ -264,12 +273,6 @@ func runTestV2Server(t testutils.TestBenchInterface) *testV2Server {
server.mu.Unlock()
return
}
server.mu.Lock()
check := server.checkRequest
server.mu.Unlock()
if check != nil {
check(token, req)
}
resp := server.nextResponse(userID, token)
body, err := json.Marshal(resp)
if err != nil {