From f4d7588e5ac75f22b8f31a6042d16ae81f5c68b5 Mon Sep 17 00:00:00 2001
From: Andrei Zhaleznichenka <zhalea@amazon.de>
Date: Tue, 10 Sep 2024 22:56:10 +0200
Subject: [PATCH] codecov

---
 src/alert/__tests__/runtime-content.test.tsx   | 18 ++++++++++++------
 .../__tests__/runtime-content.test.tsx         | 18 ++++++++++++------
 2 files changed, 24 insertions(+), 12 deletions(-)

diff --git a/src/alert/__tests__/runtime-content.test.tsx b/src/alert/__tests__/runtime-content.test.tsx
index d3972388ca2..909690f2ce7 100644
--- a/src/alert/__tests__/runtime-content.test.tsx
+++ b/src/alert/__tests__/runtime-content.test.tsx
@@ -224,7 +224,11 @@ describe('asynchronous rendering', () => {
       runReplacer(context, replacer) {
         (async () => {
           await pause(500);
+          replacer.hideHeader();
+          replacer.restoreHeader();
           replacer.replaceHeader(headerFn);
+          replacer.hideContent();
+          replacer.restoreContent();
           replacer.replaceContent(contentFn);
         })();
         return {
@@ -244,12 +248,14 @@ describe('asynchronous rendering', () => {
     unmount();
 
     await waitFor(() => {
-      expect(consoleWarnSpy).toBeCalledWith(
-        '[AwsUi] [Runtime alert content] `replaceHeader` called after component unmounted'
-      );
-      expect(consoleWarnSpy).toBeCalledWith(
-        '[AwsUi] [Runtime alert content] `replaceContent` called after component unmounted'
-      );
+      const message = (method: string) =>
+        `[AwsUi] [Runtime alert content] \`${method}\` called after component unmounted`;
+      expect(consoleWarnSpy).toBeCalledWith(message('hideHeader'));
+      expect(consoleWarnSpy).toBeCalledWith(message('restoreHeader'));
+      expect(consoleWarnSpy).toBeCalledWith(message('replaceHeader'));
+      expect(consoleWarnSpy).toBeCalledWith(message('hideContent'));
+      expect(consoleWarnSpy).toBeCalledWith(message('restoreContent'));
+      expect(consoleWarnSpy).toBeCalledWith(message('replaceContent'));
       expect(headerFn).not.toBeCalled();
       expect(contentFn).not.toBeCalled();
     });
diff --git a/src/flashbar/__tests__/runtime-content.test.tsx b/src/flashbar/__tests__/runtime-content.test.tsx
index 80cf6ac01f9..451f55c3d1c 100644
--- a/src/flashbar/__tests__/runtime-content.test.tsx
+++ b/src/flashbar/__tests__/runtime-content.test.tsx
@@ -222,7 +222,11 @@ describe('asynchronous rendering', () => {
       runReplacer(context, replacer) {
         (async () => {
           await pause(500);
+          replacer.hideHeader();
+          replacer.restoreHeader();
           replacer.replaceHeader(headerFn);
+          replacer.hideContent();
+          replacer.restoreContent();
           replacer.replaceContent(contentFn);
         })();
         return {
@@ -242,12 +246,14 @@ describe('asynchronous rendering', () => {
     unmount();
 
     await waitFor(() => {
-      expect(consoleWarnSpy).toBeCalledWith(
-        '[AwsUi] [Runtime flash content] `replaceHeader` called after component unmounted'
-      );
-      expect(consoleWarnSpy).toBeCalledWith(
-        '[AwsUi] [Runtime flash content] `replaceContent` called after component unmounted'
-      );
+      const message = (method: string) =>
+        `[AwsUi] [Runtime flash content] \`${method}\` called after component unmounted`;
+      expect(consoleWarnSpy).toBeCalledWith(message('hideHeader'));
+      expect(consoleWarnSpy).toBeCalledWith(message('restoreHeader'));
+      expect(consoleWarnSpy).toBeCalledWith(message('replaceHeader'));
+      expect(consoleWarnSpy).toBeCalledWith(message('hideContent'));
+      expect(consoleWarnSpy).toBeCalledWith(message('restoreContent'));
+      expect(consoleWarnSpy).toBeCalledWith(message('replaceContent'));
       expect(headerFn).not.toBeCalled();
       expect(contentFn).not.toBeCalled();
     });