From 2f8fb03df5661569ded2ebd30b91889fafa7d370 Mon Sep 17 00:00:00 2001
From: Tony Brix <tony@brix.ninja>
Date: Thu, 25 Apr 2019 13:33:20 -0500
Subject: [PATCH] use loadFiles in bench

---
 test/bench.js | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/test/bench.js b/test/bench.js
index d435a72f83..2f59f7e161 100644
--- a/test/bench.js
+++ b/test/bench.js
@@ -1,20 +1,19 @@
-#!/usr/bin/env node
-
 const path = require('path');
-const fs = require('fs');
 const htmlDiffer = require('./helpers/html-differ.js');
+const {loadFiles} = require('./helpers/load.js');
 
 let marked = require('../');
 
 function load() {
-  const folder = path.resolve(__dirname, './specs/commonmark');
-  const files = fs.readdirSync(folder);
-  return files.reduce((arr, file) => {
-    if (file.match(/\.json$/)) {
-      return arr.concat(require(`${folder}/${file}`));
-    }
-    return arr;
-  }, []);
+  const dir = path.resolve(__dirname, './specs/commonmark');
+  const sections = loadFiles(dir);
+  let specs = [];
+
+  for (const section in sections) {
+    specs = specs.concat(sections[section].specs);
+  }
+
+  return specs;
 }
 
 function runBench(options) {