Add unit test decryption for post-encrypt

Author: focusaurus

code/post-encrypt/lambda.js

@@ -5,11 +5,15 @@ const lambdaUtils = require("../lambda-utils");
 
 const console = lambdaUtils.console;
 
+function encrypt(clear) {
+  const cipher = crypto.createCipher("aes192", config.HL_SECRET1);
+  return cipher.update(clear, "utf8", "base64") + cipher.final("base64");
+}
+
 function postEncrypt(call, res, next) {
   console.log("Encrypting payload");
-  const cipher = crypto.createCipher("aes192", process.env.HL_SECRET1);
-  cipher.update(JSON.stringify(call.event.body));
-  call.body = {encrypted: cipher.final("base64")};
+  const encrypted = encrypt(JSON.stringify(call.event.body));
+  call.body = {encrypted};
   next();
 }
 

code/post-encrypt/unit-tests-tap.js

@@ -1,5 +1,7 @@
 "use strict";
 const {handler} = require("./lambda");
+const config = require("../config");
+const crypto = require("crypto");
 const eventSchema = require("./event-schema");
 const set = require("lodash.set");
 const tap = require("tap");
@@ -12,6 +14,11 @@ function mockEvent(path, value) {
   return event;
 }
 
+function decrypt(encrypted) {
+  const decipher = crypto.createDecipher("aes192", config.HL_SECRET1);
+  return decipher.update(encrypted, "base64", "utf8") + decipher.final("utf8");
+}
+
 const invalids = [
   ["body", 0], // not string
   ["body", false], // not string
@@ -38,6 +45,9 @@ tap.test("handler should work in base case", test => {
     const body = JSON.parse(res.body);
     test.ok(body.encrypted);
     test.same(typeof body.encrypted, "string");
+    const payloadJson = decrypt(body.encrypted);
+    const payload = JSON.parse(payloadJson);
+    test.match({foo: "FOO", bar: "BAR"}, payload);
     test.end();
   });
 });