serverspec-clc 2.1.0

data/lib/serverspec/matcher/be_reachable.rb

@@ -0,0 +1,17 @@
+RSpec::Matchers.define :be_reachable  do
+  match do |host|
+    proto   = 'tcp'
+    timeout = 5
+    if @attr
+      port    = @attr[:port]
+      proto   = @attr[:proto]   if @attr[:proto]
+      timeout = @attr[:timeout] if @attr[:timeout]
+    end
+
+    host.reachable?(port, proto, timeout)
+  end
+
+  chain :with do |attr|
+    @attr = attr
+  end
+end

data/lib/serverspec/matcher/be_readable.rb

@@ -0,0 +1,13 @@
+RSpec::Matchers.define :be_readable do
+  match do |file|
+    file.readable?(@by_whom, @by_user)
+  end
+
+  chain :by do |by_whom|
+    @by_whom = by_whom
+  end
+
+  chain :by_user do |by_user|
+    @by_user = by_user
+  end
+end

data/lib/serverspec/matcher/be_resolvable.rb

@@ -0,0 +1,9 @@
+RSpec::Matchers.define :be_resolvable do
+  match do |name|
+    name.resolvable?(@type)
+  end
+
+  chain :by do |type|
+    @type = type
+  end
+end

data/lib/serverspec/matcher/be_running.rb

@@ -0,0 +1,13 @@
+RSpec::Matchers.define :be_running do
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::Service'
+      subject.running?(@under)
+    else
+      subject.running?
+    end
+  end
+
+  chain :under do |under|
+    @under = under
+  end
+end

data/lib/serverspec/matcher/be_writable.rb

@@ -0,0 +1,13 @@
+RSpec::Matchers.define :be_writable do
+  match do |file|
+    file.writable?(@by_whom, @by_user)
+  end
+
+  chain :by do |by_whom|
+    @by_whom = by_whom
+  end
+
+  chain :by_user do |by_user|
+    @by_user = by_user
+  end
+end

data/lib/serverspec/matcher/belong_to_group.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :belong_to_group do |group|
+  match do |user|
+    user.belongs_to_group?(group)
+  end
+end

data/lib/serverspec/matcher/belong_to_primary_group.rb

@@ -0,0 +1,5 @@
+RSpec::Matchers.define :belong_to_primary_group do |group|
+  match do |user|
+    user.belongs_to_primary_group?(group)
+  end
+end

data/lib/serverspec/matcher/contain.rb

@@ -0,0 +1,24 @@
+RSpec::Matchers.define :contain do |pattern|
+  match do |file|
+    file.contain(pattern, @from, @to)
+  end
+
+  # for contain(pattern).from(/A/).to(/B/)
+  chain :from do |from|
+    @from = Regexp.new(from).inspect
+  end
+
+  chain :to do |to|
+    @to = Regexp.new(to).inspect
+  end
+
+  # for contain(pattern).after(/A/)
+  chain :after do |after|
+    @from = Regexp.new(after).inspect
+  end
+
+  # for contain(pattern).before(/B/)
+  chain :before do |before|
+    @to = Regexp.new(before).inspect
+  end
+end

data/lib/serverspec/matcher/have_entry.rb

@@ -0,0 +1,14 @@
+RSpec::Matchers.define :have_entry do |entry|
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::Cron'
+      subject.has_entry?(@user, entry)
+    elsif subject.respond_to?(:has_entry?)
+      subject.has_entry?(entry)
+    end
+  end
+
+  # For cron type
+  chain :with_user do |user|
+    @user = user
+  end
+end

data/lib/serverspec/matcher/have_rule.rb

@@ -0,0 +1,17 @@
+RSpec::Matchers.define :have_rule do |rule|
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::Iptables' || subject.class.name == 'Serverspec::Type::Ip6tables'
+      subject.has_rule?(rule, @table, @chain)
+    else
+      subject.has_rule?(rule)
+    end
+  end
+
+  chain :with_table do |table|
+    @table = table
+  end
+
+  chain :with_chain do |chain|
+    @chain = chain
+  end
+end

