def test_passthru_2(self):
self.auth_func.side_effect = MockPopper(
AuthException("boom"), download.addinfourl(StringIO.StringIO("SUCCESS"), {}, "", 200)
)
self.credentials.search.return_value.append(("andy", "penguin55", True))
self.assertEquals(self.func("http://www.isotoma.com/").read(), "SUCCESS")
self.assertEquals(self.auth_func.call_count, 2)
def setUp(self):
self.credentials = mock.Mock()
self.credentials.search.return_value = [(None, None, False)]
self.auth_func = mock.Mock()
self.auth_func.return_value = download.addinfourl(StringIO.StringIO("SUCCESS"), {}, "", 200)
self.func = download.inject_credentials(self.credentials)(self.auth_func)
