testlab 0.3.1 → 0.4.0

data/lib/testlab/provisioners/templates/chef/omnitruck.erb

@@ -0,0 +1,102 @@
+#!/bin/env bash
+<%= ZTK::Template.do_not_edit_notice(:message => "OmniTruck Bootstrap") %>
+set -x
+
+DEBIAN_FRONTEND="noninteractive"
+CHEF_SOLO_ROOT="/tmp/chef-solo"
+KNIFE_CONFIG_EXP_FILE="/tmp/knife-config.exp"
+export DEBIAN_FRONTEND CHEF_SOLO_ROOT KNIFE_CONFIG_EXP_FILE
+
+mkdir -p /tmp/chef-solo/
+cd ${CHEF_SOLO_ROOT}
+
+apt-get -y --force-yes update
+apt-get -y --force-yes install build-essential expect wget curl
+
+mkdir -pv /etc/chef /var/log/chef ${HOME}/.chef
+
+cat <<EOF | tee /etc/chef/solo.rb
+file_cache_path "${CHEF_SOLO_ROOT}/"
+cookbook_path %w(${CHEF_SOLO_ROOT}/cookbooks/ ${CHEF_SOLO_ROOT}/site-cookbooks/)
+role_path "${CHEF_SOLO_ROOT}/roles/"
+EOF
+
+cat <<EOF | tee ${CHEF_SOLO_ROOT}/attributes.json
+<%= @chef_solo_attributes.to_json %>
+EOF
+
+curl -L https://www.opscode.com/chef/install.sh | bash
+# bash install.sh -v <%= @version %>
+<% if (@chef_pre_11 == false) -%>
+
+mkdir -pv /var/chef/cache ${CHEF_SOLO_ROOT}/cookbooks/chef-server
+wget -qO- https://github.com/opscode-cookbooks/chef-server/archive/master.tar.gz | tar xvzC ${CHEF_SOLO_ROOT}/cookbooks/chef-server --strip-components=1
+<% else -%>
+
+export PATH=/opt/chef/embedded/bin:$PATH
+
+cat <<EOF | tee ${CHEF_SOLO_ROOT}/Gemfile
+source 'https://rubygems.org'
+gem 'chef', '<%= @version %>'
+gem 'berkshelf'
+gem 'moneta', '< 0.7.0'
+EOF
+
+cat <<EOF | tee ${CHEF_SOLO_ROOT}/Berksfile
+site :opscode
+cookbook 'chef-server'
+EOF
+
+gem install bundler --no-ri --no-rdoc
+bundle install --standalone --path vendor/bundle --binstubs
+mv -v ${CHEF_SOLO_ROOT}/cookbooks ${CHEF_SOLO_ROOT}/site-cookbooks
+bin/berks install --path ${CHEF_SOLO_ROOT}/cookbooks/
+<% end -%>
+
+chef-solo --config /etc/chef/solo.rb --json-attributes ${CHEF_SOLO_ROOT}/attributes.json --logfile /var/log/chef/chef-solo.log --log_level debug
+<% if (@chef_pre_11 == false) -%>
+
+cat << EOF | tee /etc/chef-server/chef-server.rb
+<%= ZTK::Template.do_not_edit_notice(:message => "OmniTruck Bootstrap") %>
+server_name = "<%= @server_name %>"
+#
+topology                "standalone"
+api_fqdn                server_name
+nginx['url']          = "https://#{server_name}"
+nginx['server_name']  = server_name
+lb['fqdn']            = server_name
+bookshelf['vip']      = server_name
+EOF
+sudo chef-server-ctl reconfigure
+<% end -%>
+
+echo -n "Waiting on <%= File.basename(@chef_validator) %> and <%= File.basename(@chef_webui) %> to appear..."
+until [ -f <%= @chef_validator %> ] && [ -f <%= @chef_webui %> ]; do
+  echo -n "."
+  sleep 1
+done
+echo "done."
+
+cp -v <%= @chef_validator %> <%= @chef_webui %> ~/.chef
+<% if (@chef_pre_11 == false) -%>
+
+ln -sv <%= @chef_validator %> /etc/chef/validation.pem
+ln -sv <%= @chef_admin %> /etc/chef/admin.pem
+<% end -%>
+
+cat <<EOF | tee ${KNIFE_CONFIG_EXP_FILE}
+#!/usr/bin/expect -f
+set timeout 10
+<% if (@chef_pre_11 == false) -%>
+spawn knife configure -i --server-url https://127.0.0.1 --admin-client-key <%= @chef_admin %> -u ${SUDO_USER} -r '' --defaults --yes -VV
+<% else -%>
+spawn knife configure -i --server-url https://127.0.0.1 -u ${SUDO_USER} -r '' --defaults --yes -VV
+<% end -%>
+expect "Please enter a password for the new user:" { send "<%= @default_password %>\n" }
+interact
+EOF
+chmod +x ${KNIFE_CONFIG_EXP_FILE}
+${KNIFE_CONFIG_EXP_FILE}
+
+knife client create <%= @user %> -a -f ${HOME}/.chef/<%= @user %>.pem --disable-editing --yes -VV
+chown -Rv ${SUDO_USER}:${SUDO_USER} ${HOME}

