From ca2448bd2fe4270dda31e5f1e7d5ead30358f586 Mon Sep 17 00:00:00 2001 From: Michael Turner Date: Thu, 1 Aug 2024 13:17:44 -0400 Subject: [PATCH] Update key provider test --- sdk/tests/key-provider.test.ts | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/sdk/tests/key-provider.test.ts b/sdk/tests/key-provider.test.ts index 55d4bee46..4e6aae9ad 100644 --- a/sdk/tests/key-provider.test.ts +++ b/sdk/tests/key-provider.test.ts @@ -35,7 +35,7 @@ describe('KeyProvider', () => { expect(cachedVerifyingKey).toBeInstanceOf(Uint8Array); // Ensure the functionKeys method to get the keys and that the cache is used to do so - const [fetchedProvingKey, fetchedVerifyingKey] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.transfer_private.prover, CREDITS_PROGRAM_KEYS.transfer_private.verifier) + const [fetchedProvingKey, fetchedVerifyingKey] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.transfer_private) expect(keyProvider.cache.size).toBe(2); expect(fetchedProvingKey).toBeInstanceOf(ProvingKey); expect(fetchedVerifyingKey).toBeInstanceOf(VerifyingKey); @@ -51,18 +51,18 @@ describe('KeyProvider', () => { it.skip("Should not fetch offline keys that haven't already been stored", async () => { // Download the credits.aleo function keys - const [bondPublicProver, bondPublicVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.bond_public.prover, CREDITS_PROGRAM_KEYS.bond_public.verifier, CREDITS_PROGRAM_KEYS.bond_public.locator); - const [claimUnbondPublicProver, claimUnbondVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.claim_unbond_public.prover, CREDITS_PROGRAM_KEYS.claim_unbond_public.verifier, CREDITS_PROGRAM_KEYS.claim_unbond_public.locator); - const [feePrivateProver, feePrivateVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.fee_private.prover, CREDITS_PROGRAM_KEYS.fee_private.verifier, CREDITS_PROGRAM_KEYS.fee_private.locator); - const [feePublicProver, feePublicVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.fee_public.prover, CREDITS_PROGRAM_KEYS.fee_public.verifier, CREDITS_PROGRAM_KEYS.fee_public.locator); - const [joinProver, joinVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.join.prover, CREDITS_PROGRAM_KEYS.join.verifier, CREDITS_PROGRAM_KEYS.join.locator); - const [setValidatorStateProver, setValidatorStateVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.set_validator_state.prover, CREDITS_PROGRAM_KEYS.set_validator_state.verifier, CREDITS_PROGRAM_KEYS.set_validator_state.locator); - const [splitProver, splitVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.split.prover, CREDITS_PROGRAM_KEYS.split.verifier, CREDITS_PROGRAM_KEYS.split.locator); - const [transferPrivateProver, transferPrivateVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.transfer_private.prover, CREDITS_PROGRAM_KEYS.transfer_private.verifier, CREDITS_PROGRAM_KEYS.transfer_private.locator); - const [transferPrivateToPublicProver, transferPrivateToPublicVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.transfer_private_to_public.prover, CREDITS_PROGRAM_KEYS.transfer_private_to_public.verifier, CREDITS_PROGRAM_KEYS.transfer_private_to_public.locator); - const [transferPublicProver, transferPublicVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.transfer_public.prover, CREDITS_PROGRAM_KEYS.transfer_public.verifier, CREDITS_PROGRAM_KEYS.transfer_public.locator); - const [transferPublicToPrivateProver, transferPublicToPrivateVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.transfer_public_to_private.prover, CREDITS_PROGRAM_KEYS.transfer_public_to_private.verifier, CREDITS_PROGRAM_KEYS.transfer_public_to_private.locator); - const [unbondPublicProver, unbondPublicVerifier] = await keyProvider.fetchRemoteKeys(CREDITS_PROGRAM_KEYS.unbond_public.prover, CREDITS_PROGRAM_KEYS.unbond_public.verifier, CREDITS_PROGRAM_KEYS.unbond_public.locator); + const [bondPublicProver, bondPublicVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.bond_public); + const [claimUnbondPublicProver, claimUnbondVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.claim_unbond_public); + const [feePrivateProver, feePrivateVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.fee_private); + const [feePublicProver, feePublicVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.fee_public); + const [joinProver, joinVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.join); + const [setValidatorStateProver, setValidatorStateVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.set_validator_state); + const [splitProver, splitVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.split); + const [transferPrivateProver, transferPrivateVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.transfer_private); + const [transferPrivateToPublicProver, transferPrivateToPublicVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.transfer_private_to_public); + const [transferPublicProver, transferPublicVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.transfer_public); + const [transferPublicToPrivateProver, transferPublicToPrivateVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.transfer_public_to_private); + const [unbondPublicProver, unbondPublicVerifier] = await keyProvider.fetchCreditsKeys(CREDITS_PROGRAM_KEYS.unbond_public); // Ensure the insertion methods work as expected without throwing an exception offlineKeyProvider.insertBondPublicKeys(bondPublicProver);