finish firefox home testcase

Author: sarinali

modules/data/about_newtab.components.json

@@ -73,5 +73,11 @@
         "selectorData": "//div[following-sibling::div[@class='section-top-bar']]",
         "strategy": "xpath",
         "groups": []
+    },
+
+    "body-logo": {
+        "selectorData": "logo",
+        "strategy": "class",
+        "groups": []
     }
 }

modules/data/about_prefs.components.json

@@ -133,5 +133,17 @@
         "selectorData": "#addresses option",
         "strategy": "css",
         "groups": []
+    },
+
+    "home-new-tabs-dropdown": {
+        "selectorData": "newTabMode",
+        "strategy": "id",
+        "groups": []
+    },
+
+    "home-new-tabs-dropdown-option-default": {
+        "selectorData": "menuitem[data-l10n-id=\"home-mode-choice-default-fx\"]",
+        "strategy": "css",
+        "groups": []
     }
 }

tests/preferences/conftest.py

@@ -0,0 +1,12 @@
+import pytest
+
+
+@pytest.fixture()
+def suite_id():
+    return ("S2241", "Preferences")
+
+
+@pytest.fixture()
+def set_prefs():
+    """Set prefs"""
+    return []

tests/preferences/test_firefox_home_new_tabs.py

@@ -0,0 +1,27 @@
+from time import sleep
+
+from selenium.webdriver import Firefox
+from modules.page_object import AboutPrefs, AboutNewtab
+from modules.browser_object import TabBar
+
+def test_firefox_home_new_tab(driver: Firefox):
+    """
+    C161472: setting the default new window to be Firefox Home
+    """
+    about_prefs = AboutPrefs(driver, category="home").open()
+    tabs = TabBar(driver)
+    about_new_tab = AboutNewtab(driver)
+
+    drop_down = about_prefs.get_element("home-new-tabs-dropdown")
+    drop_down.click()
+
+    home_option = about_prefs.get_element("home-new-tabs-dropdown-option-default", parent_element=drop_down)
+    home_option.click()
+
+    assert drop_down.get_attribute("label") == "Firefox Home (Default)"
+    tabs.new_tab_by_button()
+
+    tabs.wait_for_num_tabs(2)
+    driver.switch_to.window(driver.window_handles[-1])
+
+    assert about_new_tab.get_element("body-logo") is not None