data/lib/testlab/version.rb

@@ -1,6 +1,6 @@
 class TestLab
   unless const_defined?(:VERSION)
     # TestLab Gem Version
-    VERSION = "0.3.1"
+    VERSION = "0.4.0"
   end
 end

data/lib/testlab.rb

@@ -27,6 +27,7 @@ require 'testlab/monkeys'
 #
 #   node :localhost do
 #     components %w(resolv bind)
+#     route true
 #
 #     provider    TestLab::Provider::Vagrant
 #     config      Hash[
@@ -54,10 +55,7 @@ require 'testlab/monkeys'
 #       release       "precise"
 #
 #       provisioner   TestLab::Provisioner::Shell
-#       config        Hash[
-#         :shell => "/bin/bash",
-#         :setup => shell_provision_script
-#       ]
+#       config        Hash[:setup => shell_provision_script]
 #
 #       interface do
 #         name       :eth0

data/spec/container_spec.rb

@@ -21,7 +21,11 @@ require "spec_helper"
 
 describe TestLab::Container do
 
-  subject { @testlab = TestLab.new(:labfile => LABFILE); @testlab.containers.first }
+  subject {
+    @ui = ZTK::UI.new(:stdout => StringIO.new, :stderr => StringIO.new)
+    @testlab = TestLab.new(:labfile => LABFILE, :ui => @ui)
+    @testlab.containers.first
+  }
 
   describe "class" do
 
@@ -29,18 +33,22 @@ describe TestLab::Container do
       subject.should be_an_instance_of TestLab::Container
     end
 
-  end
-
-  describe "methods" do
+    describe "methods" do
 
-    describe "domains" do
-      it "should return the domains for all defined containers" do
-        subject.class.domains.should be_kind_of(Array)
-        subject.class.domains.should_not be_empty
-        subject.class.domains.should == ["default.zone"]
+      describe "domains" do
+        it "should return the domains for all defined containers" do
+          subject.class.domains.should be_kind_of(Array)
+          subject.class.domains.should_not be_empty
+          subject.class.domains.should == ["default.zone"]
+        end
       end
+
     end
 
+  end
+
+  describe "methods" do
+
     describe "#status" do
       it "should return a hash of status information about the container" do
         subject.lxc.stub(:state) { :not_created }
@@ -58,7 +66,7 @@ describe TestLab::Container do
 
     describe "#fqdn" do
       it "should return the FQDN for the container" do
-        subject.fqdn.should == "server-1.default.zone"
+        subject.fqdn.should == "server-dual-nic.default.zone"
       end
     end
 
@@ -169,23 +177,50 @@ describe TestLab::Container do
     end
 
     describe "#setup" do
-      it "should create and online the container" do
-        subject.stub(:create) { true }
-        subject.stub(:up) { true }
-        subject.instance_variable_get(:@provisioner) and subject.instance_variable_get(:@provisioner).stub(:setup) { true }
+      context "with no provisioner" do
+        it "should create and online the container" do
+          subject.stub(:create) { true }
+          subject.stub(:up) { true }
+          subject.instance_variable_get(:@provisioner) and subject.instance_variable_get(:@provisioner).stub(:setup) { true }
 
