import unittest

from selenium import webdriver

class SearchProducts(unittest.TestCase):

def setUp(self):

desired_caps = {}

desired_caps['platform'] = 'WINDOWS '

desired_caps['browserName'] = 'firefox'

self.driver = \

webdriver.Remote('http://192.168.1.102:4444/wd/hub', desired_caps)

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

self.driver.implicitly_wait(30)

self.driver.maximize_window()

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 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()

if __name__ == '__main__':

unittest.main(verbosity=2)