testlab 1.11.2 → 1.11.3

data/lib/testlab/container/actions.rb

@@ -13,6 +13,7 @@ class TestLab
         @ui.logger.debug { "Container Create: #{self.id}" }
 
         self.node.alive? or return false
+        self.node.ok?
 
         persistent_operation_check(:create)
 
@@ -36,6 +37,7 @@ class TestLab
         @ui.logger.debug { "Container Destroy: #{self.id}" }
 
         self.node.alive? or return false
+        self.node.ok?
 
         please_wait(:ui => @ui, :message => format_object_action(self, 'Destroy', :red)) do
           self.lxc.destroy(%(-f))
@@ -56,6 +58,7 @@ class TestLab
         @ui.logger.debug { "Container Up: #{self.id}" }
 
         self.node.alive? or return false
+        self.node.ok?
 
         please_wait(:ui => @ui, :message => format_object_action(self, 'Up', :green)) do
           configure
@@ -101,6 +104,7 @@ class TestLab
         @ui.logger.debug { "Container Down: #{self.id}" }
 
         self.node.alive? or return false
+        self.node.ok?
 
         please_wait(:ui => @ui, :message => format_object_action(self, 'Down', :red)) do
 

data/lib/testlab/container/io.rb

@@ -8,9 +8,9 @@ class TestLab
 
       PBZIP2_MEMORY    = 1024
       READ_SIZE        = ((64 * 1024) - 1)
-      TRANSFER_MESSAGE = "transferring '%s' at %0.2fMB/s -- %0.2fMB of %0.2fMB -- %d%% (%01d:%02dT-%01d:%02d)   \r"
+      TRANSFER_MESSAGE = "%s '%s' at %0.2fMB/s -- %0.2fMB of %0.2fMB -- %d%% (%01d:%02dT-%01d:%02d)   \r"
 
-      def transfer_message(filename, current_size, total_size, elapsed)
+      def transfer_message(what, filename, current_size, total_size, elapsed)
         total_size_mb   = (total_size.to_f / (1024 * 1024).to_f)
         current_size_mb = (current_size.to_f / (1024 * 1024).to_f)
 
@@ -27,7 +27,7 @@ class TestLab
 
         percentage_done = ((current_size * 100) / total_size)
 
-        @ui.stdout.print(format_message(TRANSFER_MESSAGE.yellow % [File.basename(filename), speed_mb, current_size_mb, total_size_mb, percentage_done, minutes, seconds, est_minutes, est_seconds]))
+        @ui.stdout.print(format_message(TRANSFER_MESSAGE.yellow % [what, File.basename(filename), speed_mb, current_size_mb, total_size_mb, percentage_done, minutes, seconds, est_minutes, est_seconds]))
       end
 
       def progress_callback(action, args)
@@ -44,7 +44,7 @@ class TestLab
           elapsed      = (Time.now - @start_time)
           current_size = (args[1] + args[2].length)
 
-          transfer_message(args[0].local, current_size, @total_size, elapsed)
+          transfer_message(%(uploading to node:), args[0].local, current_size, @total_size, elapsed)
 
         when :finish
           @ui.stdout.puts
@@ -59,6 +59,9 @@ class TestLab
       def export(compression=9, local_file=nil)
         @ui.logger.debug { "Container Export: #{self.id} " }
 
+        self.node.alive? or return false
+        self.node.ok?
+
         (self.state == :not_created) and raise ContainerError, 'You must create a container before you can export it!'
 
         # Throw an exception if we are attempting to export a container in a
@@ -119,6 +122,9 @@ EOF
       def import(local_file)
         @ui.logger.debug { "Container Import: #{self.id}" }
 
+        self.node.alive? or return false
+        self.node.ok?
+
         import_tempfile = Tempfile.new('import')
         remote_filename = File.basename(import_tempfile.path.dup)
         import_tempfile.close!
@@ -254,7 +260,7 @@ EOF
               elapsed  = (Time.now - start_time)
               current_size += chunk.size
 
-              transfer_message(local_file, current_size, total_size, elapsed)
+              transfer_message(%(downloading locally:), local_file, current_size, total_size, elapsed)
             end
             @ui.stdout.puts
 

data/lib/testlab/container/provision.rb

@@ -13,6 +13,7 @@ class TestLab
         @ui.logger.debug { "Container Provision: #{self.id} " }
 
         self.node.alive? or return false
+        self.node.ok?
 
         please_wait(:ui => @ui, :message => format_object_action(self, :provision, :green)) do
           do_provisioner_callbacks(self, :provision, @ui)
@@ -31,6 +32,7 @@ class TestLab
         @ui.logger.debug { "Container Deprovision: #{self.id} " }
 
         self.node.alive? or return false
+        self.node.ok?
 
         persistent_operation_check(:deprovision)
 

data/lib/testlab/node/status.rb

@@ -20,6 +20,22 @@ class TestLab
         }
       end
 
+      def ok?
+        result = true
+
+        # make sure the node has some free space
+        free_space_percent = (self.exec(%(df -P /), :ignore_exit_status => true).output.split("\n")[1].split[-2].to_i rescue nil)
+        if free_space_percent.nil?
+          @ui.stderr.puts(format_message("ERROR: We could not determine how much free space node #{self.id.inspect} has!".red.bold))
+          result = false
+        elsif (free_space_percent >= 90)
+          @ui.stderr.puts(format_message("WARNING: Your TestLab node #{self.id.inspect} is using #{free_space_percent}% of its available disk space!".red.bold))
+          result = false
+        end
+
+        result
+      end
+
     end
 
   end

data/lib/testlab/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: testlab
 version: !ruby/object:Gem::Version
-  version: 1.11.2
+  version: 1.11.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-11-04 00:00:00.000000000 Z
+date: 2013-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gli