feat: add support for json modules

testdouble · Nov 10, 2021 · f810a4c · f810a4c
1 parent 1a19c6e
commit f810a4c
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 22 deletions.
diff --git a/example-esm/lib/foo.json b/example-esm/lib/foo.json
@@ -0,0 +1,3 @@
+{
+  "quibble": "rocks"
+}
diff --git a/example-esm/test/helper.js b/example-esm/test/helper.js
@@ -4,6 +4,7 @@ const quibble = require('quibble')
 beforeEach(function () {
   quibble.esm('../lib/animals/bear.mjs', undefined, function () { return 'a fake bear' })
   quibble.esm('../lib/animals/lion.mjs', undefined, function () { return 'a fake lion' })
+  quibble.esm('../lib/foo.json', undefined, { 'hello': 'quibble' })
 })
 
 afterEach(function () {

diff --git a/example-esm/test/lib/zoo-spec.mjs b/example-esm/test/lib/zoo-spec.mjs
@@ -5,9 +5,11 @@ const { expect } = chai
 
 describe('zoo', function () {
   var subject
+  var json
 
   beforeEach(async function () {
     subject = await import('../../lib/zoo.mjs')
+    json = await import('../../lib/foo.json')
   })
 
   it('contains a fake bear', function () {
@@ -17,4 +19,8 @@ describe('zoo', function () {
   it('contains a fake lion', function () {
     expect(subject.default().animals).to.contain('a fake lion')
   })
+
+  it('manipulates json', function () {
+    expect(json.default).to.deep.equal({ hello: 'quibble' })
+  })
 })
diff --git a/lib/quibble.mjs b/lib/quibble.mjs
@@ -85,37 +85,37 @@ function getStubsInfo (moduleUrl) {
 }
 
 function transformModuleSource ([moduleKey, stubs]) {
-  return `
-${Object.keys(stubs.namedExportStubs || {})
-  .map(
-    (name) =>
-      `export let ${name} = global.__quibble.quibbledModules.get(${JSON.stringify(
-        moduleKey
-      )}).namedExportStubs["${name}"]`
-  )
-  .join(';\n')};
-${
-  stubs.defaultExportStub
-    ? `export default global.__quibble.quibbledModules.get(${JSON.stringify(
-        moduleKey
-      )}).defaultExportStub;`
-    : ''
+  const stringified = JSON.stringify(moduleKey)
+  const namedExportStubs = Object.keys(stubs.namedExportStubs || {})
+    .map(name => `export const ${name} = global.__quibble.quibbledModules.get(${stringified}).namedExportStubs["${name}"]`)
+    .join('\n')
+  const defaultExportStub = stubs.defaultExportStub ?
+    `export default global.__quibble.quibbledModules.get(${stringified}).defaultExportStub` : ''
+
+  return [namedExportStubs, defaultExportStub].join('\n')
 }
-`
+
+function transformJsonSource([moduleKey, stubs]) {
+  return JSON.stringify(global.__quibble.quibbledModules.get(moduleKey).defaultExportStub)
 }
 
 /**
  * @param {string} url
- * @param {{
- *   format: string,
- * }} context
+ * @param {{ format: string }} context
  * @param {Function} defaultLoad
  * @returns {Promise<{ source: !(string | SharedArrayBuffer | Uint8Array), format: string}>}
  */
 export async function load (url, context, defaultLoad) {
   const stubsInfo = getStubsInfo(new URL(url))
+  const isJson = url.includes('.json')
 
-  return stubsInfo
-    ? { source: transformModuleSource(stubsInfo), format: 'module' }
-    : defaultLoad(url, context, defaultLoad)
+  if (stubsInfo) {
+    if (isJson) {
+      return { source: transformJsonSource(stubsInfo), format: 'json' }
+    }
+
+    return { source: transformModuleSource(stubsInfo), format: 'module' }
+  }
+
+  return defaultLoad(url, context, defaultLoad)
 }