Skip to content

Commit

Permalink
fix_: Flaky backup keypairs test
Browse files Browse the repository at this point in the history
Signed-off-by: Mohamed Javid <[email protected]>
  • Loading branch information
smohamedjavid committed Sep 7, 2024
1 parent c24eba8 commit 7c7cac2
Showing 1 changed file with 14 additions and 10 deletions.
24 changes: 14 additions & 10 deletions protocol/messenger_backup_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import (
"encoding/json"
"fmt"
"reflect"
"slices"
"testing"
"time"

Expand Down Expand Up @@ -737,36 +738,39 @@ func (s *MessengerBackupSuite) TestBackupKeypairs() {
s.Require().NoError(err)
s.Require().True(accounts.SameKeypairsWithDifferentSyncedFrom(seedKp, dbSeedKp2, false, accounts.SyncedFromBackup, accounts.AccountNonOperable))

keypairs, err := bob2.settings.GetAllKeypairs()
s.Require().NoError(err)

// Check whether accounts added event is sent
expectedAddresses := [][]common.Address{}
profileKpWalletAddresses := []common.Address{}
seedKpAddresses := []common.Address{}
expectedAddresses := []common.Address{}
for _, acc := range dbProfileKp2.Accounts {
if acc.Chat {
continue
}
profileKpWalletAddresses = append(profileKpWalletAddresses, common.Address(acc.Address))
expectedAddresses = append(expectedAddresses, common.Address(acc.Address))
}
expectedAddresses = append(expectedAddresses, profileKpWalletAddresses)

for _, acc := range dbSeedKp2.Accounts {
seedKpAddresses = append(seedKpAddresses, common.Address(acc.Address))
expectedAddresses = append(expectedAddresses, common.Address(acc.Address))
}
expectedAddresses = append(expectedAddresses, seedKpAddresses)

for i := 0; i < len(expectedAddresses); i++ {
receivedEventAddresses := []common.Address{}
for i := 0; i < len(keypairs); i++ {
select {
case <-time.After(1 * time.Second):
s.Fail("Timed out waiting for accountsevent")
case event := <-ch:
switch event.Type {
case accountsevent.EventTypeAdded:
s.Require().Len(event.Accounts, len(expectedAddresses[i]))
s.Require().True(reflect.DeepEqual(expectedAddresses[i], event.Accounts))
receivedEventAddresses = append(receivedEventAddresses, event.Accounts...)
}
}
}
sub.Unsubscribe()
s.Require().Equal(len(expectedAddresses), len(receivedEventAddresses))
for _, addr := range receivedEventAddresses {
s.Require().True(slices.Contains(expectedAddresses, addr))
}
}

func (s *MessengerBackupSuite) TestBackupKeycards() {
Expand Down

0 comments on commit 7c7cac2

Please sign in to comment.