Simplify chain unwrapped mocks

Joystream · Jul 24, 2023 · e0d19fb · e0d19fb
1 parent 7cf3349
commit e0d19fb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 12 deletions.
diff --git a/packages/ui/src/app/pages/Validators/ValidatorList.stories.tsx b/packages/ui/src/app/pages/Validators/ValidatorList.stories.tsx
@@ -59,10 +59,8 @@ export default {
 
             staking: {
               activeEra: {
-                unwrap: {
-                  index: 700,
-                  start: Date.now() - 5400000,
-                },
+                index: 700,
+                start: Date.now() - 5400000,
               },
               counterForValidators: 12,
               counterForNominators: 20,

diff --git a/packages/ui/src/mocks/helpers/asChainData.ts b/packages/ui/src/mocks/helpers/asChainData.ts
@@ -1,15 +1,10 @@
 import { createType } from '@joystream/types'
-import { isFunction, mapValues } from 'lodash'
-
-export const asChainData = (data: any, key?: string | number): any => {
-  if (key === 'unwrap' && !isFunction(data)) {
-    const unwrappedValue = asChainData(data)
-    return () => unwrappedValue
-  }
+import { mapValues } from 'lodash'
 
+export const asChainData = (data: any): any => {
   switch (Object.getPrototypeOf(data).constructor.name) {
     case 'Object':
-      return mapValues(data, asChainData)
+      return withUnwrap(mapValues(data, asChainData))
 
     case 'Array':
       return data.map(asChainData)
@@ -24,3 +19,5 @@ export const asChainData = (data: any, key?: string | number): any => {
       return data
   }
 }
+
+const withUnwrap = (data: Record<any, any>) => Object.defineProperty(data, 'unwrap', { value: () => data })