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

Other: Fix user logout hangs due to sync

Browse Source
This commit is contained in:
James Houlahan
2022-10-20 12:35:00 +02:00
parent bf29090ffa
commit 1729c085c7
4 changed files with 51 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
internal/bridge/user_test.go
View File

@ -306,6 +306,10 @@ func TestBridge_FailLoginRecover(t *testing.T) {
// We should now be able to log the user in.
withBridge(ctx, t, s.GetHostURL(), netCtl, locator, storeKey, func(bridge *bridge.Bridge, mocks *bridge.Mocks) {
require.NoError(t, getErr(bridge.LoginFull(ctx, username, password, nil, nil)))
// The user should be there, now connected.
require.Equal(t, []string{userID}, bridge.GetUserIDs())
require.Equal(t, []string{userID}, getConnectedUserIDs(t, bridge))
})
})
}
@ -400,6 +404,20 @@ func TestBridge_AddressMode(t *testing.T) {
})
}
func TestBridge_LoginLogoutRepeated(t *testing.T) {
withEnv(t, func(ctx context.Context, s *server.Server, netCtl *liteapi.NetCtl, locator bridge.Locator, storeKey []byte) {
withBridge(ctx, t, s.GetHostURL(), netCtl, locator, storeKey, func(bridge *bridge.Bridge, mocks *bridge.Mocks) {
for i := 0; i < 10; i++ {
// Log the user in.
userID := must(bridge.LoginFull(ctx, username, password, nil, nil))
// Log the user out.
require.NoError(t, bridge.LogoutUser(ctx, userID))
}
})
})
}
// getErr returns the error that was passed to it.
func getErr[T any](val T, err error) error {
return err