test: add tests for custom blocks

Author: haoqunjiang

Diff for: test/custom.spec.ts

@@ -0,0 +1,100 @@
+import * as path from 'path'
+import { bundle, mockBundleAndRun } from './utils'
+
+test('add custom blocks to the webpack output', async () => {
+  const { code } = await bundle({
+    entry: 'custom-language.vue',
+    module: {
+      rules: [
+        {
+          test: /\.js/,
+          loader: 'babel-loader',
+          options: {
+            presets: ['@babel/preset-env'],
+          },
+        },
+      ],
+    },
+  })
+
+  // should also be transpiled
+  expect(code).toContain(
+    `
+describe('example', function () {
+  it('basic', function (done) {
+    done();
+  });
+});
+  `.trim()
+  )
+})
+
+test('custom blocks should work with src imports', async () => {
+  const { code } = await bundle({
+    entry: 'custom-import.vue',
+    module: {
+      rules: [
+        {
+          test: /\.js/,
+          loader: 'babel-loader',
+          options: {
+            presets: ['@babel/preset-env'],
+          },
+        },
+      ],
+    },
+  })
+
+  expect(code).toContain(
+    `
+describe('example', function () {
+  it('basic', function (done) {
+    done();
+  });
+});
+  `.trim()
+  )
+})
+
+test('passes Component to custom block loaders', async () => {
+  const { componentModule } = await mockBundleAndRun({
+    entry: 'custom-language.vue',
+    module: {
+      rules: [
+        {
+          resourceQuery: /blockType=documentation/,
+          loader: require.resolve('./mock-loaders/docs'),
+        },
+      ],
+    },
+  })
+
+  expect(componentModule.__docs).toContain(
+    'This is example documentation for a component.'
+  )
+})
+
+test('custom blocks can be ignored', async () => {
+  const { code } = await bundle({
+    entry: 'custom-language.vue',
+  })
+  expect(code).not.toContain(`describe('example'`)
+})
+
+test('custom blocks can be ignored even if cache-loader processes them', async () => {
+  const { code } = await bundle({
+    entry: 'custom-language.vue',
+    module: {
+      rules: [
+        {
+          test: /.vue$/,
+          loader: 'cache-loader',
+          options: {
+            cacheDirectory: path.resolve(__dirname, '.cache'),
+          },
+        },
+      ],
+    },
+  })
+  expect(code).not.toContain(`describe('example'`)
+})

Diff for: test/fixtures/custom-import.vue

@@ -0,0 +1,22 @@
+<unit-test src="./unit-test.js">
+</unit-test>
+
+<template>
+  <h2 class="red">{{msg}}</h2>
+</template>
+
+<script>
+export default {
+  data () {
+    return {
+      msg: 'Hello from Component A!'
+    }
+  }
+}
+</script>
+
+<style>
+comp-a h2 {
+  color: #f00;
+}
+</style>

Diff for: test/fixtures/custom-language.vue

@@ -0,0 +1,31 @@
+<unit-test lang="js">
+  describe('example', () => {
+    it('basic', done => {
+      done();
+    })
+  })
+</unit-test>
+
+<documentation>
+  This is example documentation for a component.
+</documentation>
+
+<template>
+  <h2 class="red">{{msg}}</h2>
+</template>
+
+<script>
+export default {
+  data () {
+    return {
+      msg: 'Hello from Component A!'
+    }
+  }
+}
+</script>
+
+<style>
+comp-a h2 {
+  color: #f00;
+}
+</style>

Diff for: test/fixtures/unit-test.js

@@ -0,0 +1,5 @@
+describe('example', () => {
+  it('basic', (done) => {
+    done()
+  })
+})

Diff for: test/mock-loaders/docs.js

@@ -0,0 +1,9 @@
+module.exports = function (source, map) {
+  this.callback(
+    null,
+    `export default Component => {
+      Component.__docs = ${JSON.stringify(source)}
+    }`,
+    map
+  )
+}