1 files changed, 17 insertions, 1 deletions
diff --git a/spec/frontend/editor/components/helpers.js b/spec/frontend/editor/components/helpers.js
index 3e6cd2a236d..12f90390c18 100644
--- a/spec/frontend/editor/components/helpers.js
+++ b/spec/frontend/editor/components/helpers.js
@@ -1,12 +1,28 @@
 import { EDITOR_TOOLBAR_RIGHT_GROUP } from '~/editor/constants';
+import { apolloProvider } from '~/editor/components/source_editor_toolbar_graphql';
+import getToolbarItemsQuery from '~/editor/graphql/get_items.query.graphql';
 
 export const buildButton = (id = 'foo-bar-btn', options = {}) => {
   return {
     __typename: 'Item',
     id,
     label: options.label || 'Foo Bar Button',
-    icon: options.icon || 'foo-bar',
+    icon: options.icon || 'check',
     selected: options.selected || false,
     group: options.group || EDITOR_TOOLBAR_RIGHT_GROUP,
+    onClick: options.onClick || (() => {}),
+    category: options.category || 'primary',
+    selectedLabel: options.selectedLabel || 'smth',
   };
 };
+
+export const warmUpCacheWithItems = (items = []) => {
+  apolloProvider.defaultClient.cache.writeQuery({
+    query: getToolbarItemsQuery,
+    data: {
+      items: {
+        nodes: items,
+      },
+    },
+  });
+};