aboutsummaryrefslogtreecommitdiffstats
path: root/actionpack/lib/action_dispatch/system_testing/test_helpers/screenshot_helper.rb
blob: 6de8fb74dc63c2f01d8468bd8ab586ae2c8a8e72 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
module ActionDispatch
  module SystemTesting
    module TestHelpers
      # Screenshot helper for system testing
      module ScreenshotHelper
        # Takes a screenshot of the current page in the browser.
        #
        # +take_screenshot+ can be used at any point in your system tests to take
        # a screenshot of the current state. This can be useful for debugging or
        # automating visual testing.
        def take_screenshot
          save_image
          puts "[Screenshot]: #{image_path}"
          puts display_image
        end

        # Takes a screenshot of the current page in the browser if the test
        # failed.
        #
        # +take_failed_screenshot+ is included in <tt>application_system_test_case.rb</tt>
        # that is generated with the application. To take screenshots when a test
        # fails add +take_failed_screenshot+ to the teardown block before clearing
        # sessions.
        def take_failed_screenshot
          take_screenshot if failed? && supports_screenshot?
        end

        private
          def image_name
            failed? ? "failures_#{method_name}" : method_name
          end

          def image_path
            "tmp/screenshots/#{image_name}.png"
          end

          def save_image
            page.save_screenshot(Rails.root.join(image_path))
          end

          def display_image
            if ENV["CAPYBARA_INLINE_SCREENSHOT"] == "artifact"
              "\e]1338;url=artifact://#{image_path}\a"
            else
              name = inline_base64(File.basename(image_path))
              image = inline_base64(File.read(image_path))
              "\e]1337;File=name=#{name};height=400px;inline=1:#{image}\a"
            end
          end

          def inline_base64(path)
            Base64.encode64(path).gsub("\n", "")
          end

          def failed?
            !passed? && !skipped?
          end

          def supports_screenshot?
            Capybara.current_driver != :rack_test
          end
      end
    end
  end
end