cucumber-chef 1.0.3 → 2.0.0.pre

data/lib/cucumber/chef/helpers/command.rb

@@ -0,0 +1,57 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber::Chef::Helpers::Command
+
+################################################################################
+
+  def command_run_remote(name, command, expected_exit_code=0)
+    output = %x(ssh #{name} '#{command}' 2>&1)
+    raise "command_run_remote(#{command}) failed (#{$?})" if ($? != expected_exit_code)
+    output
+#  rescue RuntimeError => e
+#    if $? == 65280
+#      puts "Exit Code #{$?}: Retrying..."
+#      retry
+#    end
+  end
+
+################################################################################
+
+  def command_run_chroot(name, command, expected_exit_code=0)
+    output = %x(chroot #{container_root(name)} /bin/bash -c '#{command}' 2>&1)
+    raise "command_run_chroot(#{command}) failed (#{$?})" if ($? != expected_exit_code)
+    output
+  end
+
+################################################################################
+
+  def command_run_local(command, expected_exit_code=0)
+    output = %x(#{command} 2>&1)
+    raise "command_run_local(#{command}) failed (#{$?})" if ($? != expected_exit_code)
+    output
+  end
+
+################################################################################
+
+end
+
+################################################################################

data/lib/cucumber/chef/helpers/container.rb

@@ -0,0 +1,154 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber::Chef::Helpers::Container
+
+################################################################################
+
+  def container_create(name, distro, release, arch)
+    unless container_exists?(name)
+      cache_rootfs = container_cache_root(name, distro, release, arch)
+      log(name, "has triggered first time lxc distro cache build; this will take a while") if !File.exists?(cache_rootfs)
+
+      command_run_local(container_create_command(name, distro, release, arch))
+
+      # install omnibus into the distro/release file cache if it's not already there
+      omnibus_chef_client = File.join("/", "opt", "opscode", "bin", "chef-client")
+      if !File.exists?(File.join(cache_rootfs, omnibus_chef_client))
+        case distro.downcase
+        when "ubuntu":
+          %x(chroot #{cache_rootfs} /bin/bash -c 'apt-get -y --force-yes install wget' 2>&1)
+        when "fedora":
+          %x( yum --nogpgcheck --installroot=#{cache_rootfs} -y install wget openssh-server )
+        end
+        %x( chroot #{cache_rootfs} /bin/bash -c 'wget http://opscode.com/chef/install.sh -O - | bash' 2>&1 )
+        if distro.downcase == "fedora"
+          %x( chroot #{cache_rootfs} /bin/bash -c 'rpm -Uvh --nodeps /tmp/*rpm' 2>&1 )
+        end
+        command_run_local("lxc-destroy -n #{name} 2>&1")
+        command_run_local(container_create_command(name, distro, release, arch))
+      end
+
+      command_run_local("mkdir -p #{container_root(name)}/root/.ssh")
+      command_run_local("chmod 0755 #{container_root(name)}/root/.ssh")
+      command_run_local("cat /root/.ssh/id_rsa.pub | tee -a #{container_root(name)}/root/.ssh/authorized_keys")
+      command_run_local("cat /home/ubuntu/.ssh/id_rsa.pub | tee -a #{container_root(name)}/root/.ssh/authorized_keys")
+
+      command_run_local("mkdir -p #{container_root(name)}/home/ubuntu/.ssh")
+      command_run_local("chmod 0755 #{container_root(name)}/home/ubuntu/.ssh")
+      command_run_local("cat /root/.ssh/id_rsa.pub | tee -a #{container_root(name)}/home/ubuntu/.ssh/authorized_keys")
+      command_run_local("cat /home/ubuntu/.ssh/id_rsa.pub | tee -a #{container_root(name)}/home/ubuntu/.ssh/authorized_keys")
+      command_run_local("chown -R ubuntu:ubuntu #{container_root(name)}/home/ubuntu/.ssh")
+
+      command_run_local("rm #{container_root(name)}/etc/motd")
+      command_run_local("cp /etc/motd #{container_root(name)}/etc/motd")
+      command_run_local("echo \"    You are now logged in to the LXC '#{name}'\\n\" >> #{container_root(name)}/etc/motd")
+      command_run_local("sed -i \"s/localhost #{name}/#{name}.test-lab #{name} localhost/\" #{container_root(name)}/etc/hosts")
+      command_run_local("echo \"#{name}.test-lab\" | tee #{container_root(name)}/etc/hostname")
+    end
+    container_start(name)
+  end
+
+  def container_destroy(name)
+    if container_exists?(name)
+      container_stop(name)
+      command_run_local("lxc-destroy -n #{name} 2>&1")
+      chef_server_node_destroy(name)
+      chef_server_client_destroy(name)
+    end
+  end
+
+################################################################################
+
+  def container_start(name)
+    status = command_run_local("lxc-info -n #{name} 2>&1")
+    if status.include?("STOPPED")
+      command_run_local("lxc-start -d -n #{name}")
+    end
+  end
+
+  def container_stop(name)
+    status = command_run_local("lxc-info -n #{name} 2>&1")
+    if status.include?("RUNNING")
+      command_run_local("lxc-stop -n #{name}")
+    end
+  end
+
+################################################################################
+
+  def container_running?(name)
+    status = command_run_local("lxc-info -n #{name} 2>&1")
+    status.include?("RUNNING")
+  end
+
+################################################################################
+
+  def container_config_network(name)
+    lxc_network_config = File.join("/etc/lxc", name)
+    File.open(lxc_network_config, 'w') do |f|
+      f.puts(Cucumber::Chef.generate_do_not_edit_warning("LXC Container Configuration"))
+      f.puts("")
+      f.puts("lxc.network.type = veth")
+      f.puts("lxc.network.flags = up")
+      f.puts("lxc.network.link = br0")
+      f.puts("lxc.network.name = eth0")
+      f.puts("lxc.network.hwaddr = #{$servers[name][:mac]}")
+      f.puts("lxc.network.ipv4 = 0.0.0.0")
+    end
+  end
+
+################################################################################
+
+  def containers
+    command_run_local("lxc-ls 2>&1").split("\n").uniq
+  end
+
+  def container_exists?(name)
+    (File.directory?(container_root(name)) ? true : false)
+  end
+
+  def container_root(name)
+    File.join("/", "var", "lib", "lxc", name, "rootfs")
+  end
+
+  def container_cache_root(name, distro, release, arch)
+    case distro.downcase
+    when "ubuntu":
+      cache_root = File.join("/", "var", "cache", "lxc", release, "rootfs-#{arch}")
+    when "fedora":
+      cache_root = File.join("/", "var", "cache", "lxc", distro, arch, release, "rootfs")
+    end
+  end
+
+  def container_create_command(name, distro, release, arch)
+    case distro.downcase
+    when "ubuntu":
+      "lxc-create -n #{name} -f /etc/lxc/#{name} -t #{distro} -- --release #{release} --arch #{arch}"
+    when "fedora":
+      "lxc-create -n #{name} -f /etc/lxc/#{name} -t #{distro} -- --release #{release}"
+    end
+  end
+
+################################################################################
+
+end
+
+################################################################################

data/lib/cucumber/chef/helpers/minitest.rb

@@ -0,0 +1,35 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber::Chef::Helpers::MiniTest
+
+  def enable_minitest(name)
+    @chef_client_attributes[:run_list].unshift("recipe[minitest-handler]")
+  end
+
+  def run_minitests(name)
+    chef_run = chef_run_client(name, "-l info")
+    test_result = chef_run.drop_while {|e| e !~ /^# Running tests/}.take_while {|e| e !~ /^[.*] INFO/}
+    puts test_result
+    test_result
+  end
+
+end

data/lib/cucumber/chef/helpers/server.rb

@@ -0,0 +1,81 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber::Chef::Helpers::Server
+
+################################################################################
+
+  def detect_arch(distro)
+    case distro.downcase
+    when "ubuntu":
+      ((RUBY_PLATFORM =~ /x86_64/) ? "amd64" : "i386")
+    when "fedora":
+      ((RUBY_PLATFORM =~ /x86_64/) ? "amd64" : "i686")
+    end
+  end
+
+  def server_create(name, attributes={})
+    if ((attributes[:persist] && $servers[name]) || ($servers[name] && $servers[name][:persist]))
+      attributes = $servers[name]
+    else
+      server_destroy(name) if (container_exists?(name) && (ENV['DESTROY'] == "1"))
+      attributes = { :ip => generate_ip,
+                     :mac => generate_mac,
+                     :persist => true,
+                     :distro => "ubuntu",
+                     :release => "lucid",
+                     :arch => detect_arch(attributes[:distro] || "ubuntu") }.merge(attributes)
+    end
+    $servers = ($servers || Hash.new(nil)).merge(name => attributes)
+    $current_server = $servers[name][:ip]
+    if !server_running?(name)
+      log(name, "is being provisioned") if $servers[name]
+
+      test_lab_config_dhcpd
+      container_config_network(name)
+      container_create(name, $servers[name][:distro], $servers[name][:release], $servers[name][:arch])
+      Cucumber::Chef::TCPSocket.new($servers[name][:ip], 22).wait
+    end
+  end
+
+  def server_destroy(name)
+    log(name, "is being destroyed") if $servers[name]
+
+    container_destroy(name)
+  end
+
+################################################################################
+
+  def server_running?(name)
+    container_running?(name)
+  end
+
+################################################################################
+
+  def servers
+    containers
+  end
+
+################################################################################
+
+end
+
+################################################################################

data/lib/cucumber/chef/helpers/test_lab.rb

@@ -0,0 +1,46 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber::Chef::Helpers::TestLab
+
+################################################################################
+
+  def test_lab_config_dhcpd
+    dhcpd_config = File.join("/etc/dhcp/test-lab.conf")
+    File.open(dhcpd_config, 'w') do |f|
+      f.puts(Cucumber::Chef.generate_do_not_edit_warning("DHCPD Configuration"))
+      $servers.each do |key, value|
+        f.puts
+        f.puts("host #{key} {")
+        f.puts("  hardware ethernet #{value[:mac]};")
+        f.puts("  fixed-address #{value[:ip]};")
+        f.puts("  ddns-hostname \"#{key}\";")
+        f.puts("}")
+      end
+    end
+    command_run_local("/etc/init.d/isc-dhcp-server restart")
+  end
+
+################################################################################
+
+end
+
+################################################################################

data/lib/cucumber/chef/helpers/utility.rb

@@ -0,0 +1,73 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber::Chef::Helpers::Utility
+
+################################################################################
+
+  def log(name, message)
+    STDOUT.puts("\033[34m  >>> \033[1m#{name}\033[0m\033[34m #{message}\033[0m")
+    STDOUT.flush if STDOUT.respond_to?(:flush)
+  end
+
+################################################################################
+
+  def generate_ip
+    octets = [ 192..192,
+               168..168,
+               0..254,
+               1..254 ]
+    ip = ""
+    for x in 1..4 do
+      ip += octets[x-1].to_a[rand(octets[x-1].count)].to_s
+      ip += "." if x != 4
+    end
+    ip
+  end
+
+################################################################################
+
+  def generate_mac
+    digits = [ %w( 0 ),
+               %w( 0 ),
+               %w( 0 ),
+               %w( 0 ),
+               %w( 5 ),
+               %w( e ),
+               %w( 0 1 2 3 4 5 6 7 8 9 a b c d e f ),
+               %w( 0 1 2 3 4 5 6 7 8 9 a b c d e f ),
+               %w( 5 6 7 8 9 a b c d e f ),
+               %w( 3 4 5 6 7 8 9 a b c d e f ),
+               %w( 0 1 2 3 4 5 6 7 8 9 a b c d e f ),
+               %w( 0 1 2 3 4 5 6 7 8 9 a b c d e f ) ]
+    mac = ""
+    for x in 1..12 do
+      mac += digits[x-1][rand(digits[x-1].count)]
+      mac += ":" if (x.modulo(2) == 0) && (x != 12)
+    end
+    mac
+  end
+
+################################################################################
+
+end
+
+################################################################################

data/lib/cucumber/chef/helpers.rb

@@ -0,0 +1,56 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+module Cucumber
+  module Chef
+
+    module Helpers
+
+################################################################################
+
+      require 'cucumber/chef/helpers/chef_client'
+      require 'cucumber/chef/helpers/chef_server'
+      require 'cucumber/chef/helpers/command'
+      require 'cucumber/chef/helpers/container'
+      require 'cucumber/chef/helpers/server'
+      require 'cucumber/chef/helpers/test_lab'
+      require 'cucumber/chef/helpers/utility'
+
+################################################################################
+
+      def self.included(base)
+        base.send(:include, Cucumber::Chef::Helpers::ChefClient)
+        base.send(:include, Cucumber::Chef::Helpers::ChefServer)
+        base.send(:include, Cucumber::Chef::Helpers::Command)
+        base.send(:include, Cucumber::Chef::Helpers::Container)
+        base.send(:include, Cucumber::Chef::Helpers::Server)
+        base.send(:include, Cucumber::Chef::Helpers::TestLab)
+        base.send(:include, Cucumber::Chef::Helpers::Utility)
+      end
+
+################################################################################
+
+    end
+
+  end
+end
+
+################################################################################

data/lib/cucumber/chef/logger.rb

@@ -0,0 +1,90 @@
+################################################################################
+#
+#      Author: Stephen Nelson-Smith <stephen@atalanta-systems.com>
+#      Author: Zachary Patten <zachary@jovelabs.com>
+#   Copyright: Copyright (c) 2011-2012 Atalanta Systems Ltd
+#     License: Apache License, Version 2.0
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+################################################################################
+
+
+module Cucumber
+  module Chef
+
+    class LoggerError < Error; end
+
+    class Logger < ::Logger
+
+      SEVERITIES = Severity.constants.inject([]) {|arr,c| arr[Severity.const_get(c)] = c; arr}
+
+################################################################################
+
+      def initialize(file=nil)
+        if file.nil?
+          config_path = File.join(Cucumber::Chef.locate_parent(".chef"), ".cucumber-chef")
+          FileUtils.mkdir_p(config_path)
+          file = File.join(config_path, "cucumber-chef.log")
+        end
+
+        #super(file, 7, (1024 * 1024))
+        super(file)
+        set_log_level
+      end
+
+################################################################################
+
+      def parse_caller(at)
+        if /^(.+?):(\d+)(?::in `(.*)')?/ =~ at
+          file = Regexp.last_match[1]
+          line = Regexp.last_match[2]
+          method = Regexp.last_match[3]
+          "#{File.basename(file)}:#{line}:#{method} | "
+        else
+          ""
+        end
+      end
+
+################################################################################
+
+      def add(severity, message = nil, progname = nil, &block)
+        return if (@level > severity)
+
+        called_by = parse_caller(caller[1])
+
+        msg = (block && block.call)
+        return if (msg.nil? || msg.strip.empty?)
+        message = [message, progname, msg].delete_if{|i| i == nil}.join(": ")
+        message = "%19s.%06d | %5s | %5s | %s%s\n" % [Time.now.utc.strftime("%Y-%m-%d %H:%M:%S"), Time.now.utc.usec, Process.pid.to_s, SEVERITIES[severity], called_by, message]
+
+        @logdev.write(message)
+
+        true
+      end
+
+################################################################################
+
+      def set_log_level(level="INFO")
+        log_level = (ENV['LOG_LEVEL'] || level)
+        self.level = Cucumber::Chef::Logger.const_get(log_level.to_s.upcase)
+      end
+
+################################################################################
+
+    end
+
+  end
+end
+
+################################################################################