-        subject.setup
+          subject.setup
+        end
+      end
+
+      context "with the shell provisioner" do
+        it "should create and online the container" do
+          subject = TestLab::Container.first('server-shell')
+          subject.stub(:create) { true }
+          subject.stub(:up) { true }
+          subject.instance_variable_get(:@provisioner) and subject.instance_variable_get(:@provisioner).stub(:setup) { true }
+
+          subject.setup
+        end
       end
     end
 
     describe "#teardown" do
-      it "should create and online the container" do
-        subject.stub(:down) { true }
-        subject.stub(:destroy) { true }
-        subject.instance_variable_get(:@provisioner) and subject.instance_variable_get(:@provisioner).stub(:teardown) { true }
+      context "with no provisioner" do
+        it "should create and online the container" do
+          subject.stub(:down) { true }
+          subject.stub(:destroy) { true }
+          subject.instance_variable_get(:@provisioner) and subject.instance_variable_get(:@provisioner).stub(:teardown) { true }
 
-        subject.teardown
+          subject.teardown
+        end
       end
+
+      context "with the shell provisioner" do
+        it "should create and online the container" do
+          subject = TestLab::Container.first('server-shell')
+          subject.stub(:down) { true }
+          subject.stub(:destroy) { true }
+          subject.instance_variable_get(:@provisioner) and subject.instance_variable_get(:@provisioner).stub(:teardown) { true }
+
+          subject.teardown
+        end
+      end
+
     end
 
   end

data/spec/network_spec.rb

@@ -21,7 +21,11 @@ require "spec_helper"
 
 describe TestLab::Network do
 
-  subject { @testlab = TestLab.new(:labfile => LABFILE); @testlab.networks.first }
+  subject {
+    @ui = ZTK::UI.new(:stdout => StringIO.new, :stderr => StringIO.new)
+    @testlab = TestLab.new(:labfile => LABFILE, :ui => @ui)
+    @testlab.networks.first
+  }
 
   describe "class" do
 

data/spec/node_spec.rb

@@ -21,7 +21,11 @@ require "spec_helper"
 
 describe TestLab::Node do
 
-  subject { @testlab = TestLab.new(:labfile => LABFILE); @testlab.nodes.first }
+  subject {
+    @ui = ZTK::UI.new(:stdout => StringIO.new, :stderr => StringIO.new)
+    @testlab = TestLab.new(:labfile => LABFILE, :ui => @ui)
+    @testlab.nodes.first
+  }
 
   describe "class" do
 
@@ -39,6 +43,22 @@ describe TestLab::Node do
       end
     end
 
+    describe "#bind_setup" do
+      it "should setup bind" do
+        subject.ssh.stub(:exec) { true }
+        subject.ssh.stub(:file).and_yield(StringIO.new)
+        subject.bind_setup
+      end
+    end
+
+    describe "#build_resolv_conf" do
+      it "should build resolv.conf" do
+        subject.ssh.stub(:exec) { true }
+        subject.ssh.stub(:file).and_yield(StringIO.new)
+        subject.build_resolv_conf
+      end
+    end
+
     describe "#status" do
       it "should return a hash of status information about the node" do
         subject.instance_variable_get(:@provider).stub(:state) { :not_created }
@@ -90,6 +110,36 @@ describe TestLab::Node do
       end
     end
 
+    describe "setup" do
+      it "should setup the node" do
+        subject.ssh.stub(:bootstrap) { true }
+        subject.stub(:route_setup) { true }
+        subject.stub(:build_resolv_conf) { true }
+        subject.stub(:bind_setup) { true }
+        subject.stub(:bind_reload) { true }
+        subject.containers.each do |container|
+          container.stub(:setup) { true }
+        end
+        subject.networks.each do |network|
+          network.stub(:setup) { true }
+        end
+        subject.setup
+      end
+    end
+
+    describe "teardown" do
+      it "should teardown the node" do
+        subject.stub(:route_setup) { true }
+        subject.containers.each do |container|
+          container.stub(:teardown) { true }
+        end
+        subject.networks.each do |network|
+          network.stub(:teardown) { true }
+        end
+        subject.teardown
+      end
+    end
+
   end
 
 end

