Get rid of non-null assertion

wojtekmaj · Apr 17, 2024 · 043b0e4 · 043b0e4
1 parent 12a61f2
commit 043b0e4
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 5 deletions.
diff --git a/sample/index.html b/sample/index.html
@@ -1,12 +1,12 @@
-<!DOCTYPE html>
+<!doctype html>
 <html lang="en-US">
   <head>
     <meta charset="utf-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>react-calendar sample page</title>
   </head>
   <body>
-    <div id="react-root"></div>
+    <div id="root"></div>
     <script type="module" src="./index.tsx"></script>
   </body>
 </html>
diff --git a/sample/index.tsx b/sample/index.tsx
@@ -2,4 +2,10 @@ import { createRoot } from 'react-dom/client';
 
 import Sample from './Sample.js';
 
-createRoot(document.getElementById('react-root')!).render(<Sample />);
+const root = document.getElementById('root');
+
+if (!root) {
+  throw new Error('Could not find root element');
+}
+
+createRoot(root).render(<Sample />);
diff --git a/test/index.html b/test/index.html
@@ -6,7 +6,7 @@
     <title>react-calendar test page</title>
   </head>
   <body>
-    <div id="react-root"></div>
+    <div id="root"></div>
     <script type="module" src="./index.tsx"></script>
   </body>
 </html>
diff --git a/test/index.tsx b/test/index.tsx
@@ -3,7 +3,13 @@ import { createRoot } from 'react-dom/client';
 
 import Test from './Test.js';
 
-createRoot(document.getElementById('react-root')!).render(
+const root = document.getElementById('root');
+
+if (!root) {
+  throw new Error('Could not find root element');
+}
+
+createRoot(root).render(
   <StrictMode>
     <Test />
   </StrictMode>,