def test_get_or_save(self):
self.sources[0].save()
comment = self.comments[0]
self.assertEqual(0, Comment.all().count())
self.assertEqual(0, len(self.taskqueue_stub.GetTasks('propagate')))
# new. should add a propagate task.
saved = comment.get_or_save()
self.assertEqual(1, Comment.all().count())
self.assertTrue(saved.is_saved())
self.assertEqual(comment.key(), saved.key())
self.assertEqual(comment.source, saved.source)
self.assertEqual(comment.dest, saved.dest)
tasks = self.taskqueue_stub.GetTasks('propagate')
self.assertEqual(1, len(tasks))
self.assertEqual(str(comment.key()),
testutil.get_task_params(tasks[0])['comment_key'])
self.assertEqual('/_ah/queue/propagate', tasks[0]['url'])
# existing. no new task.
same = saved.get_or_save()
self.assertEqual(saved.source.key(), same.source.key())
self.assertEqual(saved.dest.key(), same.dest.key())
self.assertEqual(1, len(tasks))
def _test_create_new(self):
FakeSource.create_new(self.handler)
self.assertEqual(1, FakeSource.all().count())
tasks = self.taskqueue_stub.GetTasks('poll')
self.assertEqual(1, len(tasks))
source = FakeSource.all().get()
self.assertEqual('/_ah/queue/poll', tasks[0]['url'])
params = testutil.get_task_params(tasks[0])
self.assertEqual(str(source.key()), params['source_key'])
self.assertEqual('1970-01-01-00-00-00',
params['last_polled'])
def test_get_or_save(self):
self.assertEqual(0, salmon.Salmon.all().count())
self.assertEqual(0, len(self.taskqueue_stub.GetTasks('propagate')))
# new. should add a propagate task.
saved = self.salmon.get_or_save()
self.assertTrue(saved.is_saved())
self.assertEqual(self.salmon.key(), saved.key())
tasks = self.taskqueue_stub.GetTasks('propagate')
self.assertEqual(1, len(tasks))
self.assertEqual(str(self.salmon.key()),
testutil.get_task_params(tasks[0])['salmon_key'])
self.assertEqual('/_ah/queue/propagate', tasks[0]['url'])
# existing. no new task.
same = saved.get_or_save()
self.assertEqual(1, len(tasks))
def test_get_or_save(self):
self.assertEqual(0, Migratable.all().count())
self.assertEqual(0, len(self.taskqueue_stub.GetTasks('propagate')))
# new. should add a propagate task.
saved = self.post.get_or_save()
self.assertTrue(saved.is_saved())
self.assertEqual(self.post.key(), saved.key())
tasks = self.taskqueue_stub.GetTasks('propagate')
self.assertEqual(1, len(tasks))
self.assertEqual(str(self.post.key()),
testutil.get_task_params(tasks[0])['post_key'])
self.assertEqual('/_ah/queue/propagate', tasks[0]['url'])
# existing. no new task.
same = saved.get_or_save()
self.assertEqual(1, len(tasks))
def test_poll(self):
"""A normal poll task."""
self.assertFalse(db.get(self.salmon.key()))
self.assertEqual([], self.taskqueue_stub.GetTasks('poll'))
self.post_task()
self.assertTrue(db.get(self.salmon.key()))
source = db.get(self.source.key())
self.assertEqual(self.now, source.last_polled)
tasks = self.taskqueue_stub.GetTasks('poll')
self.assertEqual(1, len(tasks))
self.assertEqual('/_ah/queue/poll', tasks[0]['url'])
params = testutil.get_task_params(tasks[0])
self.assertEqual(str(source.key()),
params['source_key'])
self.assertEqual(self.now.strftime(Source.POLL_TASK_DATETIME_FORMAT),
params['last_polled'])
def test_scan(self):
"""A normal scan task."""
self.assertFalse(db.get(self.salmon.key()))
self.assertEqual([], self.taskqueue_stub.GetTasks('scan'))
self.post_task()
self.assertTrue(db.get(self.salmon.key()))
source = db.get(self.source.key())
self.assertEqual(self.now, source.last_scanned)
tasks = self.taskqueue_stub.GetTasks('scan')
self.assertEqual(1, len(tasks))
self.assertEqual('/_ah/queue/scan', tasks[0]['url'])
params = testutil.get_task_params(tasks[0])
self.assertEqual(str(source.key()),
params['source_key'])
self.assertEqual(self.now.strftime(Source.SCAN_TASK_DATETIME_FORMAT),
params['last_scanned'])
