chore: fix an array of types

aws-amplify · Aug 1, 2024 · bdff20c · bdff20c
1 parent d2d939d
commit bdff20c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/packages/core/__tests__/singleton/Auth/type.test.ts b/packages/core/__tests__/singleton/Auth/type.test.ts
@@ -4,12 +4,26 @@ describe('type validity', () => {
 	describe('JWT type', () => {
 		it('can contain property that has a value as array of JsonObject', () => {
 			const value: JWT = {
-				payload: { otherProperty: 'value' },
+				payload: { otherProperty: [{ key: '123' }] },
 				toString: () => 'mock',
 			};
 			const a = value.payload.otherProperty as { key: string }[];
 
-			expect(a).toEqual('value');
+			expect(a).toEqual([{ key: '123' }]);
+		});
+
+		it('can contain property that has a value as array of JsonObject and JsonPrimitive', () => {
+			const value: JWT = {
+				payload: { otherProperty: [1, 2, 'hi', { key: 'value' }] },
+				toString: () => 'mock',
+			};
+			const a = value.payload.otherProperty as (
+				| { key: string }
+				| number
+				| string
+			)[];
+
+			expect(a).toEqual([1, 2, 'hi', { key: 'value' }]);
 		});
 	});
 });
diff --git a/packages/core/src/singleton/Auth/types.ts b/packages/core/src/singleton/Auth/types.ts
@@ -20,7 +20,7 @@ interface JwtPayloadStandardFields {
 type JsonPrimitive = null | string | number | boolean;
 
 /** JSON array type */
-type JsonArray = JsonPrimitive[] | JsonObject[];
+type JsonArray = (JsonPrimitive | JsonObject)[];
 
 /** JSON Object type */
 interface JsonObject {