More tests

1 files changed, 42 insertions, 10 deletions

diff --git a/__tests__/index.js b/__tests__/index.js
index 5d4c995..d4729df 100644
--- a/__tests__/index.js
+++ b/__tests__/index.js
@@ -1,11 +1,14 @@
-import test from 'ava';
-import unified from 'unified';
-import raw from 'rehype-raw';
-import reParse from 'remark-parse';
-import stringify from 'rehype-stringify';
-import remark2rehype from 'remark-rehype';
+'user strict';
 
-import plugin from '../app';
+import plugin from '..';
+
+const test = require('ava');
+const unified = require('unified');
+const raw = require('rehype-raw');
+const reParse = require('remark-parse');
+const stringify = require('rehype-stringify');
+const remark2rehype = require('remark-rehype');
+const parse5 = require('parse5');
 
 const render = text => unified()
   .use(reParse)
@@ -24,15 +27,44 @@ const renderRaw = text => unified()
 
 test('simple citation', t => {
   const {contents} = render('>This is a citation');
-  t.is(contents, '<blockquote>\n<p>This is a citation</p>\n</blockquote>');
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<blockquote>\n<p>This is a citation</p>\n</blockquote>')
+  );
 });
 
 test('simple citation raw', t => {
   const {contents} = renderRaw('>This is a citation');
-  t.is(contents, '<blockquote>\n<p>This is a citation</p>\n</blockquote>');
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<blockquote>\n<p>This is a citation</p>\n</blockquote>')
+  );
 });
 
 test('question', t => {
   const {contents} = renderRaw('>!question\nWhat does the 🦊 say ?');
-  t.snapshot(contents);
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<div class="special-box question"><div class="special-box-content"><p>What does the 🦊 say ?</p></div></div>'));
+});
+
+test('attention', t => {
+  const {contents} = renderRaw('>!attention\nBe carefull');
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<div class="special-box attention"><div class="special-box-content"><p>Be carefull</p></div></div>'));
+});
+
+test('good', t => {
+  const {contents} = renderRaw('>!good\n🎉');
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<div class="special-box good"><div class="special-box-content"><p>🎉</p></div></div>'));
+});
+
+test('bad', t => {
+  const {contents} = renderRaw('>!bad\n☢️');
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<div class="special-box bad"><div class="special-box-content"><p>☢️</p></div></div>'));
+});
+
+test('secret', t => {
+  const {contents} = renderRaw('>!secret Don\'t tell others about it\n🤫');
+  t.deepEqual(parse5.parse(contents),
+    parse5.parse('<details class="special-box secret"><div><p>🤫</p></div><summary>Don\'t tell others about it</summary></details>'));
 });