def test_run_check_update_task_with_bad_response(self):
with mock.patch('sentry.tasks.check_update.fetch_url_content') as fetch:
fetch.return_value = (None, None, '')
check_update() # latest_version == current_version
fetch.assert_called_once_with(PYPI_URL)
self.assertEqual(get_option(key=self.KEY), None)
def test_run_check_update_task(self, safe_urlread, safe_urlopen):
safe_urlread.return_value = json.dumps({'info': {'version': self.NEW}})
check_update() # latest_version > current_version
safe_urlopen.assert_called_once_with(PYPI_URL)
safe_urlread.assert_called_once_with(safe_urlopen.return_value)
self.assertEqual(options.get(key=self.KEY), self.NEW)
def test_run_check_update_task(self):
with mock.patch('sentry.tasks.check_update.fetch_url_content') as fetch:
fetch.return_value = (
None, None, json.dumps({'info': {'version': self.NEW}})
)
check_update() # latest_version > current_version
fetch.assert_called_once_with(PYPI_URL)
self.assertEqual(get_option(key=self.KEY), self.NEW)
def test_run_check_update_task_with_bad_response(self, safe_urlread,
safe_urlopen):
safe_urlread.return_value = ''
check_update() # latest_version == current_version
safe_urlopen.assert_called_once_with(PYPI_URL)
safe_urlread.assert_called_once_with(safe_urlopen.return_value)
self.assertEqual(options.get(key=self.KEY), '')
