fix tests

src/characters/cbd-recipient.ts

@@ -120,9 +120,14 @@ export class CbdTDecDecrypter {
  variant: number,
  expectedRitualId: number
  ) {
- const decryptedResponses = Object.entries(encryptedResponses).map(
- ([ursula, encryptedResponse]) =>
- encryptedResponse.decrypt(sessionSharedSecret[ursula])
+ const decryptedResponses = Object.entries(sessionSharedSecret).map(
+ ([ursula, sharedSecret]) => {
+ const encryptedResponse = encryptedResponses[ursula];
+ if (!encryptedResponse) {
+ throw new Error(`Missing encrypted response from ${ursula}`);
+ }
+ return encryptedResponse.decrypt(sharedSecret);
+ }
  );
 
  const ritualIds = decryptedResponses.map(({ ritualId }) => ritualId);

test/unit/cbd-strategy.test.ts

@@ -49,7 +49,7 @@ const makeCbdStrategy = async () => {
 async function makeDeployedCbdStrategy() {
  const strategy = await makeCbdStrategy();
 
- const mockedDkg = fakeDkgFlow(variant, 0);
+ const mockedDkg = fakeDkgFlow(variant, 0, 4, 4);
  const mockedDkgRitual = fakeDkgRitual(mockedDkg, mockedDkg.threshold);
  const web3Provider = fakeWeb3Provider(aliceSecretKey.toBEBytes());
  const getUrsulasSpy = mockGetUrsulas(ursulas);

test/utils.ts

@@ -216,8 +216,8 @@ export const mockDetectEthereumProvider = () => {
 export const fakeDkgFlow = (
  variant: FerveoVariant | FerveoVariant.Precomputed,
  ritualId: number,
- sharesNum = 4,
- threshold = 3
+ sharesNum: number,
+ threshold: number
 ) => {
  if (
  variant !== FerveoVariant.Simple &&
@@ -358,9 +358,11 @@ export const fakeDkgTDecFlowE2e = (
  variant: FerveoVariant,
  message = toBytes('fake-message'),
  aad = toBytes('fake-aad'),
- ritualId = 0
+ ritualId = 0,
+ sharesNum = 4,
+ threshold = 4
 ) => {
- const ritual = fakeDkgFlow(variant, ritualId, 4, 3);
+ const ritual = fakeDkgFlow(variant, ritualId, sharesNum, threshold);
 
  // In the meantime, the client creates a ciphertext and decryption request
  const ciphertext = ferveoEncrypt(message, aad, ritual.dkg.publicKey());