Improve key address generation test

packages/massa-web3/test/experimental/unit/address.spec.ts

@@ -1,16 +1,16 @@
 import { Account } from '../../../src/experimental/account'
-import { Address } from '../../../src/experimental/basicElements'
 import { Version } from '../../../src/experimental/crypto/interfaces/versioner'
-
+import { Address } from '../../../src/experimental/basicElements'
 import { Address as LegacyAddress } from '../../../src/utils/keyAndAddresses'
 
 const unsupported_version = -1 as Version
-
+const contractAddress = 'AS1eK3SEXGDAWN6pZhdr4Q7WJv6UHss55EB14hPy4XqBpiktfPu6'
 class TestAddress extends Address {
   public static callInitFromVersion(version: Version = Version.V0): Address {
     return this.initFromVersion(version)
   }
 }
+
 describe('Address tests', () => {
   let account: Account
 
@@ -39,19 +39,17 @@ describe('Address tests', () => {
 
   test('fromString throws error for invalid address string', () => {
     expect(() => Address.fromString('invalid_address_string')).toThrow(
-      'invalid address string:'
+      /invalid address string:/
     )
   })
 
   test('toString returns string with user prefix for EOA', () => {
     const address = Address.fromPublicKey(account.publicKey)
-    address.isEOA = true
     expect(address.toString()).toMatch(/^AU/)
   })
 
   test('toString returns string with contract prefix for contract account', () => {
-    const address = Address.fromPublicKey(account.publicKey)
-    address.isEOA = false
+    const address = Address.fromString(contractAddress)
     expect(address.toString()).toMatch(/^AS/)
   })
 
@@ -64,9 +62,7 @@ describe('Address tests', () => {
   })
 
   test('toBytes returns bytes with contract prefix for contract account', () => {
-    const address = Address.fromPublicKey(account.publicKey)
-    address.isEOA = false
-
+    const address = Address.fromString(contractAddress)
     const bytes = address.toBytes()
     // The first byte should be 1 for contract account
     expect(bytes[0]).toBe(1)

packages/massa-web3/test/experimental/unit/key.spec.ts

@@ -4,6 +4,8 @@ import {
   PublicKey,
 } from '../../../src/experimental/basicElements/keys'
 import { Version } from '../../../src/experimental/crypto/interfaces/versioner'
+import { SecretKey as LegacyPrivateKey } from '../../../src/utils/keyAndAddresses'
+import { Address as LegacyAddress } from '../../../src/utils/keyAndAddresses'
 
 class TestPublicKey extends PublicKey {
   public static callInitFromVersion(version: Version) {
@@ -39,10 +41,23 @@ describe('PrivateKey and PublicKey tests', () => {
   })
 
   describe('Address Generation', () => {
-    test('PublicKey getAddress', () => {
-      // TODO be sure address is right
-      const address = publicKey.getAddress()
-      expect(address).toBeInstanceOf(Address)
+    test('PublicKey getAddress', async () => {
+      // Generate address from Legacy keys objects
+      const legacyPrivKey = new LegacyPrivateKey(
+        'S1eK3SEXGDAWN6pZhdr4Q7WJv6UHss55EB14hPy4XqBpiktfPu6'
+      )
+      const legacyPubKey = await legacyPrivKey.getPublicKey()
+      const legacyAddress = LegacyAddress.fromPublicKey(legacyPubKey)
+
+      // Generate address from new keys objects
+      const privKey = PrivateKey.fromString(
+        'S1eK3SEXGDAWN6pZhdr4Q7WJv6UHss55EB14hPy4XqBpiktfPu6'
+      )
+      const pubKey = await PublicKey.fromPrivateKey(privKey)
+      const address = Address.fromPublicKey(pubKey)
+
+      // Verify that the addresses are the same
+      expect(address.toString()).toBe(legacyAddress.base58Encoded)
     })
   })