1 files changed, 25 insertions, 0 deletions
diff --git a/spec/frontend/content_editor/components/content_editor_alert_spec.js b/spec/frontend/content_editor/components/content_editor_alert_spec.js
index 12484cb13c6..ee9ead8f8a7 100644
--- a/spec/frontend/content_editor/components/content_editor_alert_spec.js
+++ b/spec/frontend/content_editor/components/content_editor_alert_spec.js
@@ -51,6 +51,16 @@ describe('content_editor/components/content_editor_alert', () => {
     },
   );
 
+  it('does not show primary action by default', async () => {
+    const message = 'error message';
+
+    createWrapper();
+    eventHub.$emit(ALERT_EVENT, { message });
+    await nextTick();
+
+    expect(findErrorAlert().attributes().primaryButtonText).toBeUndefined();
+  });
+
   it('allows dismissing the error', async () => {
     const message = 'error message';
 
@@ -62,4 +72,19 @@ describe('content_editor/components/content_editor_alert', () => {
 
     expect(findErrorAlert().exists()).toBe(false);
   });
+
+  it('allows dismissing the error with a primary action button', async () => {
+    const message = 'error message';
+    const actionLabel = 'Retry';
+    const action = jest.fn();
+
+    createWrapper();
+    eventHub.$emit(ALERT_EVENT, { message, action, actionLabel });
+    await nextTick();
+    findErrorAlert().vm.$emit('primaryAction');
+    await nextTick();
+
+    expect(action).toHaveBeenCalled();
+    expect(findErrorAlert().exists()).toBe(false);
+  });
 });