Skip to content

Commit 0f5cf9c

Browse files
committed
Updated tests and fixed a couple issues
- Updated unit test to look for a tuple - Fixed issue with code on return statement - Fixed issue with unit tests
1 parent 1a6a2ba commit 0f5cf9c

File tree

2 files changed

+13
-13
lines changed

2 files changed

+13
-13
lines changed

src/SeleniumLibrary/keywords/screenshot.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def _capture_page_screen_to_log(self, return_val):
130130
screenshot_as_base64 = self.driver.get_screenshot_as_base64()
131131
base64_str = self._embed_to_log_as_base64(screenshot_as_base64, 800)
132132
if return_val == BASE64:
133-
return_val base64_str
133+
return base64_str
134134
return EMBED
135135

136136
@keyword
@@ -194,12 +194,12 @@ def _screenshot_root_directory(self, value):
194194
def _decide_embedded(self, filename):
195195
filename = filename.upper()
196196
if (
197-
filename == DEFAULT_FILENAME_PAGE
197+
filename == DEFAULT_FILENAME_PAGE.upper()
198198
and self._screenshot_root_directory in EMBEDDED_OPTIONS
199199
):
200200
return True, self._screenshot_root_directory
201201
if (
202-
filename == DEFAULT_FILENAME_ELEMENT
202+
filename == DEFAULT_FILENAME_ELEMENT.upper()
203203
and self._screenshot_root_directory in EMBEDDED_OPTIONS
204204
):
205205
return True, self._screenshot_root_directory

utest/test/keywords/test_screen_shot.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -22,24 +22,24 @@ def teardown_function():
2222

2323

2424
def test_defaults(screen_shot):
25-
assert screen_shot._decide_embedded(SCREENSHOT_FILE_NAME) is False
26-
assert screen_shot._decide_embedded(ELEMENT_FILE_NAME) is False
25+
assert screen_shot._decide_embedded(SCREENSHOT_FILE_NAME) == (False, None)
26+
assert screen_shot._decide_embedded(ELEMENT_FILE_NAME) == (False, None)
2727

2828

2929
def test_screen_shotdir_embeded(screen_shot):
3030
screen_shot.ctx.screenshot_root_directory = EMBED
31-
assert screen_shot._decide_embedded(SCREENSHOT_FILE_NAME) is True
32-
assert screen_shot._decide_embedded(SCREENSHOT_FILE_NAME.upper()) is True
33-
assert screen_shot._decide_embedded(ELEMENT_FILE_NAME) is True
34-
assert screen_shot._decide_embedded(ELEMENT_FILE_NAME.upper()) is True
35-
assert screen_shot._decide_embedded("other.psn") is False
31+
assert screen_shot._decide_embedded(SCREENSHOT_FILE_NAME) == (True, EMBED)
32+
assert screen_shot._decide_embedded(SCREENSHOT_FILE_NAME.upper()) == (True, EMBED)
33+
assert screen_shot._decide_embedded(ELEMENT_FILE_NAME) == (True, EMBED)
34+
assert screen_shot._decide_embedded(ELEMENT_FILE_NAME.upper()) == (True, EMBED)
35+
assert screen_shot._decide_embedded("other.psn") == (False, None)
3636

3737

3838
def test_file_name_embeded(screen_shot):
39-
assert screen_shot._decide_embedded(EMBED) is True
40-
assert screen_shot._decide_embedded("other.psn") is False
39+
assert screen_shot._decide_embedded(EMBED) == (True, EMBED)
40+
assert screen_shot._decide_embedded("other.psn") == (False, None)
4141
screen_shot.ctx.screenshot_root_directory = EMBED
42-
assert screen_shot._decide_embedded(EMBED) is True
42+
assert screen_shot._decide_embedded(EMBED) == (True, EMBED)
4343

4444

4545
def test_screenshot_path_embedded(screen_shot):

0 commit comments

Comments
 (0)