aws_account_utils 0.1.0

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/aws_account_utils.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'aws_account_utils/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "aws_account_utils"
+  spec.version       = AwsAccountUtils::VERSION
+  spec.authors       = ["keviny22"]
+  spec.email         = ["kevin_young@intuit.com"]
+
+  spec.summary       = %q{A set of helper methods for configuring aspects of AWS that do not have an API.}
+  spec.description   = %q{A set of helper methods that are webdriven.}
+  spec.homepage      = "https://github.com/intuit/aws_account_utils"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency 'bundler', '~> 1.9'
+  spec.add_development_dependency 'webmock', '~> 1.21', '>= 1.21.0'
+  spec.add_development_dependency 'guard-rspec', '~> 4.3', '>= 4.3.1'
+  spec.add_development_dependency 'simplecov', '~> 0.8', '>= 0.8.2'
+  spec.add_development_dependency 'fakefs', '~> 0.6', '>= 0.6.7'
+  spec.add_runtime_dependency 'rake', '~> 10.3.2', '~> 10.0'
+  spec.add_runtime_dependency 'watir-webdriver', '~> 0.8', '>= 0.8.0'
+  spec.add_runtime_dependency 'random-word', '~> 1.3', '>= 1.3.0'
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "aws_account_utils"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,7 @@
+#!/bin/bash
+set -euo pipefail
+IFS=$'\n\t'
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/aws_account_utils.rb

@@ -0,0 +1,200 @@
+require 'aws_account_utils/base'
+require 'aws_account_utils/version'
+require 'aws_account_utils/account_logger'
+require 'aws_account_utils/watir_browser'
+require 'aws_account_utils/account_registration'
+require 'aws_account_utils/customer_information'
+require 'aws_account_utils/settings'
+require 'aws_account_utils/iam_billing'
+require 'aws_account_utils/email_preferences'
+require 'aws_account_utils/enterprise_support'
+require 'aws_account_utils/consolidated_billing'
+require 'aws_account_utils/root_access_keys'
+require 'aws_account_utils/password'
+require 'aws_account_utils/alternate_contacts'
+require 'aws_account_utils/logout'
+
+module AwsAccountUtils
+  class AwsAccountUtils
+    attr_reader :options, :logger, :browser, :screenshots
+
+    def initialize(logger: nil, browser: nil, screenshots: nil)
+      @browser = browser
+      @logger = logger
+      @screenshots = screenshots
+
+      Settings.set_screenshot_dir screenshots if screenshots
+    end
+
+    def create_account(account_name:, account_email:, account_password:, account_details:)
+      raise ArgumentError, "account_detials: must be a hash." unless account_details.is_a?(Hash)
+      account_registration.signup account_name, account_email, account_password
+      resp = customer_information.submit account_email, account_password, account_details
+      logger.info 'Successfully created account.' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def check_enterprise_support(account_email:, account_password:)
+      resp = enterprise_support.existing_support? account_email, account_password
+      logger.info 'Enterprise support was already enabled' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def change_root_password(account_email:, account_password:, new_password:)
+      resp = password.change(account_email, account_password, new_password)
+      logger.info 'Changed root password.' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def confirm_consolidated_billing(account_email:, account_password:, confirmation_link:)
+      resp = consolidated_billing.confirm account_email, account_password, confirmation_link
+      logger.info 'Consolidated billing has been confirmed' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def create_root_access_keys(account_email:, account_password:)
+      resp = root_access_keys.create account_email, account_password
+      logger.info 'Created root access keys.' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def delete_root_access_keys(account_email:, account_password:)
+      resp = root_access_keys.delete account_email, account_password
+      logger.info 'Deleted all root access keys.' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def email_opt_out(account_email:, account_password:)
+      resp = email_preferences.opt_out account_email, account_password
+      logger.info 'Successfully opted out of all emails' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def enable_enterprise_support(account_email:, account_password:)
+      resp = enterprise_support.enable account_email, account_password
+      logger.info 'Enabled enterprise support' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def enable_iam_billing(account_email:, account_password:)
+      resp = iam_billing.enable account_email, account_password
+      logger.info 'Successfully enabled IAM billing' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def existing_consolidated_billing?(account_email:, account_password:)
+      resp = consolidated_billing.existing? account_email, account_password
+      logger.info 'Consolidated billing has already been setup' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def logout_from_console
+      resp = logout.execute
+      logger.info 'Logged out of console' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def request_consolidated_billing(master_account_email:, master_account_password:, account_email:)
+      resp = consolidated_billing.request master_account_email, master_account_password, account_email
+      logger.info 'Consolidated billing has been requested' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def set_challenge_questions(account_email:, account_password:, answers:)
+      raise ArgumentError, "answers: must be a hash." unless answers.is_a?(Hash)
+
+      resp = challenge_questions.create account_email, account_password, answers
+      logger.info 'Security Challenge Questions have been setup' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    def set_alternate_contacts(account_email:, account_password:, contact_info:)
+      raise ArgumentError, "contact_info: must be a hash." unless contact_info.is_a?(Hash)
+
+      resp = alternate_contacts.set account_email, account_password, contact_info
+      logger.info 'Set alterante contacts.' if resp
+      resp
+    ensure
+      browser.close rescue nil
+    end
+
+    private
+    def account_registration
+      @account_registration ||= AccountRegistration.new logger, browser
+    end
+
+    def alternate_contacts
+      @alternate_contacts ||= AlternateContacts.new logger, browser
+    end
+
+    def browser
+      @browser ||= WatirBrowser.new(logger).create
+    end
+
+    def challenge_questions
+      @challenge_questions ||= ChallengeQuestions.new logger, browser
+    end
+
+    def consolidated_billing
+      @consolidated_billing ||= ConsolidatedBilling.new logger, browser
+    end
+
+    def customer_information
+      @customer_information ||= CustomerInformation.new logger, browser
+    end
+
+    def email_preferences
+      @email_preferences ||= EmailPreferences.new logger, browser
+    end
+
+    def enterprise_support
+      @enterprise_support ||= EnterpriseSupport.new logger, browser
+    end
+
+    def iam_billing
+      @iam_billing ||= IamBilling.new logger, browser
+    end
+
+    def logger
+      @logger ||= AccountLogger.new(options[:log_level])
+    end
+
+    def password
+      @password ||= Password.new logger, browser
+    end
+
+    def root_access_keys
+      @root_access_keys ||= RootAccessKeys.new logger, browser
+    end
+
+    def logout
+      @logout ||= Logout.new logger, browser
+    end
+  end
+end

