features/sts/step_definitions/sts.js

module.exports = function() {
  this.Before('@sts', function (callback) {
    this.service = new this.AWS.STS({
      stsRegionalEndpoints: "regional"
    });
    callback();
  });

  this.Given(/^I get an STS session token with a duration of (\d+) seconds$/, function(duration, callback) {
    this.request(null, 'getSessionToken', {DurationSeconds: parseInt(duration)}, callback, false);
  });

  this.Then(/^the result should contain an access key ID and secret access key$/, function(callback) {
    this.assert.compare(this.data.Credentials.AccessKeyId.length, '>', 0);
    this.assert.compare(this.data.Credentials.SecretAccessKey.length, '>', 0);
    callback();
  });

  this.Given(/^I try to assume role with web identity$/, function(callback) {
    var params = {RoleArn: 'arn:aws:iam::123456789:role/WebIdentity', RoleSessionName: 'name', WebIdentityToken: 'token'};
    this.request(null, 'assumeRoleWithWebIdentity', params, callback, false);
  });

  this.Given(/^I try to assume role with SAML$/, function(callback) {
    var arn = 'arn:aws:iam::123456789:role/Role';
    var token = 'TOKENVALUETOKENVALUETOKENVALUETOKENVALUE';
    var params = {RoleArn: arn, PrincipalArn: arn, SAMLAssertion: token};
    this.request(null, 'assumeRoleWithSAML', params, callback, false);
  });
};