cleanup

AztecProtocol · May 10, 2024 · 28084d6 · 28084d6
1 parent 218fde6
commit 28084d6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 48 deletions.
diff --git a/noir-projects/noir-contracts/contracts/key_registry_contract/src/main.nr b/noir-projects/noir-contracts/contracts/key_registry_contract/src/main.nr
@@ -60,10 +60,7 @@ contract KeyRegistry {
             outgoing_public_key,
             tagging_public_key
         );
-        let computed_address = AztecAddress::compute(
-            public_keys_hash,
-            partial_address
-        );
+        let computed_address = AztecAddress::compute(public_keys_hash, partial_address);
 
         assert(computed_address.eq(address), "Computed address does not match supplied address");
 

diff --git a/yarn-project/circuits.js/src/structs/complete_address.ts b/yarn-project/circuits.js/src/structs/complete_address.ts
@@ -58,30 +58,7 @@ export class CompleteAddress {
   }
 
   static random(): CompleteAddress {
-    // TODO: the following should be cleaned up
-    const secretKey = Fr.random();
-    const partialAddress = Fr.random();
-    const address = computeContractAddressFromPartial({ secretKey, partialAddress });
-    const {
-      masterNullifierPublicKey,
-      masterIncomingViewingPublicKey,
-      masterOutgoingViewingPublicKey,
-      masterTaggingPublicKey,
-    } = deriveKeys(secretKey);
-    return new CompleteAddress(
-      address,
-      masterNullifierPublicKey,
-      masterIncomingViewingPublicKey,
-      masterOutgoingViewingPublicKey,
-      masterTaggingPublicKey,
-      partialAddress,
-    );
-  }
-
-  static fromRandomSecretKey() {
-    const secretKey = Fr.random();
-    const partialAddress = Fr.random();
-    return { secretKey, completeAddress: CompleteAddress.fromSecretKeyAndPartialAddress(secretKey, partialAddress) };
+    return this.fromSecretKeyAndPartialAddress(Fr.random(), Fr.random());
   }
 
   static fromSecretKeyAndPartialAddress(secretKey: Fr, partialAddress: Fr): CompleteAddress {
@@ -107,21 +84,7 @@ export class CompleteAddress {
     instance: Parameters<typeof computePartialAddress>[0],
   ): CompleteAddress {
     const partialAddress = computePartialAddress(instance);
-    const address = computeContractAddressFromPartial({ secretKey, partialAddress });
-    const {
-      masterNullifierPublicKey,
-      masterIncomingViewingPublicKey,
-      masterOutgoingViewingPublicKey,
-      masterTaggingPublicKey,
-    } = deriveKeys(secretKey);
-    return new CompleteAddress(
-      address,
-      masterNullifierPublicKey,
-      masterIncomingViewingPublicKey,
-      masterOutgoingViewingPublicKey,
-      masterTaggingPublicKey,
-      partialAddress,
-    );
+    return CompleteAddress.fromSecretKeyAndPartialAddress(secretKey, partialAddress);
   }
 
   /** Throws if the address is not correctly derived from the public key and partial address.*/

diff --git a/yarn-project/end-to-end/src/e2e_deploy_contract/deploy_test.ts b/yarn-project/end-to-end/src/e2e_deploy_contract/deploy_test.ts
@@ -7,7 +7,7 @@ import {
   type ContractArtifact,
   type ContractBase,
   type DebugLogger,
-  type Fr,
+  Fr,
   type PXE,
   type Wallet,
   createDebugLogger,
@@ -81,10 +81,8 @@ export class DeployTest {
   }
 
   async registerRandomAccount(): Promise<AztecAddress> {
-    const pxe = this.pxe;
-    const { completeAddress: owner, secretKey } = CompleteAddress.fromRandomSecretKey();
-    await pxe.registerAccount(secretKey, owner.partialAddress);
-    return owner.address;
+    const completeAddress = await this.pxe.registerAccount(Fr.random(), Fr.random());
+    return completeAddress.address;
   }
 }