Skip to content

Commit dbd2ad1

Browse files
committed
Add Demo Mode notifications for basic assert statements
1 parent 2e2a429 commit dbd2ad1

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

seleniumbase/fixtures/base_case.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1794,19 +1794,39 @@ def assert_downloaded_file(self, file):
17941794

17951795
def assert_true(self, expr, msg=None):
17961796
self.assertTrue(expr, msg=msg)
1797+
if self.demo_mode:
1798+
messenger_post = ("ASSERT TRUE: {%s}" % expr)
1799+
js_utils.post_messenger_success_message(
1800+
self.driver, messenger_post, self.message_duration)
17971801

17981802
def assert_false(self, expr, msg=None):
17991803
self.assertFalse(expr, msg=msg)
1804+
if self.demo_mode:
1805+
messenger_post = ("ASSERT FALSE: {%s}" % expr)
1806+
js_utils.post_messenger_success_message(
1807+
self.driver, messenger_post, self.message_duration)
18001808

18011809
def assert_equal(self, first, second, msg=None):
18021810
self.assertEqual(first, second, msg=msg)
1811+
if self.demo_mode:
1812+
messenger_post = ("ASSERT EQUAL: {%s == %s}" % (first, second))
1813+
js_utils.post_messenger_success_message(
1814+
self.driver, messenger_post, self.message_duration)
18031815

18041816
def assert_not_equal(self, first, second, msg=None):
18051817
self.assertNotEqual(first, second, msg=msg)
1818+
if self.demo_mode:
1819+
messenger_post = ("ASSERT NOT EQUAL: {%s != %s}" % (first, second))
1820+
js_utils.post_messenger_success_message(
1821+
self.driver, messenger_post, self.message_duration)
18061822

18071823
def assert_title(self, title):
18081824
""" Asserts that the web page title matches the expected title. """
18091825
assert self.get_title() == title
1826+
if self.demo_mode:
1827+
messenger_post = ("ASSERT TITLE: {%s}" % title)
1828+
js_utils.post_messenger_success_message(
1829+
self.driver, messenger_post, self.message_duration)
18101830

18111831
def assert_no_js_errors(self):
18121832
""" Asserts that there are no JavaScript "SEVERE"-level page errors.

0 commit comments

Comments
 (0)