Unit tests for Grovers, Measure nodes

nodes/errors.js

@@ -127,6 +127,7 @@ module.exports = {
   QUBITS_FROM_DIFFERENT_CIRCUITS,
   SAME_QUBIT_RECEIVED_TWICE,
   BLOCH_SPHERE_WITH_MEASUREMENT,
+  NOT_BIT_STRING,
   validateQubitInput,
   validateRegisterInput,
   validateQubitsFromSameCircuit,

nodes/quantum-algorithms/grovers/grovers.js

@@ -10,6 +10,7 @@ module.exports = function(RED) {
   function GroversNode(config) {
     RED.nodes.createNode(this, config);
     this.name = config.name || 'Grovers';
+    const node = this;
 
     this.on('input', async function(msg, send, done) {
       let error = errors.validateGroversInput(msg);

test/nodes/grovers_spec.js

@@ -1,9 +1,10 @@
 const groversNode = require('../../nodes/quantum-algorithms/grovers/grovers.js');
 const testUtil = require('../test-util');
 const nodeTestHelper = testUtil.nodeTestHelper;
+const {FlowBuilder} = require('../flow-builder');
+const errors = require('../../nodes/errors');
 const assert = require('chai').assert;
 
-
 describe('GroversNode', function() {
   beforeEach(function(done) {
     nodeTestHelper.startServer(done);
@@ -19,34 +20,38 @@ describe('GroversNode', function() {
   });
 
   it('default name outputs correctly', function(done) {
-    let flow = [{id: 'groversNode', type: 'grovers', wires: []}];
-    nodeTestHelper.load(groversNode, flow, function() {
+    flow = new FlowBuilder();
+    flow.add('grovers', 'groversNode', []);
+
+    nodeTestHelper.load(flow.nodes, flow.flow, function() {
       let groversTestNode = nodeTestHelper.getNode('groversNode');
-      groversTestNode.should.have.property('name', 'Grovers');
+      groversTestNode.should.have.property('name', 'grovers');
       done();
     });
   });
 
   it('return success output on valid input', function(done) {
-    let flow = [{id: 'groversNode', type: 'grovers', wires: [['helperNode']]},
-      {id: 'helperNode', type: 'helper'}];
+    flow = new FlowBuilder();
+    flow.add('grovers', 'n1', [['n2']]);
+    flow.addOutput('n2');
+
+    const givenInput = {payload: '111111'};
+    const expectedOutput = {topMeasurement: '111111', iterationsNum: 6};
+    testUtil.correctOutputReceived(flow, givenInput, expectedOutput, done);
+  });
+
+  it('should fail on invalid input', function(done) {
+    flow = new FlowBuilder();
+    flow.add('grovers', 'groversNode', []);
 
-    nodeTestHelper.load(groversNode, flow, function() {
+    nodeTestHelper.load(flow.nodes, flow.flow, function() {
       let groversTestNode = nodeTestHelper.getNode('groversNode');
-      let helperNode = nodeTestHelper.getNode('helperNode');
-
-      helperNode.on('input', function(msg) {
-        const expectedPayload = {topMeasurement: '111111', iterationsNum: 6};
-        try {
-          assert.strictEqual(msg.payload.topMeasurement, expectedPayload.topMeasurement);
-          assert.strictEqual(msg.payload.iterationsNum, expectedPayload.iterationsNum);
-          done();
-        } catch (err) {
-          done(err);
-        }
+      groversTestNode.on('call:error', (call)=> {
+        const actualError = call.firstArg;
+        assert.strictEqual(actualError.message, errors.NOT_BIT_STRING);
+        done();
       });
-
-      groversTestNode.receive({payload: '111111'});
+      groversTestNode.receive({payload: '111112'});
     });
   });
 });

test/nodes/measure_spec.js

@@ -30,4 +30,20 @@ describe('MeasureNode', function() {
 
     testUtil.commandExecuted(flow, command, done);
   });
+
+  it('should return correct output', function(done) {
+    let flow = new FlowBuilder();
+
+    flow.add('quantum-circuit', 'n0', [['n1']],
+        {structure: 'qubits', outputs: '1', qbitsreg: '1', cbitsreg: '1'});
+    flow.add('not-gate', 'n1', [['n2']]);
+    flow.add('measure', 'n2', [['n3']], {selectedBit: '0'});
+    flow.add('local-simulator', 'n3', [['n4']], {shots: 1});
+    flow.addOutput('n4');
+
+    const givenInput = 'dummy input';
+    const expectedOutput = {1: 1};
+
+    testUtil.correctOutputReceived(flow, givenInput, expectedOutput, done);
+  });
 });

test/test-util.js

@@ -42,8 +42,27 @@ function commandExecuted(flowBuilder, command, done) {
   });
 }
 
+function correctOutputReceived(flow, givenInput, expectedOutput, done) {
+  nodeTestHelper.load(flow.nodes, flow.flow, function() {
+    const inputNode = nodeTestHelper.getNode(flow.inputId);
+    const outputNode = nodeTestHelper.getNode(flow.outputId);
+    outputNode.on('input', function(msg) {
+      try {
+        assert.deepEqual(msg.payload, expectedOutput);
+        done();
+      } catch (err) {
+        done(err);
+      } finally {
+        shell.stop();
+      }
+    });
+    inputNode.receive(givenInput);
+  });
+}
+
 module.exports = {
   nodeTestHelper,
   isLoaded,
   commandExecuted,
+  correctOutputReceived,
 };