From 8b3ed473c19bcb8e71f1b16882b45d61117e8788 Mon Sep 17 00:00:00 2001
From: Brendan Dahl <brendan.dahl@gmail.com>
Date: Fri, 3 Aug 2018 17:24:24 -0700
Subject: [PATCH] Utility script to add a reference test.
---
test/add_test.js | 68 ++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 68 insertions(+)
create mode 100644 test/add_test.js
diff --git a/test/add_test.js b/test/add_test.js
new file mode 100644
index 000000000..abbdafa92
--- /dev/null
+++ b/test/add_test.js
@@ -0,0 +1,68 @@
+const fs = require('fs');
+const crypto = require('crypto');
+const execSync = require('child_process').execSync;
+
+const testManifest = 'test/test_manifest.json';
+const pdfFolder = 'test/pdfs/';
+const gitIgnore = 'test/pdfs/.gitignore';
+
+if (process.argv.length < 3) {
+ console.log('\nUsage: node add_test.js FILE\n');
+ console.log(' Add a PDF as a reference test. FILE must be located in ' +
+ `${pdfFolder}`);
+ process.exit(1);
+}
+
+let file = process.argv[2];
+if (!file.startsWith(pdfFolder)) {
+ throw new Error(`PDF file must be in '${pdfFolder}' directory.`);
+}
+if (!fs.existsSync(file)) {
+ throw new Error(`PDF file does not exist '${file}'.`);
+}
+
+function calculateMD5(file, callback) {
+ var hash = crypto.createHash('md5');
+ var stream = fs.createReadStream(file);
+ stream.on('data', function (data) {
+ hash.update(data);
+ });
+ stream.on('error', function (err) {
+ callback(err);
+ });
+ stream.on('end', function() {
+ var result = hash.digest('hex');
+ callback(null, result);
+ });
+}
+
+function getRandomArbitrary(min, max) {
+ return Math.floor(Math.random() * (max - min) + min);
+}
+
+calculateMD5(file, (err, md5) => {
+ if (err) {
+ throw new Error(err);
+ }
+ let contents = fs.readFileSync(gitIgnore, 'utf8').split('\n');
+ let randomLine = getRandomArbitrary(10, contents.length - 2);
+ contents.splice(randomLine, 0, '!' + file.substr(file.lastIndexOf('/') + 1));
+ fs.writeFileSync('test/pdfs/.gitignore', contents.join('\n'));
+
+ contents = fs.readFileSync(testManifest, 'utf8');
+ let pdf = file.substring(file.lastIndexOf('/') + 1, file.length - 4);
+ let randomPoint = getRandomArbitrary(100, contents.length - 20);
+ let bracket = contents.indexOf('},\n', randomPoint);
+ let out = contents.substr(0, bracket) +
+ '},\n' +
+ ` { "id": "${pdf}",\n` +
+ ` "file": "pdfs/${pdf}.pdf",\n` +
+ ` "md5": "${md5}",\n` +
+ ' "rounds": 1,\n' +
+ ' "type": "eq"\n' +
+ ' ' +
+ contents.substr(bracket);
+ fs.writeFileSync('test/test_manifest.json', out);
+ execSync(`git add ${testManifest} ${gitIgnore}`);
+ execSync(`git add ${file}`);
+});