Contents

require_relative 'image_wrangler'

module Gatling
  class CaptureElement

    def initialize element_to_capture, *element_to_exclude
      @reference_image_path = Gatling::Configuration.reference_image_path
      @element_to_capture = element_to_capture
      @element_to_exclude = element_to_exclude.first

    end

    def capture

      screenshot = self.take_screenshot
      screenshot = exclude(screenshot, @element_to_exclude) if @element_to_exclude
      Gatling::ImageWrangler.crop_element(screenshot, @element_to_capture)
    end

    def take_screenshot
      temp_dir = "#{@reference_image_path}/temp"
      #captures the uncropped full screen
      begin
        Capybara.page.driver.browser.save_screenshot("#{temp_dir}/temp.png")
        temp_screenshot = Magick::Image.read("#{temp_dir}/temp.png").first
      rescue
        raise "Could not save screenshot to #{temp_dir}. Please make sure you have permission"
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
gatling-1.0.7	lib/gatling/capture_element.rb