data/spec/provisioners/shell_spec.rb

@@ -0,0 +1,66 @@
+################################################################################
+#
+#      Author: Zachary Patten <zachary AT jovelabs DOT com>
+#   Copyright: Copyright (c) Zachary Patten
+#     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.
+#
+################################################################################
+require "spec_helper"
+
+describe TestLab::Provisioner::Shell do
+
+  subject {
+    @ui = ZTK::UI.new(:stdout => StringIO.new, :stderr => StringIO.new)
+    @testlab = TestLab.new(:labfile => LABFILE, :ui => @ui)
+    TestLab::Container.first('server-shell')
+  }
+
+  describe "class" do
+
+    it "should be an instance of TestLab::Provisioner::Shell" do
+      subject.instance_variable_get(:@provisioner).should be_an_instance_of TestLab::Provisioner::Shell
+    end
+
+  end
+
+  describe "methods" do
+
+    describe "setup" do
+      context "bootstrap successful" do
+        it "should provision the container" do
+          subject.node.ssh.stub(:file).and_yield(StringIO.new)
+          subject.lxc.stub(:attach) { "" }
+          subject.instance_variable_get(:@provisioner).setup(subject)
+        end
+      end
+
+      context "bootstrap fails" do
+        it "should raise an exception" do
+          subject.node.ssh.stub(:file).and_yield(StringIO.new)
+          subject.lxc.stub(:attach) { "No such file or directory" }
+          lambda{ subject.instance_variable_get(:@provisioner).setup(subject) }.should raise_error TestLab::ContainerError
+        end
+      end
+    end
+
+    describe "teardown" do
+      it "should decomission the container" do
+        subject.instance_variable_get(:@provisioner).teardown(subject)
+      end
+    end
+
+  end
+
+end

data/spec/support/Labfile

@@ -1,12 +1,21 @@
 #!/usr/bin/env ruby
 #^syntax detection
 
+shell_provision_script = <<-EOF
+set -x
+apt-get -y update
+apt-get -y install dnsutils
+EOF
+
 config Hash[
   :domain => "default.zone"
 ]
 
 node :localhost do
 
