Skip to content

Commit 384b718

Browse files
committed
Reuse MessageSigner instance in test to avoid worker startup overhead
1 parent 5f158ab commit 384b718

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

packages/sdk/test/unit/waitForAssignmentsToPropagate.test.ts

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import { StreamID, toStreamID, toStreamPartID, utf8ToBinary, wait } from '@streamr/utils'
1+
import { StreamID, toStreamID, toStreamPartID, UserID, utf8ToBinary, wait } from '@streamr/utils'
22
import range from 'lodash/range'
33
import shuffle from 'lodash/shuffle'
44
import { MessageSigner } from '../../src/signature/MessageSigner'
@@ -7,7 +7,6 @@ import { waitForAssignmentsToPropagate } from '../../src/utils/waitForAssignment
77
import { createRandomIdentity, mockLoggerFactory } from '../test-utils/utils'
88
import { MessageID } from './../../src/protocol/MessageID'
99
import { StreamMessage, StreamMessageType } from './../../src/protocol/StreamMessage'
10-
import { Identity } from '../../src/identity/Identity'
1110
import { ContentType, EncryptionType, SignatureType } from '@streamr/trackerless-network'
1211

1312
const RACE_TIMEOUT_IN_MS = 20
@@ -22,11 +21,12 @@ describe(waitForAssignmentsToPropagate, () => {
2221
let messageStream: MessageStream
2322
let propagatePromiseState: 'rejected' | 'resolved' | 'pending'
2423
let propagatePromise: Promise<any>
25-
let identity: Identity
24+
let messageSigner: MessageSigner
25+
let publisherId: UserID
2626

2727
async function makeMsg(ts: number, content: unknown): Promise<StreamMessage> {
28-
return new MessageSigner(identity).createSignedMessage({
29-
messageId: new MessageID(toStreamID('assignmentStreamId'), 0, ts, 0, await identity.getUserId(), 'msgChain'),
28+
return messageSigner.createSignedMessage({
29+
messageId: new MessageID(toStreamID('assignmentStreamId'), 0, ts, 0, publisherId, 'msgChain'),
3030
messageType: StreamMessageType.MESSAGE,
3131
content: utf8ToBinary(JSON.stringify(content)),
3232
contentType: ContentType.JSON,
@@ -46,7 +46,13 @@ describe(waitForAssignmentsToPropagate, () => {
4646
}
4747

4848
beforeAll(async () => {
49-
identity = await createRandomIdentity()
49+
const identity = await createRandomIdentity()
50+
messageSigner = new MessageSigner(identity)
51+
publisherId = await identity.getUserId()
52+
})
53+
54+
afterAll(() => {
55+
messageSigner.destroy()
5056
})
5157

5258
beforeEach(() => {

0 commit comments

Comments
 (0)