'use strict';
var sinon = require('sinon');
var IsemInjector = require('../../../app/src/scripts/isem-injector');

function mock() {}
mock.prototype.onAddVariable = () => {};
mock.prototype.onImportFile = () => {};

sinon.stub(IsemInjector, 'NetworkDiagramDispatcher').returns(mock.prototype);

var stub = {
  onAddVariable: sinon.stub(mock.prototype, 'onAddVariable'),
  onImportFile:  sinon.stub(mock.prototype, 'onImportFile')
};

module.exports = {
  mock: mock,
  stub: stub
};
@@ -127,16 +127,57 @@
 ck() %7B%7D%0A
+mock.prototype.onAddRelation = () =%3E %7B%7D;%0A
 mock.pro
@@ -342,16 +342,78 @@
 tub = %7B%0A
+  onAddRelation: sinon.stub(mock.prototype, 'onAddRelation'),%0A
   onAddV

fix a missing stub