rspec-infrastructure 0.1

data/.gitignore

@@ -0,0 +1,2 @@
+pkg/*
+Gemfile.lock

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+
+gemspec

data/README.md

@@ -0,0 +1,6 @@
+Infrastructure testing with rspec
+
+
+require 'rspec/infrastructure'
+
+

data/Rakefile

@@ -0,0 +1,12 @@
+#!/usr/bin/env rake
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+
+Bundler.require :development
+
+RSpec::Core::RakeTask.new(:spec) do |task|
+  task.rspec_opts = [ '-c', '-f documentation' ]
+  task.pattern    = 'spec/**/*_spec.rb'
+end
+
+task :default => :spec

data/lib/rspec/infrastructure.rb

@@ -0,0 +1,3 @@
+require 'rspec/infrastructure/dsl'
+require 'rspec/infrastructure/nagios_matchers'
+require 'rspec/infrastructure/host_manager'

data/lib/rspec/infrastructure/dsl.rb

@@ -0,0 +1,36 @@
+module RSpec
+  module Core
+    #class ExampleGroup
+
+    module SharedExampleGroup
+      alias_method :application,      :shared_examples
+    end
+
+    module DSL
+      def environment(*args, &block)
+        *args = [ 'asdasd Shrubbery', { :environment => :shrubbery }]
+        env = RSpec::Core::ExampleGroup.describe(*args, &block)
+        env.register
+      end
+
+      def roles(roles, &block)
+        roles.each do |role|
+          Nagios::HostManager.hostnames_for_roles(metadata[:environment], role).each do |hostname|
+            *args = [ "Host: #{hostname}", { :host => hostname }]
+            bleh = RSpec::Core::ExampleGroup.describe(*args, &block)
+            bleh.before :all do
+              @host_manager = Nagios::HostManager.new
+            end
+            bleh.after :all do
+              @host_manager.hosts.each { |hostname, host| host.ssh.close }
+            end
+            bleh.before do
+              @host = @host_manager.manage(hostname)
+            end
+            bleh.register
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rspec/infrastructure/host_manager.rb

@@ -0,0 +1,41 @@
+require 'awesome_print'
+require 'hashie'
+require 'net/ssh/simple'
+
+module Nagios
+
+  class HostManager
+
+    attr_reader :hosts
+
+    def initialize
+      @hosts = {}
+    end
+
+    def manage hostname
+      @hosts[hostname] ||= Host.new(hostname)
+    end
+
+    def self.hostnames_for_roles environment, roles
+      [ *roles ].map do |role|
+        $hosts[environment][role]
+      end.flatten
+    end
+  end
+
+  class Host
+
+    attr_reader :hostname, :ssh
+
+    def initialize(hostname)
+      @hostname = hostname
+      @user = 'deploy'
+      @keys = [ '/Users/james/.ssh/id_rsa' ]
+      @ssh = Net::SSH::Simple.new
+    end
+
+    def exec(command)
+      Hashie::Mash.new(@ssh.ssh(hostname, command, { :user => @user, :keys => @keys, :timeout => 300 }))
+    end
+  end
+end

data/lib/rspec/infrastructure/nagios_matchers.rb

@@ -0,0 +1,53 @@
+require 'socket'
+require 'timeout'
+
+RSpec::Matchers.define :have_remote_file do |file, options|
+
+  options ||= {}.reverse_merge!({ :sudo => false })
+
+  match do |host|
+    command = "[ -f #{file} ]"
+    command = "sudo #{command}" if options.fetch(:sudo)
+    host.exec(command).exit_code == 0
+  end
+
+  failure_message_for_should do |host|
+    "#{host.hostname} is missing #{@expected.first} "
+  end
+end
+
+RSpec::Matchers.define :listen_to_port do |port, options|
+
+  match do |host|
+    port_open?(host.hostname, port)
+  end
+
+  failure_message_for_should do |host|
+    "#{host.hostname} is not listening on port #{@expected.first} "
+  end
+end
+
+RSpec::Matchers.define :listen_on_local_port do |port, options|
+
+  match do |host|
+    command = "nc -z #{host.hostname} #{port}"
+    host.exec(command).exit_code == 0
+  end
+
+  failure_message_for_should do |host|
+    "#{host.hostname} is not listening on local port #{@expected.first} "
+  end
+end
+
+def port_open?(ip, port, seconds=1)
+  Timeout::timeout(seconds) do
+    begin
+      TCPSocket.new(ip, port).close
+      true
+    rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
+      false
+    end
+  end
+rescue Timeout::Error
+  false
+end

data/lib/rspec/infrastructure/version.rb

@@ -0,0 +1,5 @@
+module RSpec
+  module Infrastructure
+    VERSION = '0.1'
+  end
+end

data/rspec-infrastructure.gemspec

@@ -0,0 +1,19 @@
+require File.expand_path('../lib/rspec/infrastructure/version', __FILE__)
+
+Gem::Specification.new do |gem|
+ gem.name        = "rspec-infrastructure"
+ gem.version     = RSpec::Infrastructure::VERSION
+ gem.authors     = [ "James Martelletti", "Ash McKenzie" ]
+ gem.email       = [ "james@hooroo.com",  "ash@hooroo.com" ]
+
+ gem.summary     = "Test your application infrastructure and environments with rspec"
+ gem.description = "Perform high level specification and testing of your application infrastructure and environments using familiar rspec syntax and matchers"
+ gem.homepage    = "http://github.com/hooroo/rspec-infrastructure"
+
+ gem.executables = `git ls-files -- bin/*`.split("\n").map { |f| File.basename(f) }
+ gem.files       = `git ls-files`.split("\n")
+ gem.test_files  = `git ls-files -- spec/*`.split("\n")
+
+ gem.add_dependency             'rspec',       '~>2.11'
+ gem.add_development_dependency 'rake',        '~>0.9'
+end

metadata

@@ -0,0 +1,96 @@
+--- !ruby/object:Gem::Specification
+name: rspec-infrastructure
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- James Martelletti
+- Ash McKenzie
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.11'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.9'
+description: Perform high level specification and testing of your application infrastructure
+  and environments using familiar rspec syntax and matchers
+email:
+- james@hooroo.com
+- ash@hooroo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- lib/rspec/infrastructure.rb
+- lib/rspec/infrastructure/dsl.rb
+- lib/rspec/infrastructure/host_manager.rb
+- lib/rspec/infrastructure/nagios_matchers.rb
+- lib/rspec/infrastructure/version.rb
+- rspec-infrastructure.gemspec
+homepage: http://github.com/hooroo/rspec-infrastructure
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -2192991719490512288
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -2192991719490512288
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Test your application infrastructure and environments with rspec
+test_files: []