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