soda 0.0.5 → 0.0.6

data/lib/Soda.rb

@@ -46,6 +46,7 @@ require "SodaCSV"
 require "SodaXML"
 require 'SodaFireFox'
 require 'SodaTestCheck'
+require 'SodaScreenShot'
 require "utils/sodalookups"
 require "fields/SodaField"
 require "fields/TextField"
@@ -2667,6 +2668,17 @@ JSCode
                   @rep.ReportFailure(msg)
                   PrintDebug("Global Time was: #{$global_time}\n")
                   PrintDebug("Timeout Time was: #{time_check}\n")
+
+						begin
+							result_dir = @rep.GetResultDir()
+							shooter = SodaScreenShot.new(result_dir)
+							image_file = shooter.GetOutputFile()
+							@rep.log("ScreenShot taken: #{image_file}\n")
+						rescue Excaption => e
+            			@rep.ReportException(e, false, false)
+						ensure
+						end
+
                   result = -1
                   thread_soda.exit()
                   break

data/lib/SodaLogReporter.rb

@@ -614,6 +614,21 @@ HTML
       return row_data
    end
 
+   def FormatScreenShot(line)
+      row_data = Hash.new()
+      line =~ /\[(\d+\/\d+\/\d+-\d+:\d+:\d+)\](\(.\))(.*)/
+      row_data['date'] = "#{$1}"
+      row_data['msg_type'] = "#{$2}"
+      msg = "#{$3}"
+      row_html = ""
+
+		data = msg.split(/:/)
+		msg = "<b>#{data[0]}:</b> <a href=\"file://#{data[1]}\">#{data[1]}</a>"
+      row_data['msg'] = msg 
+
+      return row_data
+   end
+
 ###############################################################################
 # FormatReplacingString -- Method
 #     This method finds the replace string message and reformats it a little.
@@ -725,6 +740,8 @@ HTML
             row_data = FormatClickingElement(line)
          when /element:/i
             row_data = FormatClickingElement(line)
+			when /screenshot\staken/i
+				row_data = FormatScreenShot(line)
          else
             line =~ /\[(\d+\/\d+\/\d+-\d+:\d+:\d+)\](\(.\))(.*)/
             row_data['date'] = "#{$1}"

data/lib/SodaReporter.rb

@@ -113,6 +113,20 @@ class SodaReporter
       log("Saving HTML files => #{@saveHtmlFiles.to_s()}.\n")
 	end
 
+###############################################################################
+# GetResultDir -- Method
+#		This method returns the current result dir.
+#
+# Input:
+#		None.
+#
+# Output:
+#		returns the current result directory.
+#
+###############################################################################
+	def GetResultDir()
+		return @ResultsDir
+	end
 
 ###############################################################################
 # IncSkippedTest -- Method

data/lib/SodaScreenShot.rb

