@@ -80,6 +80,19 @@ def setup
80
80
assert_equal "simple" , @new_test . send ( :output_type )
81
81
end
82
82
83
+ test "take_screenshot saves image and shows link to it" do
84
+ display_image_actual = nil
85
+
86
+ Rails . stub :root , Pathname . getwd do
87
+ @new_test . stub :save_image , nil do
88
+ @new_test . stub :show , -> ( img ) { display_image_actual = img } do
89
+ @new_test . take_screenshot
90
+ end
91
+ end
92
+ end
93
+ assert_match %r|\[ Screenshot Image\] .+?tmp/screenshots/1_x\. png | , display_image_actual
94
+ end
95
+
83
96
test "take_screenshot saves HTML and shows link to it when using RAILS_SYSTEM_TESTING_SCREENSHOT_HTML env" do
84
97
original_html_setting = ENV [ "RAILS_SYSTEM_TESTING_SCREENSHOT_HTML" ]
85
98
ENV [ "RAILS_SYSTEM_TESTING_SCREENSHOT_HTML" ] = "1"
@@ -97,7 +110,7 @@ def setup
97
110
end
98
111
end
99
112
assert called_save_html
100
- assert_match %r|\[ Screenshot HTML\] .+?tmp/screenshots/1_x\. html| , display_image_actual
113
+ assert_match %r|\[ Screenshot HTML\] .+?tmp/screenshots/1_x\. html | , display_image_actual
101
114
ensure
102
115
ENV [ "RAILS_SYSTEM_TESTING_SCREENSHOT_HTML" ] = original_html_setting
103
116
end
@@ -116,7 +129,7 @@ def setup
116
129
end
117
130
end
118
131
assert called_save_html
119
- assert_match %r|\[ Screenshot HTML\] .+?tmp/screenshots/1_x\. html| , display_image_actual
132
+ assert_match %r|\[ Screenshot HTML\] .+?tmp/screenshots/1_x\. html | , display_image_actual
120
133
end
121
134
122
135
test "take_screenshot allows changing screenshot display format via RAILS_SYSTEM_TESTING_SCREENSHOT env" do
0 commit comments