def test_get_patch_invalid_patch(self): bz = Bugzilla(api_url="http://fake", attachment_url="fake") with mock.patch("requests.get", autospec=True) as get: r = get.return_value r.content = "The attachment id 1111 is invalid" with self.assertRaises(autoland.errors.InvalidAttachment): bz.download_patch(patch_id=1111, path=self.tmpdir)
def test_get_patch_http_failure(self): bz = Bugzilla(api_url="http://fake", attachment_url="fake") with mock.patch("requests.get", autospec=True) as get: r = get.return_value r.raise_for_status.side_effect = requests.HTTPError() with self.assertRaises(requests.RequestException): bz.download_patch(patch_id=1111, path=self.tmpdir)
def test_get_patch_overwrite_refetches(self): bz = Bugzilla(api_url="http://fake", attachment_url="fake") with mock.patch("requests.get", autospec=True) as get: r = get.return_value content = "Super patch" r.content = content bz.download_patch(patch_id=1111, path=self.tmpdir) r.raise_for_status.side_effect = requests.HTTPError() with self.assertRaises(requests.RequestException): bz.download_patch(patch_id=1111, path=self.tmpdir, overwrite_patch=True)
def test_get_patch(self): bz = Bugzilla(api_url="http://fake", attachment_url="fake") with mock.patch("requests.get", autospec=True) as get: r = get.return_value content = "Super patch" r.content = content patch_file = bz.download_patch(patch_id=1111, path=self.tmpdir) self.assertEquals(content, open(patch_file).read())