add Address.equals(address: Address)

ryanio · ryanio · commit 93c43f76a0ea · 2021-01-21T09:40:34.000-08:00
diff --git a/src/address.ts b/src/address.ts
@@ -75,11 +75,18 @@ export class Address {
     return new Address(generateAddress2(from.buf, salt, initCode))
   }
 
+  /**
+   * Is address equal to another.
+   */
+  equals(address: Address): boolean {
+    return this.buf.equals(address.buf)
+  }
+
   /**
    * Is address zero.
    */
   isZero(): boolean {
-    return this.buf.equals(Address.zero().buf)
+    return this.equals(Address.zero())
   }
 
   /**
diff --git a/test/address.spec.ts b/test/address.spec.ts
@@ -84,4 +84,22 @@ describe('Address', () => {
     addressBuf.fill(0)
     assert.equal(address.toString(), str)
   })
+
+  it('should compare equality properly', () => {
+    const str = '0x2f015c60e0be116b1f0cd534704db9c92118fb6a'
+    const address1 = Address.fromString(str)
+    const address2 = new Address(Buffer.from(str.slice(2), 'hex'))
+    assert.ok(address1.equals(address2))
+    assert.ok(address1.buf.equals(address2.buf))
+
+    const str2 = '0xcd4EC7b66fbc029C116BA9Ffb3e59351c20B5B06'
+    const address3 = Address.fromString(str2)
+    assert.ok(!address1.equals(address3))
+
+    const address3LowerCase = Address.fromString(str2.toLowerCase())
+    assert.ok(address3.equals(address3LowerCase))
+
+    const address4 = Address.zero()
+    assert.ok(!address1.equals(address4))
+  })
 })
