testlab 1.8.5 → 1.9.0

data/lib/testlab/provisioner.rb

@@ -11,6 +11,7 @@ class TestLab
     autoload :AptCacherNG,  'testlab/provisioners/apt_cacher_ng'
     autoload :Bind,         'testlab/provisioners/bind'
     autoload :Chef,         'testlab/provisioners/chef'
+    autoload :HostsFile,    'testlab/provisioners/hosts_file'
     autoload :NFSMount,     'testlab/provisioners/nfs_mount'
     autoload :Raring,       'testlab/provisioners/raring'
     autoload :Resolv,       'testlab/provisioners/resolv'

data/lib/testlab/provisioners/hosts_file.rb

@@ -0,0 +1,84 @@
+class TestLab
+
+  class Provisioner
+
+    # HostsFile Provisioner Error Class
+    class HostsFileError < ProvisionerError; end
+
+    # HostsFile Provisioner Class
+    #
+    # @author Zachary Patten <zachary AT jovelabs DOT com>
+    class HostsFile
+
+      def initialize(config={}, ui=nil)
+        @config  = (config || Hash.new)
+        @ui      = (ui     || TestLab.ui)
+        @command = ZTK::Command.new(:ui => @ui, :silence => true, :ignore_exit_status => true)
+
+        @ui.logger.debug { "config(#{@config.inspect})" }
+      end
+
+      # HostsFile: Container Provision
+      def on_container_callback(container)
+        remove_hosts
+        add_hosts(container)
+
+        true
+      end
+      alias :on_container_create :on_container_callback
+      alias :on_container_up :on_container_callback
+      alias :on_container_provision :on_container_callback
+
+      alias :on_container_deprovision:on_container_callback
+      alias :on_container_down :on_container_callback
+      alias :on_container_destroy :on_container_callback
+
+    private
+
+      def add_hosts(container)
+        @command.exec(<<-EOF)
+set -x
+cat <<EOI | #{sudo} tee -a /etc/hosts
+#{def_tag}
+#{hosts_blob(container)}
+#{end_tag}
+EOI
+        EOF
+      end
+
+      def remove_hosts
+        @command.exec(sed_hostsfile)
+      end
+
+      def hosts_blob(container)
+        blob = Array.new
+        container.node.containers.each do |con|
+          blob << "#{con.primary_interface.ip}\t#{con.id} #{con.fqdn}"
+        end
+
+        blob.join("\n")
+      end
+
+      def sed_hostsfile
+        case RUBY_PLATFORM
+        when /darwin/ then
+          %(#{sudo} sed -i '' '/#{def_tag}/,/#{end_tag}/d' /etc/hosts)
+        when /linux/ then
+          %(#{sudo} sed -i '/#{def_tag}/,/#{end_tag}/d' /etc/hosts)
+        end
+      end
+
+      # NFS Exports Start Definition Tag
+      def def_tag
+        "#TESTLAB-HOSTSFILE"
+      end
+
+      # NFS Exports End Definition Tag
+      def end_tag
+        "#TESTLAB-HOSTSFILE"
+      end
+
+    end
+
+  end
+end

data/lib/testlab/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: testlab
 version: !ruby/object:Gem::Version
-  version: 1.8.5
+  version: 1.9.0
   prerelease: 
 platform: ruby
 authors:
@@ -286,6 +286,7 @@ files:
 - lib/testlab/provisioners/chef/omni_truck.rb
 - lib/testlab/provisioners/chef/ruby_gem_client.rb
 - lib/testlab/provisioners/chef/ruby_gem_server.rb
+- lib/testlab/provisioners/hosts_file.rb
 - lib/testlab/provisioners/nfs_mount.rb
 - lib/testlab/provisioners/raring.rb
 - lib/testlab/provisioners/resolv.rb
@@ -344,7 +345,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3128709550097862356
+      hash: 1545777099270245561
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -353,7 +354,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3128709550097862356
+      hash: 1545777099270245561
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.25