infrataster 0.1.0

data/infrataster.gemspec

@@ -0,0 +1,30 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'infrataster/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "infrataster"
+  spec.version       = Infrataster::VERSION
+  spec.authors       = ["Ryota Arai"]
+  spec.email         = ["ryota.arai@gmail.com"]
+  spec.summary       = %q{Infrastructure Behavior Testing Framework}
+  spec.homepage      = "https://github.com/ryotarai/infrataster"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "rspec"
+  spec.add_runtime_dependency "net-ssh"
+  spec.add_runtime_dependency "net-ssh-gateway"
+  spec.add_runtime_dependency "mysql2"
+  spec.add_runtime_dependency "capybara"
+  spec.add_runtime_dependency "selenium-webdriver"
+  spec.add_runtime_dependency "browsermob-proxy"
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+end

data/lib/infrataster/browsermob_proxy.rb

@@ -0,0 +1,31 @@
+require 'browsermob/proxy'
+
+module Infrataster
+  module BrowsermobProxy
+    class << self
+      def server
+        @server ||= start_server
+      end
+
+      def bin_path
+        @bin_path
+      end
+
+      def bin_path=(path)
+        @bin_path = path
+      end
+
+      private
+      def start_server
+        BrowserMob::Proxy::Server.new(find_bin).tap do |server|
+          server.start
+        end
+      end
+
+      def find_bin
+        return @bin_path if @bin_path
+        `which browsermob-proxy`
+      end
+    end
+  end
+end

data/lib/infrataster/contexts/base_context.rb

@@ -0,0 +1,14 @@
+module Infrataster
+  module Contexts
+    class BaseContext
+      attr_reader :server
+      attr_reader :type
+
+      def initialize(server, type)
+        @server = server
+        @type = type
+      end
+    end
+  end
+end
+

data/lib/infrataster/contexts/capybara_context.rb

@@ -0,0 +1,56 @@
+require 'capybara'
+require 'capybara/rspec/matchers'
+require 'selenium-webdriver'
+
+module Infrataster
+  module Contexts
+    class CapybaraContext < BaseContext
+      def initialize(*args)
+        super(*args)
+
+        register_capybara_driver
+      end
+
+      def session
+        return @session if @session
+
+        address, port, @gateway_finalize_proc = server.from_gateway(type.uri.port)
+        Capybara.app_host = "http://#{address}:#{port}"
+        @session = Capybara::Session.new(capybara_driver_name)
+      end
+
+      def page
+        session
+      end
+
+      def before_each(example)
+        example.example_group_instance.extend(Capybara::RSpecMatchers)
+      end
+
+      def after_each(example)
+        @gateway_finalize_proc.call if @gateway_finalize_proc
+      end
+
+      def method_missing(method, *args)
+        session.public_send(method, *args)
+      end
+
+      private
+      def register_capybara_driver
+        proxy = BrowsermobProxy.server.create_proxy
+        proxy.header({"Host" => type.uri.host})
+
+        profile = Selenium::WebDriver::Firefox::Profile.new
+        profile.proxy = proxy.selenium_proxy
+
+        Capybara.register_driver capybara_driver_name do |app|
+          Capybara::Selenium::Driver.new(app, profile: profile)
+        end
+      end
+
+      def capybara_driver_name
+        :"selenium_via_proxy_#{hash}"
+      end
+    end
+  end
+end

data/lib/infrataster/contexts/http_context.rb

@@ -0,0 +1,20 @@
+require 'net/http'
+require 'net/https'
+
+module Infrataster
+  module Contexts
+    class HttpContext < BaseContext
+      def response
+        req = Net::HTTP::Get.new('/', {'Host' => type.uri.host})
+
+        server.from_gateway(type.uri.port) do |address, port|
+          Net::HTTP.start(address, port) do |http|
+            http.request(req)
+          end
+        end
+      end
+    end
+  end
+end
+
+

data/lib/infrataster/contexts/mysql_query_context.rb

@@ -0,0 +1,26 @@
+require 'mysql2'
+
+module Infrataster
+  module Contexts
+    class MysqlQueryContext < BaseContext
+      def results
+        options = {port: 3306, user: 'root', password: ''}
+        if server.options[:mysql]
+          options = options.merge(server.options[:mysql])
+        end
+
+        server.from_gateway(options[:port]) do |address, new_port|
+          client = Mysql2::Client.new(
+            host: address,
+            port: new_port,
+            username: options[:user],
+            password: options[:password],
+          )
+          client.query(type.query)
+        end
+      end
+    end
+  end
+end
+
+

data/lib/infrataster/contexts.rb

