usher/test/fixtures/bootstrap.js

Simple DSL for composing decision workflows for AWS Simple Workflow

'use strict';

var winston = require('winston'),
    Q = require('q'),
    AWS = require('aws-sdk');


module.exports = {
  setup: setup
};


function setup() {
  var domain = '_test_workflow_';
  registerDomain(domain)
    .then(registerWorkflow(domain, 'activity'))
    .then(registerWorkflow(domain, 'heartbeat-activity'))
    .then(registerWorkflow(domain, 'dynamic'))
    .then(registerWorkflow(domain, 'branch'))
    .then(registerWorkflow(domain, 'parent'))
    .then(registerWorkflow(domain, 'child'))
    .then(registerWorkflow(domain, 'dynamic-parent'))
    .then(registerWorkflow(domain, 'dynamic-child-1'))
    .then(registerWorkflow(domain, 'dynamic-child-2'))
    .then(registerWorkflow(domain, 'parent-result'))
    .then(registerWorkflow(domain, 'child-result'))
    .then(registerWorkflow(domain, 'parent-failure'))
    .then(registerWorkflow(domain, 'child-failure'))
    .then(registerWorkflow(domain, 'linear'))
    .then(registerWorkflow(domain, 'execution-test'))
    .then(registerWorkflow(domain, 'loop'))
    .then(registerWorkflow(domain, 'loop-batch'))
    .then(registerWorkflow(domain, 'loop-failure'))
    .then(registerWorkflow(domain, 'loop-child'))
    .then(registerWorkflow(domain, 'while'))
    .then(registerWorkflow(domain, 'accumulator'))
    .then(registerWorkflow(domain, 'nested-while'))
    .then(registerWorkflow(domain, 'variable'))
    .then(registerActivity(domain, 'activity1'))
    .then(registerActivity(domain, 'activity2'))
    .then(registerActivity(domain, 'activity3'))
    .then(registerActivity(domain, 'activity4'))
    .then(registerActivity(domain, 'activity5'))
    .then(registerActivity(domain, 'activity6'))
    .then(registerActivity(domain, 'failure1'))
    .then(registerActivity(domain, 'timeout1'))
    .then(registerActivity(domain, 'while-activity1'))
    .then(registerActivity(domain, 'while-activity2'))
    .then(registerActivity(domain, 'accumulator-activity1'))
    .then(registerActivity(domain, 'accumulator-results'))
    .then(registerActivity(domain, 'variable-activity1'))
    .then(registerActivity(domain, 'variable-activity2'))
    .then(registerActivity(domain, 'variable-activity3'))
    .then(registerActivity(domain, 'test-long-activity1', '1.0.0', 'test-heartbeat-activity-poller-tasklist'))
    .then(registerActivity(domain, 'test-activity1', '1.0.0', 'test-activity-poller-tasklist'))
    .then(registerActivity(domain, 'test-activity1', '1.1.0', 'test-activity-poller-tasklist'))
    .then(registerActivity(domain, 'test-activity2', '1.0.0', 'test-activity-poller-tasklist'))
    .then(registerActivity(domain, 'test-activity2', '1.1.0', 'test-activity-poller-tasklist'));
}


function registerDomain(name) {
  var defer = Q.defer();
  var swf = new AWS.SWF();

  swf.registerDomain({
      name: name,
      workflowExecutionRetentionPeriodInDays: '1'
    },
    function (err, results) {
      if (err) {
        winston.log('warn', 'Unable to register domain: %s due to: %s', name, err);
      } else {
        winston.log('info', 'Registered domain: %s', name);
      }
      defer.resolve();
  });
  return defer.promise;
}


function registerWorkflow(domain, name, version) {
  var defer = Q.defer();
  var swf = new AWS.SWF();

  swf.registerWorkflowType({
      domain: domain,
      name: name,
      version: version || '1.0.0',
      defaultTaskList: { name: 'test-' + name + '-decision-tasklist' },
      defaultChildPolicy: 'TERMINATE'
    },
    function (err, results) {
      if (err) {
        winston.log('warn', 'Unable to register workflow: %s due to: %s', name, err);
      } else {
        winston.log('info', 'Registered workflow: %s', name);
      }
      defer.resolve();
  });
  return defer.promise;
}

function registerActivity(domain, name, version, tasklist) {
  var defer = Q.defer();
  var swf = new AWS.SWF();

  swf.registerActivityType({
      domain: domain,
      name: name,
      version: version || '1.0.0',
      defaultTaskList: { name: tasklist || 'test-workflow-activity-tasklist' },
      defaultTaskHeartbeatTimeout: '30',
      defaultTaskScheduleToCloseTimeout: '90',
      defaultTaskScheduleToStartTimeout: '30',
      defaultTaskStartToCloseTimeout: '60'
    },
    function (err, results) {
      if (err) {
        winston.log('warn', 'Unable to register activity: %s due to: %s', name, err);
      } else {
        winston.log('info', 'Registered activity: %s', name);
      }
      defer.resolve();
  });
  return defer.promise;
}