@@ -0,0 +1,174 @@
+###############################################################################
+# Copyright (c) 2010, SugarCRM, Inc.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#    * Redistributions of source code must retain the above copyright
+#      notice, this list of conditions and the following disclaimer.
+#    * Redistributions in binary form must reproduce the above copyright
+#      notice, this list of conditions and the following disclaimer in the
+#      documentation and/or other materials provided with the distribution.
+#    * Neither the name of SugarCRM, Inc. nor the
+#      names of its contributors may be used to endorse or promote products
+#      derived from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 
+# ARE DISCLAIMED. IN NO EVENT SHALL SugarCRM, Inc. BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+###############################################################################
+
+require 'SodaUtils'
+
+###############################################################################
+# SodaScreenShot -- Class
+#		This class is for taking desktop screen shots of the current running os.
+#  	Currently only Winows & Linux are supported.  Really any unix platform
+#		running X11 is supported, they just have to be added to the case.
+#
+# Input:
+#		dir: This is the directory to store the snapshots in.
+#
+###############################################################################
+class SodaScreenShot
+
+	LINUX_XWD_BIN = "xwd"
+	LINUX_CONVERT_BIN = "convert"
+	OUTPUT_FILE = "screenshot-"
+
+	def initialize(dir)
+		os = nil
+		err = nil
+		time = Time.now()
+		time = time.to_i()
+		hostname = "#{ENV['HOSTNAME']}"
+		@outputfile = nil
+
+		if (!File.exist?(dir))
+			raise "Failed to find needed diectory: '#{dir}!'\n"
+		end
+
+      if ( (hostname.empty?) || (hostname.length < 5) )
+         hostname = `hostname`
+         hostname = hostname.chomp()
+      end
+
+		os = SodaUtils.GetOsType()
+		case (os)
+			when /linux/i
+				xwd = FindLinuxXwd()
+				if (xwd == nil)
+					raise "Failed to find needed program: 'xwd'!\n"
+				end
+				
+				@outputfile = "#{dir}/#{OUTPUT_FILE}#{time}-#{hostname}.xwd"
+				cmd = "#{xwd} -root -out #{@outputfile}"
+				err = Kernel.system(cmd)
+				if (!err)
+					raise "Failed trying to take screenshot!\n"
+				end
+
+				convert = FindLinuxConvert()
+				if (convert != nil)
+					old_outfile = @outputfile
+					ext = File.extname(old_outfile)
+					@outputfile = @outputfile.gsub(/#{ext}$/, "")
+					@outputfile << ".png"
+					cmd = "#{convert} #{old_outfile} #{@outputfile}"
+					err = Kernel.system(cmd)
+					if (!err)
+						@outputfile = old_outfile
+					else
+						File.unlink(old_outfile)
+					end
+				end
+			when /windows/i
+				require 'Win32/screenshot'
+				@outputfile = "#{dir}/#{OUTPUT_FILE}#{time}-#{hostname}.bmp"
+				Win32::Screenshot.desktop do |width, height, bmp|
+					fd = File.new(@outputfile, "w+")
+					fd.write(bmp)
+					fd.close()
+				end
+		end
+	end
+
+###############################################################################
+# GetOutputFile -- Method
+#		This is a getter for getting the newly created snapshot file.
+#
+# Input:
+#		None.
+#
+# Output:
+#		retutns a string with the full path and filename to the new file.
+#
+###############################################################################
+	def GetOutputFile()
+		return @outputfile
+	end
+
+###############################################################################
+# FindLinuxXwd -- Method
+#		This method trys to find the xwd X11 util in you current path.
+#
+# Input:
+#		None.
+#
+# Output:
+#		Returns the full path and exe name on success or nil on failure.
+#
+###############################################################################
+	def FindLinuxXwd()
+		tmp = nil
+		found = nil
+
+		tmp = ENV['PATH'].split(/:/)
+		tmp.each do |p|
+			path = "#{p}/#{LINUX_XWD_BIN}"
+			if (File.exist?(path))
+				found = path
+				break
+			end
+		end
+
+		return found
+	end
+	private :FindLinuxXwd
+
+###############################################################################
+# FindLinuxConvert -- Method
+#		This method trys to find the linux convert util in you current path.
+#
+# Input:
+#		None.
+#
+# Output:
+#		Returns the full path and exe name on success or nil on failure.
+#
+###############################################################################
+	def FindLinuxConvert()
+		tmp = nil
+		found = nil
+
+		tmp = ENV['PATH'].split(/:/)
+		tmp.each do |p|
+			path = "#{p}/#{LINUX_CONVERT_BIN}"
+			if (File.exist?(path))
+				found = path
+				break
+			end
+		end
+
+		return found
+	end
+	private :FindLinuxConvert
+
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: soda
 version: !ruby/object:Gem::Version 
-  hash: 21
+  hash: 19
   prerelease: false
   segments: 
   - 0
   - 0
-  - 5
-  version: 0.0.5
+  - 6
+  version: 0.0.6
 platform: ruby
 authors: 
 - Trampus Richmond
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-11-09 00:00:00 -08:00
+date: 2010-11-10 00:00:00 -08:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -59,6 +59,7 @@ extra_rdoc_files: []
 files: 
 - lib/SodaReportSummery.rb
 - lib/SodaCSV.rb
+- lib/SodaScreenShot.rb
 - lib/soda.rb
 - lib/SodaXML.rb
 - lib/SodaUtils.rb