from selenium.webdriver import Firefox
from selenium.webdriver.support.wait import WebDriverWait
from selectmenu import Selectmenu

class TestSelectmenu(object):
    def setup_method(self, method):
        self.driver = Firefox()
        self.driver.get("http://jquery-ui.googlecode.com/svn/branches/labs/selectmenu/index.html")

    def teardown_method(self, method):
        self.driver.quit()

    def test_options(self):
        select = self.driver.find_element_by_id("speedA")
        selectmenu = Selectmenu(select)
        assert(len(selectmenu.options) == 5)

    def test_select_by_visible_text(self):
        select = self.driver.find_element_by_id("speedA")
        selectmenu = Selectmenu(select)
        selectmenu.select_by_visible_text('Faster')
        assert(selectmenu.first_selected_option == 'Faster')
    
    def test_first_selected_option(self):
        select = self.driver.find_element_by_id("speedA")
        selectmenu = Selectmenu(select)
        assert(selectmenu.first_selected_option == 'Medium')