Add tests

Author: rzhade3

.eslintrc.json

@@ -7,7 +7,8 @@
     "no-invalid-this": "off",
     "@typescript-eslint/no-invalid-this": ["error"],
     "import/extensions": ["error", "always"],
-    "github/no-inner-html": "off"
+    "github/no-inner-html": "off",
+    "@typescript-eslint/no-unused-vars": ["error", { "vars": "all", "argsIgnorePattern": "^_" }]
   },
   "overrides": [
     {

test/render.ts

@@ -1,5 +1,5 @@
 import {expect} from 'chai'
-import {html, render} from '../lib/index.js'
+import {html, render, setCSPTrustedTypesPolicy} from '../lib/index.js'
 import type {TemplateResult} from '../lib/index.js'
 
 describe('render', () => {
@@ -8,6 +8,10 @@ describe('render', () => {
     surface = document.createElement('section')
   })
 
+  afterEach(() => {
+    setCSPTrustedTypesPolicy(null)
+  })
+
   it('memoizes by TemplateResult#template, updating old templates with new values', () => {
     const main = (x: string | null = null) => html`<div class="${x}"></div>`
     render(main('foo'), surface)
@@ -55,4 +59,21 @@ describe('render', () => {
       expect(surface.innerHTML).to.contain('<div><span><div></div></span><span><div></div></span></div>')
     })
   })
+
+  describe('trusted types', () => {
+    it('respects a Trusted Types Policy if it is set', () => {
+      let policyCalled = false
+      const rewrittenFragment = '<div id="bar"></div>'
+      setCSPTrustedTypesPolicy({
+        createHTML: (_html: string) => {
+          policyCalled = true
+          return rewrittenFragment
+        }
+      })
+      const main = (x: string | null = null) => html`<div class="${x}"></div>`
+      render(main('foo'), surface)
+      expect(surface.innerHTML).to.equal(rewrittenFragment)
+      expect(policyCalled).to.be.true
+    })
+  })
 })

test/trusted-types.ts

@@ -1,14 +1,20 @@
 import {expect} from 'chai'
-import {TrustedTypesPolicy} from '../lib/index.js'
+import {setCSPTrustedTypesPolicy} from '../lib/index.js'
+import {getCSPTrustedTypesPolicy} from '../lib/trusted-types.js'
 
 describe('trusted types', () => {
+  after(() => {
+    setCSPTrustedTypesPolicy(null)
+  })
+
   it('can set a CSP Trusted Types policy', () => {
     const dummyPolicy = {
       createHTML: (htmlText: string) => {
         return htmlText
       }
     }
-    TrustedTypesPolicy.setTrustedTypesPolicy(dummyPolicy)
-    expect(TrustedTypesPolicy.cspTrustedTypesPolicy).to.equal(dummyPolicy)
+    expect(getCSPTrustedTypesPolicy()).to.equal(null)
+    setCSPTrustedTypesPolicy(dummyPolicy)
+    expect(getCSPTrustedTypesPolicy()).to.equal(dummyPolicy)
   })
 })

test/unsafe-html.ts

@@ -1,7 +1,13 @@
 import {expect} from 'chai'
-import {html, render, unsafeHTML} from '../lib/index.js'
+import {html, render, setCSPTrustedTypesPolicy, unsafeHTML} from '../lib/index.js'
 
 describe('unsafeHTML', () => {
+  beforeEach(() => {
+    setCSPTrustedTypesPolicy(null)
+  })
+  afterEach(() => {
+    setCSPTrustedTypesPolicy(null)
+  })
   it('renders basic text', async () => {
     const surface = document.createElement('section')
     render(html`<div>${unsafeHTML('Hello World')}</div>`, surface)
@@ -31,4 +37,18 @@ describe('unsafeHTML', () => {
     render(fn('<a href="">Universe</a>'), surface)
     expect(surface.innerHTML).to.equal('<div><span>Hello</span><span><a href="">Universe</a></span></div>')
   })
+  it('respects trusted types', async () => {
+    let policyCalled = false
+    const rewrittenFragment = '<div id="bar">This has been rewritten by Trusted Types.</div>'
+    setCSPTrustedTypesPolicy({
+      createHTML: (_html: string) => {
+        policyCalled = true
+        return rewrittenFragment
+      }
+    })
+    const surface = document.createElement('section')
+    render(html`<div>${unsafeHTML('<span>Hello</span><span>World</span>')}</div>`, surface)
+    expect(surface.innerHTML).to.equal(rewrittenFragment)
+    expect(policyCalled).to.be.true
+  })
 })