+  components %w(resolv bind)
+  route true
+
   provider    TestLab::Provider::Vagrant
   config      Hash[
     :vagrant => {
@@ -20,14 +29,13 @@ node :localhost do
     :repo => File.dirname(__FILE__)
   ]
 
-  route        false
-
   network :testnet do
     address '192.168.255.254/16'
     bridge  :br0
   end
 
-  container "server-1" do
+  # DUAL NICs
+  container "server-dual-nic" do
     distro        "ubuntu"
     release       "precise"
 
@@ -45,7 +53,8 @@ node :localhost do
     end
   end
 
-  container "server-2" do
+  # SINGLE NIC
+  container "server-single-nic" do
     distro        "ubuntu"
     release       "precise"
 
@@ -56,4 +65,23 @@ node :localhost do
     end
   end
 
+  # SHELL PROVISIONER
+  container "server-shell" do
+    distro        "ubuntu"
+    release       "precise"
+
+    provisioner   TestLab::Provisioner::Shell
+    config        Hash[
+      :shell => "/bin/bash",
+      :setup => shell_provision_script
+    ]
+
+    interface do
+      name       :eth0
+      network_id :testnet
+      address    '192.168.0.200/16'
+    end
+  end
+
+
 end

data/spec/testlab_spec.rb

@@ -21,7 +21,10 @@ require "spec_helper"
 
 describe TestLab do
 
-  subject { TestLab.new(:labfile => LABFILE) }
+  subject {
+    @ui = ZTK::UI.new(:stdout => StringIO.new, :stderr => StringIO.new)
+    @testlab = TestLab.new(:labfile => LABFILE, :ui => @ui)
+  }
 
   describe "class" do
 

data/spec/utility_spec.rb

@@ -0,0 +1,114 @@
+################################################################################
+#
+#      Author: Zachary Patten <zachary AT jovelabs DOT com>
+#   Copyright: Copyright (c) Zachary Patten
+#     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.
+#
+################################################################################
+require "spec_helper"
+
+describe TestLab::Utility do
+
+  subject { TestLab::Utility }
+
+  describe "class" do
+
+    it "should be TestLab::Utility" do
+      subject.should be TestLab::Utility
+    end
+
+  end
+
+  describe "methods" do
+
+    describe "#cidr_octets" do
+      context "with CIDR 0" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/0"
+          subject.cidr_octets(address).should be_kind_of(Array)
+          subject.cidr_octets(address).should_not be_empty
+          subject.cidr_octets(address).should == ["192", "168", "0", "254"]
+        end
+      end
+
+      context "with CIDR 8" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/8"
+          subject.cidr_octets(address).should be_kind_of(Array)
+          subject.cidr_octets(address).should_not be_empty
+          subject.cidr_octets(address).should == ["168", "0", "254"]
+        end
+      end
+
+      context "with CIDR 16" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/16"
+          subject.cidr_octets(address).should be_kind_of(Array)
+          subject.cidr_octets(address).should_not be_empty
+          subject.cidr_octets(address).should == ["0", "254"]
+        end
+      end
+
+      context "with CIDR 24" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/24"
+          subject.cidr_octets(address).should be_kind_of(Array)
+          subject.cidr_octets(address).should_not be_empty
+          subject.cidr_octets(address).should == ["254"]
+        end
+      end
+    end
+
+    describe "#arpa_octets" do
+      context "with CIDR 0" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/0"
+          subject.arpa_octets(address).should be_kind_of(Array)
+          subject.arpa_octets(address).should be_empty
+          subject.arpa_octets(address).should == []
+        end
+      end
+
+      context "with CIDR 8" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/8"
+          subject.arpa_octets(address).should be_kind_of(Array)
+          subject.arpa_octets(address).should_not be_empty
+          subject.arpa_octets(address).should == ["192"]
+        end
+      end
+
+      context "with CIDR 16" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/16"
+          subject.arpa_octets(address).should be_kind_of(Array)
+          subject.arpa_octets(address).should_not be_empty
+          subject.arpa_octets(address).should == ["168", "192"]
+        end
+      end
+
+      context "with CIDR 24" do
+        it "should return the correct number of octets based on our CIDR" do
+          address = "192.168.0.254/24"
+          subject.arpa_octets(address).should be_kind_of(Array)
+          subject.arpa_octets(address).should_not be_empty
+          subject.arpa_octets(address).should == ["0", "168", "192"]
+        end
+      end
+    end
+
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: testlab
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-05 00:00:00.000000000 Z
+date: 2013-05-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gli
@@ -246,9 +246,11 @@ files:
 - lib/testlab/providers/templates/vagrant/Vagrantfile.erb
 - lib/testlab/providers/vagrant.rb
 - lib/testlab/provisioner.rb
-- lib/testlab/provisioners/chef.rb
+- lib/testlab/provisioners/omnibus.rb
+- lib/testlab/provisioners/omnitruck.rb
 - lib/testlab/provisioners/shell.rb
-- lib/testlab/provisioners/templates/chef/bootstrap.erb
+- lib/testlab/provisioners/templates/chef/omnibus.erb
+- lib/testlab/provisioners/templates/chef/omnitruck.erb
 - lib/testlab/utility.rb
 - lib/testlab/utility/cidr.rb
 - lib/testlab/utility/misc.rb
@@ -258,9 +260,11 @@ files:
 - spec/node_spec.rb
 - spec/provider_spec.rb
 - spec/provisioner_spec.rb
+- spec/provisioners/shell_spec.rb
 - spec/spec_helper.rb
 - spec/support/Labfile
 - spec/testlab_spec.rb
+- spec/utility_spec.rb
 - testlab.gemspec
 homepage: https://github.com/zpatten/testlab
 licenses:
@@ -277,7 +281,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -1882861325237330048
+      hash: 2598342112570301025
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -286,7 +290,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -1882861325237330048
+      hash: 2598342112570301025
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.25
@@ -299,7 +303,9 @@ test_files:
 - spec/node_spec.rb
 - spec/provider_spec.rb
 - spec/provisioner_spec.rb
+- spec/provisioners/shell_spec.rb
 - spec/spec_helper.rb
 - spec/support/Labfile
 - spec/testlab_spec.rb
+- spec/utility_spec.rb
 has_rdoc: