def setUp(self):

# create a new Firefox session

self.driver = webdriver.Firefox()

self.driver.implicitly_wait(30)

self.driver.maximize_window()

# navigate to the application home page

self.driver.get('http://demo-store.seleniumacademy.com/')

def test_search_by_category(self):

# get the search textbox

search_field = self.driver.find_element_by_name('q')

self.highlightElement(search_field)

search_field.clear()

# enter search keyword and submit

self.highlightElement(search_field)

search_field.send_keys('phones')

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("//h2[@class='product-name']/a")

# check count of products shown in results

self.assertEqual(3, len(products))

def tearDown(self):

# close the browser window

self.driver.quit()

def highlightElement(self, element):

self.driver.execute_script("arguments[0].setAttribute('style', arguments[1]);",

element, "color: green; border: 2px solid green;")

self.driver.execute_script("arguments[0].setAttribute('style', arguments[1]);",