From 0606f9b416fcdeb674b295bdd89d8cd87be02d0d Mon Sep 17 00:00:00 2001
From: Eric Peterson <ericpeterson@spotify.com>
Date: Thu, 17 Jun 2021 09:30:36 +0200
Subject: [PATCH] Clean up tests.

Signed-off-by: Eric Peterson <ericpeterson@spotify.com>
---
 .../reader/transformers/addBaseUrl.test.ts    | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/plugins/techdocs/src/reader/transformers/addBaseUrl.test.ts b/plugins/techdocs/src/reader/transformers/addBaseUrl.test.ts
index 5a3205f886..6b6eae4b0d 100644
--- a/plugins/techdocs/src/reader/transformers/addBaseUrl.test.ts
+++ b/plugins/techdocs/src/reader/transformers/addBaseUrl.test.ts
@@ -14,6 +14,7 @@
  * limitations under the License.
  */
 
+import { waitFor } from '@testing-library/react';
 import { createTestShadowDom } from '../../test-utils';
 import { addBaseUrl } from '../transformers';
 import { TechDocsStorageApi } from '../../api';
@@ -120,12 +121,9 @@ describe('addBaseUrl', () => {
       postTransformers: [],
     });
 
-    await new Promise<void>(done => {
-      process.nextTick(() => {
-        const actualSrc = root.getElementById('x')?.getAttribute('src');
-        expect(expectedSrc).toEqual(actualSrc);
-        done();
-      });
+    await waitFor(() => {
+      const actualSrc = root.getElementById('x')?.getAttribute('src');
+      expect(expectedSrc).toEqual(actualSrc);
     });
   });
 
@@ -153,12 +151,9 @@ describe('addBaseUrl', () => {
       },
     );
 
-    await new Promise<void>(done => {
-      process.nextTick(() => {
-        const actualSrc = root.getElementById('x')?.getAttribute('src');
-        expect(expectedSrc).toEqual(actualSrc);
-        done();
-      });
+    await waitFor(() => {
+      const actualSrc = root.getElementById('x')?.getAttribute('src');
+      expect(expectedSrc).toEqual(actualSrc);
     });
   });