From a33d507872ee496dc20e0929e83e1c32d696d792 Mon Sep 17 00:00:00 2001
From: exuanbo <exuanbo@protonmail.com>
Date: Sun, 30 Aug 2020 15:39:41 +0800
Subject: [PATCH] test: refactor

---
 test/index.test.js | 49 ++++++++++++++++------------------------------
 1 file changed, 17 insertions(+), 32 deletions(-)

diff --git a/test/index.test.js b/test/index.test.js
index c06831d..d645acb 100644
--- a/test/index.test.js
+++ b/test/index.test.js
@@ -3,11 +3,13 @@
 
 const fs = require('fs')
 const path = require('path')
-const should = require('should')
+const expect = require('chai').expect
 const Vinyl = require('vinyl')
 const inlineSource = require('..')
 
 const getFile = (filePath, contents) => {
+  filePath = filePath.replace(/\\/g, '/')
+  if (!path.extname(filePath)) filePath += '.html'
   return new Vinyl({
     base: path.dirname(filePath),
     path: filePath,
@@ -23,66 +25,49 @@ const getExpected = filePath => {
   return getFile(path.join(__dirname, 'expected', filePath))
 }
 
-const compare = (stream, fixtureName, expectedName, done) => {
-  stream.on('error', err => {
-    should.exist(err)
-    done(err)
-  })
-
+const compare = (done, fixtureName, expectedName, options = {}) => {
+  const stream = inlineSource(options)
+  stream.on('error', err => done(err))
   stream.on('data', file => {
-    should.exist(file)
-    should.exist(file.contents)
-
-    should.equal(
-      String(file.contents),
+    expect(String(file.contents)).to.equal(
       String(getExpected(expectedName).contents)
     )
     done()
   })
-
   stream.write(getFixture(fixtureName))
   stream.end()
 }
 
 describe('gulp-inline-source', () => {
   it('Should inline <script> tag', done => {
-    compare(inlineSource(), 'script.html', 'inlined-script.html', done)
+    compare(done, 'script', 'inlined-script')
   })
 
   it('Should inline <script> tag with ES6 source', done => {
-    compare(inlineSource(), 'script-es6.html', 'inlined-script.html', done)
+    compare(done, 'script-es6', 'inlined-script')
   })
 
   it('Should inline <link> tag', done => {
-    compare(inlineSource(), 'link.html', 'inlined-link.html', done)
+    compare(done, 'link', 'inlined-link')
   })
 
   it('Should inline <img> tag with SVG source', done => {
-    compare(inlineSource(), 'image-svg.html', 'inlined-image-svg.html', done)
+    compare(done, 'image-svg', 'inlined-image-svg')
   })
 
   it('Should inline <img> tag with PNG source', done => {
-    compare(inlineSource(), 'image-png.html', 'inlined-image-png.html', done)
+    compare(done, 'image-png', 'inlined-image-png')
   })
 
-  it('works with type and media attributes', done => {
-    compare(
-      inlineSource(),
-      'with-attributes.html',
-      'inlined-with-attributes.html',
-      done
-    )
+  it('Should work with type and media attributes', done => {
+    compare(done, 'with-attributes', 'inlined-with-attributes')
   })
 
-  it('works with relative paths', done => {
-    compare(inlineSource(), 'script-relative.html', 'inlined-script.html', done)
+  it('Should work with relative paths', done => {
+    compare(done, 'script-relative', 'inlined-script')
   })
 
   it('Should inline assets without minification', done => {
-    const stream = inlineSource({
-      compress: false
-    })
-
-    compare(stream, 'nominify.html', 'inlined-nominify.html', done)
+    compare(done, 'nominify', 'inlined-nominify', { compress: false })
   })
 })