data/lib/aws_account_utils/account_logger.rb

@@ -0,0 +1,26 @@
+require 'logger'
+require 'forwardable'
+
+module AwsAccountUtils
+  class AccountLogger
+    extend Forwardable
+    def_delegators :@logger, :fatal, :debug, :error, :info, :warn, :debug?
+
+    def initialize(log_level = 'info')
+      @log_level = log_level
+      @logger = new_logger
+    end
+
+    private
+    def new_logger
+      STDOUT.sync = true
+      Logger.new(STDOUT).tap do |l|
+        l.datetime_format = '%Y-%m-%dT%H:%M:%S%z'
+        l.formatter = proc do |severity, datetime, progname, msg|
+          "#{datetime} #{severity} : #{msg}\n"
+        end
+        l.level = Logger.const_get @log_level.upcase
+      end
+    end
+  end
+end

data/lib/aws_account_utils/account_registration.rb

@@ -0,0 +1,52 @@
+require 'aws_account_utils/base'
+
+module AwsAccountUtils
+  class AccountRegistration < Base
+    attr_reader :logger, :browser
+
+    def initialize(logger, browser)
+      @logger = logger
+      @browser = browser
+    end
+
+    def signup(account_name, account_email, account_password)
+      logger.debug "Signing up for a new account."
+      login_details = { 'ap_customer_name'  => account_name,
+                        'ap_email'          => account_email,
+                        'ap_email_check'    => account_email,
+                        'ap_password'       => account_password,
+                        'ap_password_check' => account_password }
+
+      browser.goto signup_url
+      browser.text_field(:id => 'ap_customer_name').wait_until_present
+      screenshot(browser, "1")
+      login_details.each do |k,v|
+        browser.text_field(:id => k).when_present.set v
+      end
+      screenshot(browser, "2")
+      browser.button(:id => 'continue-input').when_present.click
+      raise StandardError if browser.div(:id => /message_(error|warning)/).exist?
+      true
+    rescue StandardError
+      screenshot(browser, "error")
+      error_header = browser.div(:id => /message_(error|warning)/).h6.text
+      error_body = browser.div(:id => /message_(error|warning)/).p.text
+      raise StandardError, "AWS signup error: \"#{error_header}: #{error_body}\""
+
+    rescue Watir::Wait::TimeoutError, Net::ReadTimeout => e
+      screenshot(browser, "error")
+      raise StandardError, "#{self.class.name} - #{e}"
+    end
+
+    private
+    def signup_url
+      url =  "https://www.amazon.com/ap/register?"
+      url << "&openid.claimed_id=http://specs.openid.net/auth/2.0/identifier_select"
+      url << "&create=1&openid.assoc_handle=aws&accountStatusPolicy=P1"
+      url << "&openid.ns=http://specs.openid.net/auth/2.0"
+      url << "&openid.identity=http://specs.openid.net/auth/2.0/identifier_select"
+      url << "&openid.mode=checkid_setup"
+      url << "&openid.return_to=https://console.aws.amazon.com/billing/signup"
+    end
+  end
+end

