feat(amplify-appsync-simulator): using this.mapper in map.keySet

aws-amplify · Jul 6, 2021 · 98b4482 · 98b4482
1 parent 775b4c1
commit 98b4482
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/packages/amplify-appsync-simulator/src/__tests__/velocity/value-mapper/map.test.ts b/packages/amplify-appsync-simulator/src/__tests__/velocity/value-mapper/map.test.ts
@@ -39,7 +39,10 @@ describe('JavaMap', () => {
   it('entrySet', () => {
     const obj = { foo: 'Foo Value', bar: 'Bar Value' };
     const map = new JavaMap(obj, identityMapper);
-    expect(map.entrySet().toJSON()).toEqual([{ key: 'foo', value: 'Foo Value' }, { key: 'bar', value: 'Bar Value' }]);
+    expect(map.entrySet().toJSON()).toEqual([
+      { key: 'foo', value: 'Foo Value' },
+      { key: 'bar', value: 'Bar Value' },
+    ]);
   });
 
   it('equal', () => {
@@ -65,13 +68,13 @@ describe('JavaMap', () => {
 
   it('keySet', () => {
     const obj = { foo: 'Foo Value', bar: 'Bar Value' };
-    const map = new JavaMap(obj, identityMapper);
+    const map = new JavaMap(obj, mapper);
     expect(map.keySet().toJSON()).toEqual(['foo', 'bar']);
   });
 
   it('keySet returns a JavaArray with each element of type JavaString', () => {
     const obj = { foo: 'Foo Value', bar: 'Bar Value' };
-    const map = new JavaMap(obj, identityMapper);
+    const map = new JavaMap(obj, mapper);
     expect(map.keySet()).toEqual(new JavaArray([new JavaString('foo'), new JavaString('bar')], mapper));
   });
 

diff --git a/packages/amplify-appsync-simulator/src/velocity/value-mapper/map.ts b/packages/amplify-appsync-simulator/src/velocity/value-mapper/map.ts
@@ -1,6 +1,5 @@
 import { JavaArray } from './array';
 import { toJSON } from './to-json';
-import { map } from './mapper';
 
 export class JavaMap {
   private map: Map<string, any>;
@@ -57,7 +56,7 @@ export class JavaMap {
   }
 
   keySet() {
-    return map(Array.from(this.map.keys()));
+    return this.mapper(Array.from(this.map.keys()));
   }
 
   put(key, value) {