data/lib/serverspec/matcher/have_site_application.rb

@@ -0,0 +1,18 @@
+RSpec::Matchers.define :have_site_application do |app|
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::IisWebsite'
+      subject.has_site_application?(app, @pool, @physical_path)
+    else
+      className = subject.class.name
+      raise "not supported class #{className}"
+    end
+  end
+
+  chain :with_pool do |pool|
+    @pool = pool
+  end
+  
+  chain :with_physical_path do |physical_path|
+    @physical_path = physical_path
+  end
+end

data/lib/serverspec/matcher/have_site_bindings.rb

@@ -0,0 +1,22 @@
+RSpec::Matchers.define :have_site_bindings do |port|
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::IisWebsite'
+      subject.has_site_bindings?(port, @protocol, @ipaddress, @host_header)
+    else
+      className = subject.class.name
+      raise "not supported class #{className}"
+    end
+  end
+
+  chain :with_protocol do |protocol|
+    @protocol = protocol
+  end
+
+  chain :with_ipaddress do |ipaddress|
+    @ipaddress = ipaddress
+  end
+
+  chain :with_host_header do |host_header|
+    @host_header = host_header
+  end
+end

data/lib/serverspec/matcher/have_virtual_dir.rb

@@ -0,0 +1,14 @@
+RSpec::Matchers.define :have_virtual_dir do |vdir|
+  match do |subject|
+    if subject.class.name == 'Serverspec::Type::IisWebsite'
+      subject.has_virtual_dir?(vdir, @path)
+    else
+      className = subject.class.name
+      raise "not supported class #{className}"
+    end
+  end
+
+  chain :with_path do |path|
+    @path = path
+  end
+end

data/lib/serverspec/setup.rb

