Skip to content

Commit f7a7a3e

Browse files
committed
Fix trigger test
1 parent 16879ce commit f7a7a3e

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

rolling-shutter/keyper/epochkghandler/trigger_test.go

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ func TestHandleDecryptionTriggerIntegration(t *testing.T) {
2929

3030
identityPreimage := identitypreimage.Uint64ToIdentityPreimage(50)
3131
keyperIndex := uint64(1)
32+
keyperConfigIndex := int64(1)
3233

3334
testsetup.InitializeEon(ctx, t, dbpool, config, keyperIndex)
3435

@@ -55,22 +56,18 @@ func TestHandleDecryptionTriggerIntegration(t *testing.T) {
5556
IdentityPreimages: []identitypreimage.IdentityPreimage{identityPreimage},
5657
}
5758
decrTrigChan <- broker.NewEvent(trig)
58-
decrTrigChan <- broker.NewEvent(trig)
5959
close(decrTrigChan)
6060
err = group.Wait()
6161
cleanup()
6262
assert.NilError(t, err)
6363

6464
// send decryption key share when first trigger is received
6565
share, err := db.GetDecryptionKeyShare(ctx, database.GetDecryptionKeyShareParams{
66-
Eon: int64(config.GetEon()),
66+
Eon: keyperConfigIndex,
6767
EpochID: identityPreimage.Bytes(),
6868
KeyperIndex: int64(keyperIndex),
6969
})
7070
assert.NilError(t, err)
71-
// although we requested the trigger 2 times, the keyshare should have been
72-
// sent only once
73-
assert.Check(t, len(messaging.SentMessages) == 1)
7471

7572
msg, ok := messaging.SentMessages[0].Message.(*p2pmsg.DecryptionKeyShares)
7673
assert.Check(t, ok)

0 commit comments

Comments
 (0)