From 96c24c687b4e3e32461a97531aac9acc68dd91fb Mon Sep 17 00:00:00 2001
From: Bilawal Hameed <bil@spotify.com>
Date: Fri, 3 Jul 2020 16:04:00 +0200
Subject: [PATCH] chore(techdocs): updated tests

---
 .../src/reader/hooks/shadowDom.test.tsx       | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/plugins/techdocs/src/reader/hooks/shadowDom.test.tsx b/plugins/techdocs/src/reader/hooks/shadowDom.test.tsx
index 1d58bf26e7..b8a6a78c1f 100644
--- a/plugins/techdocs/src/reader/hooks/shadowDom.test.tsx
+++ b/plugins/techdocs/src/reader/hooks/shadowDom.test.tsx
@@ -19,26 +19,31 @@ import { renderWithEffects } from '@backstage/test-utils';
 import { useShadowDom } from './shadowDom';
 
 const ComponentWithoutHook = () => {
-  return <div data-testid="shadow-dom" />;
+  return <div data-testid="root" />;
 };
 
 const ComponentWithHook = () => {
-  const ref = useShadowDom();
-  return <div data-testid="shadow-dom" ref={ref} />;
+  const [ref] = useShadowDom('mkdocs', 'about/license/');
+  return <div data-testid="root" ref={ref} />;
 };
 
 describe('useShadowDom', () => {
   it('does not create a Shadow DOM instance', async () => {
     const rendered = await renderWithEffects(<ComponentWithoutHook />);
 
-    const divElement = rendered.getByTestId('shadow-dom');
-    expect(divElement.shadowRoot).not.toBeInstanceOf(ShadowRoot);
+    const outerDivElement = rendered.getByTestId('root');
+    expect(outerDivElement.shadowRoot).not.toBeInstanceOf(ShadowRoot);
+    expect(outerDivElement.children.length).toBe(0);
   });
 
   it('create a Shadow DOM instance', async () => {
     const rendered = await renderWithEffects(<ComponentWithHook />);
 
-    const divElement = rendered.getByTestId('shadow-dom');
-    expect(divElement.shadowRoot).toBeInstanceOf(ShadowRoot);
+    const outerDivElement = rendered.getByTestId('root');
+    expect(outerDivElement.shadowRoot).not.toBeInstanceOf(ShadowRoot);
+    expect(outerDivElement.children.length).toBe(1);
+
+    const innerDivElement = outerDivElement.children[0];
+    expect(innerDivElement.shadowRoot).toBeInstanceOf(ShadowRoot);
   });
 });