update test utitlities (#1101)

src/test-utilities/getMockDictionary.ts

@@ -1,5 +1,23 @@
-import type StyleDictionary from 'style-dictionary'
 import {getMockToken} from './getMockToken.js'
+import type {Dictionary, TransformedToken, TransformedTokens} from 'style-dictionary/types'
+
+const flattenTokens = (tokenTree: TransformedTokens): TransformedToken[] => {
+  const output: TransformedToken[] = []
+
+  const getToken = (tokens: TransformedTokens, flatTokens: TransformedToken[]) => {
+    for (const token of Object.values(tokens)) {
+      if (Object.prototype.hasOwnProperty.call(token, 'name')) {
+        flatTokens.push(token as TransformedToken)
+        continue
+      }
+      getToken(token, flatTokens)
+    }
+  }
+
+  getToken(tokenTree, output)
+
+  return output
+}
 
 const mockDictionaryDefault = {
   tokens: {
@@ -14,11 +32,7 @@ const mockDictionaryDefault = {
   },
 }
 
-export const getMockDictionary = (tokens?: StyleDictionary.TransformedTokens): StyleDictionary.Dictionary => ({
-  allTokens: Object.values((tokens || mockDictionaryDefault).tokens.subgroup),
+export const getMockDictionary = (tokens?: TransformedTokens): Dictionary => ({
+  allTokens: flattenTokens(tokens || mockDictionaryDefault),
   tokens: tokens || mockDictionaryDefault,
-  allProperties: Object.values((tokens || mockDictionaryDefault).tokens.subgroup),
-  properties: tokens || mockDictionaryDefault,
-  usesReference: _value => false,
-  getReferences: _value => [],
 })

src/test-utilities/getMockToken.ts

@@ -1,4 +1,4 @@
-import type StyleDictionary from 'style-dictionary'
+import {TransformedToken} from 'style-dictionary/types'
 
 const mockTokenDefaults = {
   name: 'tokenName',
@@ -11,15 +11,34 @@ const mockTokenDefaults = {
   isSource: true,
   $value: 'transformedValue',
   attributes: {},
+} as const
+
+type getMockTokenOptions = {
+  remove: Array<keyof typeof mockTokenDefaults>
+}
+
+const removeProps = (
+  token: Record<string, unknown>,
+  props: getMockTokenOptions['remove'] = [],
+): Partial<TransformedToken> => {
+  for (const prop of props) {
+    delete token[prop]
+  }
+  return token
 }
 /**
  *
  * @param valueOverrides partial StyleDictionary.TransformedToken
  * @returns StyleDictionary.TransformedToken - a merge of {@link mockTokenDefaults} and any valid properties provided in the valueOverrides param
  */
-export const getMockToken = (valueOverrides: {
-  [key: keyof StyleDictionary.TransformedToken]: unknown
-}): StyleDictionary.TransformedToken => ({
-  ...mockTokenDefaults,
-  ...valueOverrides,
-})
+export const getMockToken = (
+  valueOverrides: {
+    [key: keyof TransformedToken]: unknown
+  },
+  options?: getMockTokenOptions,
+) => {
+  return {
+    ...removeProps(mockTokenDefaults, options?.remove),
+    ...valueOverrides,
+  }
+}