@@ -0,0 +1,36 @@
+require 'infrataster/types'
+require 'infrataster/contexts/base_context'
+require 'infrataster/contexts/http_context'
+require 'infrataster/contexts/mysql_query_context'
+require 'infrataster/contexts/capybara_context'
+
+module Infrataster
+  module Contexts
+    class << self
+      def from_example(example)
+        example_group = example.metadata[:example_group]
+
+        server = find_described(Types::ServerType, example_group).server
+        type = find_described(Types::BaseType, example_group)
+
+        type.context_class.new(server, type)
+      end
+
+      private
+      def find_described(type_class, example_group)
+        arg = example_group[:description_args].first
+        if arg.is_a?(type_class)
+          arg
+        else
+          parent_example_group = example_group[:example_group]
+          if parent_example_group
+            find_described(type_class, parent_example_group)
+          else
+            raise Error
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/infrataster/helpers/rspec_helper.rb

@@ -0,0 +1,15 @@
+module Infrataster
+  module Helpers
+    module RSpecHelper
+      def method_missing(method, *args)
+        current_infrataster_context.public_send(method, *args)
+      end
+
+      def current_infrataster_context
+        @infrataster_context
+      end
+    end
+  end
+end
+
+

data/lib/infrataster/helpers/type_helper.rb

@@ -0,0 +1,24 @@
+require 'infrataster/types'
+
+module Infrataster
+  module Helpers
+    module TypeHelper
+      def server(*args)
+        Types::ServerType.new(*args)
+      end
+
+      def http(*args)
+        Types::HttpType.new(*args)
+      end
+
+      def mysql_query(*args)
+        Types::MysqlQueryType.new(*args)
+      end
+
+      def capybara(*args)
+        Types::CapybaraType.new(*args)
+      end
+    end
+  end
+end
+

data/lib/infrataster/helpers.rb

@@ -0,0 +1,4 @@
+require 'infrataster/helpers/type_helper'
+require 'infrataster/helpers/rspec_helper'
+
+

data/lib/infrataster/rspec.rb

@@ -0,0 +1,17 @@
+require 'infrataster'
+require 'rspec'
+
+include Infrataster::Helpers::TypeHelper
+
+RSpec.configure do |config|
+  config.include Infrataster::Helpers::RSpecHelper
+
+  config.before(:each) do
+    @infrataster_context = Infrataster::Contexts.from_example(example)
+    @infrataster_context.before_each(example) if @infrataster_context.respond_to?(:before_each)
+  end
+  config.after(:each) do
+    @infrataster_context.after_each(example) if @infrataster_context.respond_to?(:after_each)
+  end
+end
+

data/lib/infrataster/server.rb

@@ -0,0 +1,125 @@
+require 'tmpdir'
+require 'net/ssh'
+require 'net/ssh/gateway'
+require 'ipaddr'
+
+module Infrataster
+  class Server
+    Error = Class.new(StandardError)
+
+    class << self
+      @@servers = []
+
+      def define(*args)
+        @@servers << Server.new(*args)
+      end
+
+      def find_by_name(name)
+        server = @@servers.find {|s| s.name == name }
+        unless server
+          raise Error, "Server definition for '#{name}' is not found."
+        end
+        server
+      end
+
+    end
+
+    attr_reader :name, :address, :options
+
+    def initialize(name, address, options = {})
+      @name, @address, @options = name, address, options
+    end
+
+    def from
+      if @options[:from]
+        Server.find_by_name(@options[:from])
+      end
+    end
+
+    def ssh_gateway
+      gateway = Net::SSH::Gateway.new(*ssh_start_args)
+      finalize_proc = Proc.new do
+        gateway.shutdown!
+      end
+
+      [gateway, finalize_proc]
+    end
+
+    def with_ssh_gateway
+      gateway, finalize_proc = ssh_gateway
+      yield gateway
+    ensure
+      finalize_proc.call
+    end
+
+    def gateway_open(*args)
+      if block_given?
+        with_ssh_gateway do |gateway|
+          gateway.open(*args) do |port|
+            yield port
+          end
+        end
+      else
+        gateway, gateway_finalize_proc = ssh_gateway
+        port = gateway.open(*args)
+        finalize_proc = Proc.new do
+          gateway.close(port)
+          gateway_finalize_proc.call
+        end
+        [port, finalize_proc]
+      end
+    end
+
+    def from_gateway(port)
+      if from
+        if block_given?
+          from.gateway_open(@address, port) do |new_port|
+            Logger.debug("tunnel: localhost:#{new_port} -> #{from.address} -> #{@address}:#{port}")
+            yield '127.0.0.1', new_port
+          end
+        else
+          new_port, finalize_proc = from.gateway_open(@address, port)
+          Logger.debug("tunnel: localhost:#{new_port} -> #{from.address} -> #{@address}:#{port}")
+          ['127.0.0.1', new_port, finalize_proc]
+        end
+      else
+        if block_given?
+          yield @address, port
+        else
+          [@address, port, nil]
+        end
+      end
+    end
+
+    def ssh_start_args
+      @ssh_start_args ||= _ssh_start_args
+    end
+
+    private
+
+
+    def _ssh_start_args
+      config = {}
+      if @options[:ssh]
+        config = @options[:ssh]
+      elsif @options[:vagrant]
+        if @options[:vagrant] != true
+          vagrant_name = @options[:vagrant]
+        else
+          vagrant_name = @name
+        end
+
+        Dir.mktmpdir do |dir|
+          output = File.join(dir, 'ssh-config')
+          system("/usr/bin/vagrant ssh-config #{vagrant_name} > #{output}")
+          config = Net::SSH::Config.for(@name.to_s, [output])
+        end
+      else
+        raise Error, "Can't get configuration to connect to the server via SSH. Please set ssh option or vagrant option."
+      end
+
+      [config[:host], config[:user], config]
+    end
+  end
+end
+

