Fix Entity helpers

Author: leoyvens

index.ts

@@ -308,14 +308,14 @@ export class ByteArray extends Uint8Array {
   }
 
   static fromUTF8(string: String): ByteArray {
-      // AssemblyScript counts a null terminator, we don't want that.
-      let len = string.lengthUTF8 - 1;
-      let utf8 = string.toUTF8();
-      let bytes = new ByteArray(len);
-      for (let i: i32 = 0; i < len; i++) {
-        bytes[i] = load<u8>(utf8 + i)
-      }
-      return bytes;
+    // AssemblyScript counts a null terminator, we don't want that.
+    let len = string.lengthUTF8 - 1
+    let utf8 = string.toUTF8()
+    let bytes = new ByteArray(len)
+    for (let i: i32 = 0; i < len; i++) {
+      bytes[i] = load<u8>(utf8 + i)
+    }
+    return bytes
   }
 
   toHex(): string {
@@ -1008,7 +1008,7 @@ export class Entity extends TypedMap<string, Value> {
     this.set(key, Value.fromBoolean(value))
   }
 
-  setBigDecimal(key, value: BigDecimal): void {
+  setBigDecimal(key: string, value: BigDecimal): void {
     this.set(key, Value.fromBigDecimal(value))
   }
 
@@ -1025,15 +1025,15 @@ export class Entity extends TypedMap<string, Value> {
   }
 
   getBytes(key: string): Bytes {
-    this.get(key).toBytes()
+    return this.get(key).toBytes()
   }
 
   getBoolean(key: string): boolean {
-    this.get(key).toBoolean()
+    return this.get(key).toBoolean()
   }
 
   getBigDecimal(key: string): BigDecimal {
-    this.get(key).toBigDecimal()
+    return this.get(key).toBigDecimal()
   }
 }
 

test/entity.ts

@@ -0,0 +1,14 @@
+import { Entity, Bytes, BigDecimal, BigInt } from 'temp_lib/index'
+
+export function test(): void {
+  let entity = new Entity()
+
+  entity.setBytes('x', new Bytes(1))
+  assert(entity.getBytes('x') == new Bytes(1))
+
+  entity.setBoolean('x', true)
+  assert(entity.getBoolean('x') == true)
+
+  entity.setBigDecimal('x', new BigDecimal(BigInt.fromI32(2)))
+  assert(entity.getBigDecimal('x') !== null)
+}

test/test.js

@@ -20,14 +20,15 @@ fs.copyFileSync('index.ts', 'test/temp_lib/index.ts')
 let output_path = 'test/temp_out/test.wasm'
 
 const env = {
-  abort: function(message, fileName, lineNumber, columnNumber) {
+  abort: function (message, fileName, lineNumber, columnNumber) {
     console.log('aborted')
   },
 }
 
 try {
   testFile('test/test.ts')
   testFile('test/testBytes.ts')
+  testFile('test/entity.ts')
 
   // Cleanup
   fs.unlinkSync('test/temp_lib/index.ts')
@@ -56,9 +57,9 @@ function testFile(path) {
   WebAssembly.instantiate(test_wasm, {
     env,
     index: {
-      'typeConversion.bytesToHex': function() {},
+      'typeConversion.bytesToHex': function () {},
     },
-  }).then(module => {
+  }).then((module) => {
     module.instance.exports.test()
   })
 }