diff --git a/packages/core/src/dom_components/model/ComponentWrapper.ts b/packages/core/src/dom_components/model/ComponentWrapper.ts index 06a3b21c2..de29df06e 100644 --- a/packages/core/src/dom_components/model/ComponentWrapper.ts +++ b/packages/core/src/dom_components/model/ComponentWrapper.ts @@ -9,6 +9,9 @@ import { DataVariableProps } from '../../data_sources/model/DataVariable'; import { DataCollectionStateMap } from '../../data_sources/model/data_collection/types'; import ComponentWithCollectionsState, { DataVariableMap } from '../../data_sources/model/ComponentWithCollectionsState'; +export const keyRootData = '__rootData'; + + export default class ComponentWrapper extends ComponentWithCollectionsState { dataSourceWatcher?: DataResolverListener; @@ -132,7 +135,7 @@ export default class ComponentWrapper extends ComponentWithCollectionsState { wrapper.setDataResolver(createDataResolver('my_data_source_id.pages.data')); const stateMap = wrapper.collectionsStateMap; - expect(stateMap).toHaveProperty('__pages'); + expect(stateMap).toHaveProperty(keyRootData); expect(wrapper.page?.collectionsStateMap).toEqual(stateMap); expect(wrapper.head.collectionsStateMap).toEqual(stateMap); });