data/lib/infrataster/types/base_type.rb

@@ -0,0 +1,15 @@
+module Infrataster
+  module Types
+    class BaseType
+      # do nothing
+      def name
+        self.class.name.split('::').last[0...-4]
+      end
+
+      def context_class
+        Contexts.const_get("#{name}Context")
+      end
+    end
+  end
+end
+

data/lib/infrataster/types/capybara_type.rb

@@ -0,0 +1,21 @@
+require 'infrataster/types/base_type'
+require 'uri'
+
+module Infrataster
+  module Types
+    class CapybaraType < BaseType
+      Error = Class.new(StandardError)
+
+      attr_reader :uri
+
+      def initialize(url)
+        @uri = URI.parse(url)
+      end
+
+      def to_s
+        "capybara '#{@uri}'"
+      end
+    end
+  end
+end
+

data/lib/infrataster/types/http_type.rb

@@ -0,0 +1,24 @@
+require 'infrataster/types/base_type'
+require 'uri'
+
+module Infrataster
+  module Types
+    class HttpType < BaseType
+      Error = Class.new(StandardError)
+
+      attr_reader :uri
+
+      def initialize(url_str)
+        @uri = URI.parse(url_str)
+        unless %w!http https!.include?(@uri.scheme)
+          raise Error, "The provided url, '#{@uri}', is not http or https."
+        end
+      end
+
+      def to_s
+        "http '#{@uri}'"
+      end
+    end
+  end
+end
+

data/lib/infrataster/types/mysql_query_type.rb

@@ -0,0 +1,21 @@
+require 'infrataster/types/base_type'
+
+module Infrataster
+  module Types
+    class MysqlQueryType < BaseType
+      Error = Class.new(StandardError)
+
+      attr_reader :query
+
+      def initialize(query, options = {})
+        @query = query
+        @options = options
+      end
+
+      def to_s
+        "mysql '#{@query}'"
+      end
+    end
+  end
+end
+

data/lib/infrataster/types/server_type.rb

@@ -0,0 +1,25 @@
+module Infrataster
+  module Types
+    class ServerType
+      Error = Class.new(StandardError)
+
+      attr_reader :name
+
+      def initialize(name)
+        @name = name
+      end
+
+      def to_s
+        desc = "server '#{server.name}'"
+        desc += " from '#{server.from.name}'" if server.from
+        desc
+      end
+
+      def server
+        Server.find_by_name(@name)
+      end
+    end
+  end
+end
+
+

data/lib/infrataster/types.rb

@@ -0,0 +1,5 @@
+require 'infrataster/types/base_type'
+require 'infrataster/types/server_type'
+require 'infrataster/types/http_type'
+require 'infrataster/types/mysql_query_type'
+require 'infrataster/types/capybara_type'

data/lib/infrataster/version.rb

@@ -0,0 +1,3 @@
+module Infrataster
+  VERSION = "0.1.0"
+end

data/lib/infrataster.rb

@@ -0,0 +1,12 @@
+require "infrataster/version"
+require "infrataster/types"
+require "infrataster/server"
+require "infrataster/helpers"
+require "infrataster/browsermob_proxy"
+require "infrataster/contexts"
+require 'logger'
+
+module Infrataster
+  Logger = ::Logger.new($stdout)
+  Logger.level = ::Logger::ERROR
+end