Add test

Author: muan

test/test.js

@@ -39,6 +39,28 @@ const responses = {
         'Content-Type': 'text/html'
       }
     })
+  },
+  '/fragment': function(request) {
+    if (request.headers.get('Accept') === 'text/html; fragment') {
+      return new Response('<div id="fragment">fragment</div>', {
+        status: 200,
+        headers: {
+          'Content-Type': 'text/html; fragment'
+        }
+      })
+    } else {
+      return new Response('406', {
+        status: 406
+      })
+    }
+  },
+  '/test.js': function() {
+    return new Response('alert("what")', {
+      status: 200,
+      headers: {
+        'Content-Type': 'text/javascript'
+      }
+    })
   }
 }
 
@@ -159,6 +181,35 @@ suite('include-fragment-element', function() {
       })
   })
 
+  test('throws on incorrect Content-Type', function() {
+    const el = document.createElement('include-fragment')
+    el.setAttribute('src', '/test.js')
+
+    return el.data.then(
+      () => {
+        assert.ok(false)
+      },
+      error => {
+        assert.match(error, /expected text\/html but was text\/javascript/)
+      }
+    )
+  })
+
+  test('throws on non-matching Content-Type', function() {
+    const el = document.createElement('include-fragment')
+    el.setAttribute('accept', 'text/html; fragment')
+    el.setAttribute('src', '/hello')
+
+    return el.data.then(
+      () => {
+        assert.ok(false)
+      },
+      error => {
+        assert.match(error, /expected text\/html; fragment but was text\/html; charset=utf-8/)
+      }
+    )
+  })
+
   test('data is not writable', function() {
     const el = document.createElement('include-fragment')
     assert.ok(el.data !== 42)
@@ -231,6 +282,17 @@ suite('include-fragment-element', function() {
     })
   })
 
+  test('replaces with response with the right accept header', function() {
+    const div = document.createElement('div')
+    div.innerHTML = '<include-fragment src="/fragment" accept="text/html; fragment">loading</include-fragment>'
+    document.body.appendChild(div)
+
+    return when(div.firstChild, 'load').then(() => {
+      assert.equal(document.querySelector('include-fragment'), null)
+      assert.equal(document.querySelector('#fragment').textContent, 'fragment')
+    })
+  })
+
   test('error event is not cancelable or bubbles', function() {
     const div = document.createElement('div')
     div.innerHTML = '<include-fragment src="/boom">loading</include-fragment>'
@@ -262,6 +324,16 @@ suite('include-fragment-element', function() {
     )
   })
 
+  test('adds is-error class on incorrect Content-Type', function() {
+    const div = document.createElement('div')
+    div.innerHTML = '<include-fragment src="/fragment">loading</include-fragment>'
+    document.body.appendChild(div)
+
+    return when(div.firstChild, 'error').then(() =>
+      assert.ok(document.querySelector('include-fragment').classList.contains('is-error'))
+    )
+  })
+
   test('replaces element when src attribute is changed', function() {
     const elem = document.createElement('include-fragment')
     document.body.appendChild(elem)