Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-18 08:06:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Other(test): Ensure calls are protected by mutex

Browse Source
This commit is contained in:
James Houlahan
2022-10-25 10:26:34 +02:00
parent 3e54885ea0
commit 5adbf74cbe
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/bridge/bridge_test.go
View File

@ -22,6 +22,7 @@ import (
"crypto/tls" "crypto/tls"
"net/http" "net/http"
"os" "os"
"sync"
"testing" "testing"
"time" "time"
@ -133,9 +134,15 @@ func TestBridge_Focus(t *testing.T) {
func TestBridge_UserAgent(t *testing.T) { func TestBridge_UserAgent(t *testing.T) {
withEnv(t, func(ctx context.Context, s *server.Server, netCtl *liteapi.NetCtl, locator bridge.Locator, vaultKey []byte) { withEnv(t, func(ctx context.Context, s *server.Server, netCtl *liteapi.NetCtl, locator bridge.Locator, vaultKey []byte) {
var calls []server.Call var (
calls []server.Call
lock sync.Mutex
)
s.AddCallWatcher(func(call server.Call) { s.AddCallWatcher(func(call server.Call) {
lock.Lock()
defer lock.Unlock()
calls = append(calls, call) calls = append(calls, call)
}) })
@ -150,6 +157,9 @@ func TestBridge_UserAgent(t *testing.T) {
_, err := bridge.LoginFull(context.Background(), username, password, nil, nil) _, err := bridge.LoginFull(context.Background(), username, password, nil, nil)
require.NoError(t, err) require.NoError(t, err)
lock.Lock()
defer lock.Unlock()
// Assert that the user agent was sent to the API. // Assert that the user agent was sent to the API.
require.Contains(t, calls[len(calls)-1].RequestHeader.Get("User-Agent"), bridge.GetCurrentUserAgent()) require.Contains(t, calls[len(calls)-1].RequestHeader.Get("User-Agent"), bridge.GetCurrentUserAgent())
}) })