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'])