Merge branch 'master' of https://github.com/LimeChain/as-scale-codec

LimeChain · Jul 8, 2020 · c1fdc5c · c1fdc5c
2 parents 339a008 + 03439b6
commit c1fdc5c
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/assembly/Hash.ts b/assembly/Hash.ts
@@ -65,4 +65,16 @@ export class Hash extends Array<u8> {
     static fromU8a (input: u8[]): Hash {
         return new Hash(input);
     }
+
+    @inline @operator('==')
+    static eq(a: Hash, b: Hash): bool {
+        let areEqual = true;
+        for (let i = 0; i < a.length; i++) {
+            if (a[i] != b[i]) {
+                areEqual = false;
+                break;
+            }
+        }
+        return areEqual;
+    }
 }
diff --git a/assembly/Int/CompactInt.ts b/assembly/Int/CompactInt.ts
@@ -66,4 +66,9 @@ export class CompactInt implements Codec {
         const decodedData = Bytes.decodeCompactInt(value);
         return new CompactInt(decodedData.value);
     }
+
+    @inline @operator('==')
+    static eq(a: CompactInt, b: CompactInt): bool {
+        return a.value == b.value;
+    }
 }
diff --git a/assembly/__tests__/Hash.spec.ts b/assembly/__tests__/Hash.spec.ts
@@ -72,4 +72,18 @@ describe("Hash", () => {
         hash = new Hash([0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff]);
         expect<string>(hash.toString()).toStrictEqual('0x255000000000000000000000000000000255');
     });
+
+    it("equals should work", () => {
+        let hash: Hash = new Hash([0xff]);
+        let hash2: Hash = new Hash([0xff]);
+
+        assert(hash == hash2, "hashes are not equal")
+    })
+
+    it("should not be equal when hashes are different", () => {
+        let hash1: Hash = new Hash([0x01]);
+        let hash2: Hash = new Hash([0xff]);
+
+        assert(hash1 != hash2, "hashes should be different");
+    })
 });