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)