From d9eca45945a3d82927a498969410af7af434f375 Mon Sep 17 00:00:00 2001 From: Trent Palmer Date: Sat, 21 Aug 2021 16:05:41 -0700 Subject: [PATCH] add unittest test_scroll.py --- test/test_scroll.py | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 test/test_scroll.py diff --git a/test/test_scroll.py b/test/test_scroll.py new file mode 100644 index 0000000..baede42 --- /dev/null +++ b/test/test_scroll.py @@ -0,0 +1,43 @@ +from test.test_selenium import TestSelenium +from unittest import TestCase + + +class TestScroll(TestSelenium, TestCase): + def setUp(self): + super(TestScroll, self).setUp() + self.plus_button = self.driver.find_element_by_id('plus-button') + self.minus_button = self.driver.find_element_by_id('minus-button') + + def test_get_scroll_position(self): + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 0) + self.driver.execute_script( + "window.scrollTo(0,document.body.scrollHeight)" + ) + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 12461) + + def test_plus_button_scroll_to_top(self): + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 0) + self.driver.execute_script( + "window.scrollTo(0,document.body.scrollHeight)" + ) + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 12461) + self.plus_button.click() + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 0) + + def test_minus_button_scroll_to_top(self): + self.plus_button.click() + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 0) + self.driver.execute_script( + "window.scrollTo(0,document.body.scrollHeight)" + ) + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 12573) + self.minus_button.click() + self.assertEqual( + self.driver.execute_script("return window.pageYOffset"), 0)