unittest

import unittest
from selenium import webdriver

class SearchTest(unittest.TestCase):
    @classmethod
    def setUp(cls):
        # create a  new Firefox session
        cls.driver = webdriver.Firefox()
        cls.driver.implicitly_wait(30)
        cls.driver.maximize_window()

        # navigate to the application home page
        cls.driver.get("https://start.firefoxchina.cn/")

    def test_search_by_category(self):
        # get the search textbox
        self.search_field = self.driver.find_element_by_name("q")
        self.search_field.clear()

        # enter search keyword and submit
        self.search_field.send_keys("phones")
        self.search_field.submit()

        # get all the anchor element which have product names
        # displayed currently on result page using
        # find_elements_by_xpath method
        products = self.driver.find_elements_by_xpath(".//*[@id='2']/h3/a")
        self.assertEqual(1,len(products))

    def test_search_by_name(self):
        # get the search textbox
        self.search_field = self.driver.find_element_by_name("q")
        self.search_field.clear()

        # enter search keyword and submit
        self.search_field.send_keys("sait shaker")
        self.search_field.submit()

        # get all the anchor elements which have
        # product names displayed
        # currently on result page using
        # find_elements_by_xpath method
        products = self.driver.find_elements_by_xpath(".//*[@id='6']/h3/a")
        self.assertEqual(1, len(products))

    @classmethod
    def tearDown(cls):
        # close the browser window
        cls.driver.quit()