def test_do_calls_callable_correctly(self):
agent = Agent(self.driver)
agent.foo = 0
def my_func(page):
agent.foo = 1
CSSTestPage.wait = 5
agent.to_at(CSSTestPage)
agent.then(my_func)
self.assertEquals(agent.foo, 1)
class TestCaseHorace(TestCase):
def setUp(self):
agentDriver = Driver({"driver": getenv("BROWSER", "phantomjs"), "platform": getenv("PLATFORM", "ANY")})
self.agent = Agent(agentDriver)
def tearDown(self):
self.agent.close()
def to(self, page, path=None, parameters=None):
self.agent.to(page, path, parameters)
def at(self, page):
self.agent.at(page)
def to_at(self, page, path=None, parameters=None):
self.agent.to_at(page, path, parameters)
def __getattr__(self, item):
if item == "page":
return self.agent.page
elif item == "_driver":
return self.agent._driver
else:
return object.__getattribute__(self, item)
def test_do_does_not_call_an_uncallable(self):
agent = Agent(self.driver)
agent.foo = 0
agent.to_at(CSSTestPage)
agent.then(1)
self.assertEquals(agent.foo, 0)
def test_currentPage(self):
agent = Agent(self.driver)
agent.to_at(DuckDuckGoPage)
self.assertIsInstance(agent.page, DuckDuckGoPage)
def test_toAt(self):
agent = Agent(self.driver)
agent.to_at(DuckDuckGoPage)
self.assertTrue(DuckDuckGoPage.title, self.driver.title)