@@ -0,0 +1,333 @@
+require 'fileutils'
+require 'erb'
+
+module Serverspec
+  class Setup
+    def self.run
+
+      ask_os_type
+
+      if @os_type == 'UN*X'
+        ask_unix_backend
+      else
+        ask_windows_backend
+      end
+
+      if @backend_type == 'ssh'
+        print 'Vagrant instance y/n: '
+        @vagrant = $stdin.gets.chomp
+        if @vagrant =~ (/(true|t|yes|y|1)$/i)
+          @vagrant = true
+          print 'Auto-configure Vagrant from Vagrantfile? y/n: '
+          auto_config = $stdin.gets.chomp
+          if auto_config =~ (/(true|t|yes|y|1)$/i)
+            auto_vagrant_configuration
+          else
+            print('Input vagrant instance name: ')
+            @hostname = $stdin.gets.chomp
+          end
+        else
+          @vagrant = false
+          print('Input target host name: ')
+          @hostname = $stdin.gets.chomp
+        end
+      elsif @backend_type == 'winrm'
+        print('Input target host name: ')
+        @hostname = $stdin.gets.chomp
+      else
+        @hostname = 'localhost'
+      end
+
+      ['spec', "spec/#{@hostname}"].each { |dir| safe_mkdir(dir) }
+      safe_create_spec
+      safe_create_spec_helper
+      safe_create_rakefile
+      safe_create_dotrspec
+    end
+
+    def self.ask_os_type
+      prompt = <<-EOF
+Select OS type:
+
+  1) UN*X
+  2) Windows
+
+Select number: 
+EOF
+
+      print prompt.chop
+      num = $stdin.gets.to_i - 1
+      puts
+
+      @os_type = ['UN*X', 'Windows'][num] || 'UN*X'
+    end
+
+    def self.ask_unix_backend
+      prompt = <<-EOF
+Select a backend type:
+
+  1) SSH
+  2) Exec (local)
+
+Select number: 
+EOF
+      print prompt.chop
+      num = $stdin.gets.to_i - 1
+      puts
+
+      @backend_type = ['ssh', 'exec'][num] || 'exec'
+    end
+
+    def self.ask_windows_backend
+      prompt = <<-EOF
+Select a backend type:
+
+  1) WinRM
+  2) Cmd (local)
+
+Select number: 
+EOF
+      print prompt.chop
+      num = $stdin.gets.to_i - 1
+      puts
+
+      @backend_type = ['winrm', 'cmd'][num] || 'exec'
+    end
+
+    def self.safe_create_spec
+      content = <<-EOF
+require 'spec_helper'
+
+describe package('httpd'), :if => os[:family] == 'redhat' do
+  it { should be_installed }
+end
+
+describe package('apache2'), :if => os[:family] == 'ubuntu' do
+  it { should be_installed }
+end
+
+describe service('httpd'), :if => os[:family] == 'redhat' do
+  it { should be_enabled }
+  it { should be_running }
+end
+
+describe service('apache2'), :if => os[:family] == 'ubuntu' do
+  it { should be_enabled }
+  it { should be_running }
+end
+
+describe service('org.apache.httpd'), :if => os[:family] == 'darwin' do
+  it { should be_enabled }
+  it { should be_running }
+end
+
+describe port(80) do
+  it { should be_listening }
+end
+EOF
+
+      if File.exists? "spec/#{@hostname}/sample_spec.rb"
+        old_content = File.read("spec/#{@hostname}/sample_spec.rb")
+        if old_content != content
+          $stderr.puts "!! spec/#{@hostname}/sample_spec.rb already exists and differs from template"
+        end
+      else
+        File.open("spec/#{@hostname}/sample_spec.rb", 'w') do |f|
+          f.puts content
+        end
+        puts " + spec/#{@hostname}/sample_spec.rb"
+      end
+    end
+
+    def self.safe_mkdir(dir)
+      if File.exists? dir
+        unless File.directory? dir
+          $stderr.puts "!! #{dir} already exists and is not a directory"
+        end
+      else
+        FileUtils.mkdir dir
+        puts " + #{dir}/"
+      end
+    end
+
+    def self.safe_create_spec_helper
+      content = ERB.new(spec_helper_template, nil, '-').result(binding)
+      if File.exists? 'spec/spec_helper.rb'
+        old_content = File.read('spec/spec_helper.rb')
+        if old_content != content
+          $stderr.puts "!! spec/spec_helper.rb already exists and differs from template"
+        end
+      else
+        File.open('spec/spec_helper.rb', 'w') do |f|
+          f.puts content
+        end
+        puts ' + spec/spec_helper.rb'
+      end
+    end
+
+    def self.safe_create_rakefile
+      content = <<-'EOF'
+require 'rake'
+require 'rspec/core/rake_task'
+
+task :spec    => 'spec:all'
+task :default => :spec
+
+namespace :spec do
+  targets = []
+  Dir.glob('./spec/*').each do |dir|
+    next unless File.directory?(dir)
+    targets << File.basename(dir)
+  end
+
+  task :all     => targets
+  task :default => :all
+
+  targets.each do |target|
+    desc "Run serverspec tests to #{target}"
+    RSpec::Core::RakeTask.new(target.to_sym) do |t|
+      ENV['TARGET_HOST'] = target
+      t.pattern = "spec/#{target}/*_spec.rb"
+    end
+  end
+end
+      EOF
+      if File.exists? 'Rakefile'
+        old_content = File.read('Rakefile')
+        if old_content != content
+          $stderr.puts '!! Rakefile already exists and differs from template'
+        end
+      else
+        File.open('Rakefile', 'w') do |f|
+          f.puts content
+        end
+        puts ' + Rakefile'
+      end
+    end
+
+    def self.find_vagrantfile
+      Pathname.new(Dir.pwd).ascend do |dir|
+        path = File.expand_path('Vagrantfile', dir)
+        return path if File.exists?(path)
+      end
+      nil
+    end
+
+    def self.auto_vagrant_configuration
+      if find_vagrantfile
+        vagrant_list = `vagrant status`
+        list_of_vms = []
+        if vagrant_list != ''
+          vagrant_list.each_line do |line|
+            if match = /([\w-]+[\s]+)(created|aborted|not created|poweroff|running|saved)[\s](\(virtualbox\)|\(vmware\)|\(vmware_fusion\))/.match(line)
+              list_of_vms << match[1].strip!
+            end
+          end
+          if list_of_vms.length == 1
+            @hostname = list_of_vms[0]
+          else
+            list_of_vms.each_with_index { |vm, index | puts "#{index}) #{vm}\n" }
+            print 'Choose a VM from the Vagrantfile: '
+            chosen_vm = $stdin.gets.chomp
+            @hostname = list_of_vms[chosen_vm.to_i]
+          end
+        else
+          $stderr.puts 'Vagrant status error - Check your Vagrantfile or .vagrant'
+          exit 1
+        end
+      else
+        $stderr.puts 'Vagrantfile not found in directory!'
+        exit 1
+      end
+    end
+
+    def self.spec_helper_template
+      template = <<-'EOF'
+require 'serverspec'
+<% if @backend_type == 'ssh' -%>
+require 'net/ssh'
+<% end -%>
+<%- if @vagrant -%>
+require 'tempfile'
+<% end -%>
+<% if @backend_type == 'winrm' -%>
+require 'winrm'
+<% end -%>
+
+set :backend, :<%= @backend_type %>
+
+<% if @os_type == 'UN*X' && @backend_type == 'ssh' -%>
+if ENV['ASK_SUDO_PASSWORD']
+  begin
+    require 'highline/import'
+  rescue LoadError
+    fail "highline is not available. Try installing it."
+  end
+  set :sudo_password, ask("Enter sudo password: ") { |q| q.echo = false }
+else
+  set :sudo_password, ENV['SUDO_PASSWORD']
+end
+
+<%- if @backend_type == 'ssh' -%>
+host = ENV['TARGET_HOST']
+
+<%- if @vagrant -%>
+`vagrant up #{host}`
+
+config = Tempfile.new('', Dir.tmpdir)
+`vagrant ssh-config #{host} > #{config.path}`
+
+options = Net::SSH::Config.for(host, [config.path])
+<%- else -%>
+options = Net::SSH::Config.for(host)
+<%- end -%>
+
+options[:user] ||= Etc.getlogin
+
+set :host,        options[:host_name] || host
+set :ssh_options, options
+
+# Disable sudo
+# set :disable_sudo, true
+<%- end -%>
+
+
+# Set environment variables
+# set :env, :LANG => 'C', :LC_MESSAGES => 'C' 
+
+# Set PATH
+# set :path, '/sbin:/usr/local/sbin:$PATH'
+<%- end -%>
+<% if @backend_type == 'winrm'-%>
+user = <username>
+pass = <password>
+endpoint = "http://#{ENV['TARGET_HOST']}:5985/wsman"
+
+winrm = ::WinRM::WinRMWebService.new(endpoint, :ssl, :user => user, :pass => pass, :basic_auth_only => true)
+winrm.set_timeout 300 # 5 minutes max timeout for any operation
+Specinfra.configuration.winrm = winrm
+<% end -%>
+EOF
+      template
+    end
+
+    def self.safe_create_dotrspec
+      content = <<-'EOF'
+--color
+--format documentation
+      EOF
+      if File.exists? '.rspec'
+        old_content = File.read('.rspec')
+        if old_content != content
+          $stderr.puts '!! .rspec already exists and differs from template'
+        end
+      else
+        File.open('.rspec', 'w') do |f|
+          f.puts content
+        end
+        puts ' + .rspec'
+      end
+    end
+
+  end
+end