def test_ensure_when_present(self):
Author(id='author').put()
response = self.app.get(util.ensure_author_task('author'), headers={'X-AppEngine-QueueName': 'default'})
self.assertEqual(response.status_int, 200)
tasks = self.tasks.get_filtered_tasks()
self.assertEqual([], [task.url for task in tasks])
def test_delete_missing_author(self):
author = Author(id='test')
author.put()
self.respond_to_github('https://api.github.com/users/test', {'status': 404})
response = self.app.get(util.update_author_task('TEST'), headers={'X-AppEngine-QueueName': 'default'})
self.assertEqual(response.status_int, 200)
author = Author.get_by_id('test')
self.assertIsNone(author)
def test_update_suppressed_is_noop(self):
author = Author(id='test', status=Status.suppressed)
author.put()
response = self.app.get('/task/update/test', headers={'X-AppEngine-QueueName': 'default'})
self.assertEqual(response.status_int, 200)
tasks = self.tasks.get_filtered_tasks()
self.assertEqual(len(tasks), 0)
author = author.key.get()
self.assertEqual(author.status, Status.suppressed)
def test_update_all(self):
library_key = Library(id='owner/repo').put()
author_key = Author(id='owner').put()
response = self.app.get('/manage/update-all', headers={'X-AppEngine-QueueName': 'default'})
self.assertEqual(response.status_int, 200)
tasks = self.tasks.get_filtered_tasks()
self.assertEqual([
util.update_library_task(library_key.id()),
util.update_author_task(author_key.id()),
], [task.url for task in tasks])