data/lib/aws_account_utils/alternate_contacts.rb

@@ -0,0 +1,42 @@
+require 'aws_account_utils/base'
+
+module AwsAccountUtils
+  class AlternateContacts < Base
+    attr_reader :logger, :browser
+
+    def initialize(logger, browser)
+      @logger = logger
+      @browser = browser
+    end
+
+    def set(account_email, account_password, contact_info = {})
+      logger.debug "Setting alternate account contacts."
+      Login.new(logger, browser).execute url,
+                                         account_email,
+                                         account_password
+      browser.a(:xpath => '//a[@ng-click="toggleEditingAlternateContactsInfoState()"]').when_present.click
+      form_inputs(contact_info)
+      screenshot(browser, "1")
+      browser.button(:xpath => '//button[@ng-click="updateAlternateContacts()"]').when_present.click
+      browser.div(:xpath => '//div[@ng-show="options.status == \'success\'"]').wait_until_present
+      true
+    rescue Watir::Wait::TimeoutError, Net::ReadTimeout => e
+      screenshot(browser, "error")
+      raise StandardError, "#{self.class.name} - #{e}"
+    end
+
+    private
+    def form_inputs(requestor_info)
+      requestor_info.each do |type, details|
+        details.each do |key, value|
+          browser.input(:xpath => "//input[@ng-model=\"alternateContacts.#{type}Contact.#{key}\"]").to_subtype.when_present.set value
+        end
+      end
+    end
+
+    def url
+      'https://console.aws.amazon.com/billing/home?#/account'
+    end
+
+  end
+end

data/lib/aws_account_utils/base.rb

@@ -0,0 +1,26 @@
+module AwsAccountUtils
+  class Base
+
+    def screenshot(browser, file)
+      return unless Settings.screenshot_dir
+      calling_method = /(?<=`)(.*)(?=')/.match(caller.first).to_s.gsub(" ",'-')
+      calling_class = self.class.to_s.gsub("::",'_')
+      browser.screenshot.save screenshot_file(file, calling_class, calling_method)
+    end
+
+    private
+    def increment(path)
+      _, filename, count, extension = *path.match(/(\A.*?)(?:_#(\d+))?(\.[^.]*)?\Z/)
+      count = (count || '0').to_i + 1
+      "#{filename}_##{count}#{extension}"
+    end
+
+    def screenshot_file(file, calling_class, calling_method)
+      file_path = File.join Settings.screenshot_dir, "#{calling_class}_#{calling_method}_#{file}_#1.png"
+      File.exists?(file_path) ? increment(file_path) : file_path
+    end
+
+  end
+end
+
+

data/lib/aws_account_utils/challenge_questions.rb

@@ -0,0 +1,49 @@
+require 'aws_account_utils/base'
+require 'random-word'
+
+module AwsAccountUtils
+  class ChallengeQuestions < Base
+    attr_reader :logger, :browser
+
+    def initialize(logger, browser)
+      @logger = logger
+      @browser = browser
+    end
+
+    def create(account_email, account_password, challenge_words = {})
+      logger.debug "Entering customer details."
+      Login.new(logger, browser).execute url,
+                                         account_email,
+                                         account_password
+
+      browser.a(:xpath => '//a[@ng-click="toggleEditingSecurityQuestionsInfoState()"]').when_present.click
+
+      challenge_words.each do |num, word|
+        browser.input(:xpath => "//input[@ng-model=\"selectedSecurityQuestions.question#{num}.answer\"]").to_subtype.when_present.set word
+      end
+
+      screenshot(browser, "1")
+      browser.button(:xpath => '//button[@ng-click="updateSecurityQuestions()"]').when_present.click
+      browser.a(:xpath => '//a[@ng-click="toggleEditingSecurityQuestionsInfoState()"]').wait_until_present
+      challenge_words
+    rescue Watir::Wait::TimeoutError, Net::ReadTimeout => e
+      screenshot(browser, "error")
+      raise StandardError, "#{self.class.name} - #{e}"
+    end
+
+    private
+    def challenge_words
+      RandomWord.exclude_list << (/_/) if RandomWord.exclude_list.none?
+
+      @challenge_words ||= (1..3).inject({}) do |hash, num|
+        hash[num] = RandomWord.nouns.next
+        hash
+      end
+    end
+
+    def url
+      'https://portal.aws.amazon.com/gp/aws/manageYourAccount'
+    end
+
+  end
+end