Exemplo n.º 1
0
 def testParseTaskRelative(self):
     expected_self_link = (
         '{}/projects/my-project/locations/us-central1/queues/my-queue'
         '/tasks/my-task'.format(_SELF_LINK_PREFIX))
     actual_task_ref = parsers.ParseTask(
         'projects/my-project/locations/us-central1/queues/my-queue'
         '/tasks/my-task')
     self.assertEqual(actual_task_ref.SelfLink(), expected_self_link)
Exemplo n.º 2
0
 def Run(self, args):
     tasks_client = GetApiAdapter(self.ReleaseTrack()).tasks
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.task, queue_ref)
     return tasks_client.Get(
         task_ref,
         response_view=self.TASK_RESPONSE_VIEW_MAPPER.GetEnumForChoice(
             args.response_view))
Exemplo n.º 3
0
 def Run(self, args):
   tasks_client = tasks.Tasks()
   queue_ref = parsers.ParseQueue(args.queue)
   task_ref = parsers.ParseTask(args.id, queue_ref) if args.id else None
   task_config = parsers.ParseCreateTaskArgs(
       args, constants.APP_ENGINE_QUEUE, tasks_client.api.messages)
   create_response = tasks_client.Create(
       queue_ref, task_ref,
       schedule_time=task_config.scheduleTime,
       app_engine_http_request=task_config.appEngineHttpRequest)
   log.CreatedResource(create_response.name, 'task')
   return create_response
 def Run(self, args):
     tasks_client = tasks.Tasks()
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.id, queue_ref) if args.id else None
     task_config = parsers.ParseCreateTaskArgs(args, constants.PULL_QUEUE,
                                               tasks_client.api.messages)
     create_response = tasks_client.Create(
         queue_ref,
         task_ref,
         schedule_time=task_config.scheduleTime,
         pull_message=task_config.pullMessage)
     log.CreatedResource(create_response.name, 'task')
     return create_response
Exemplo n.º 5
0
 def Run(self, args):
   api = GetApiAdapter(self.ReleaseTrack())
   tasks_client = api.tasks
   queue_ref = parsers.ParseQueue(args.queue, args.location)
   task_ref = parsers.ParseTask(args.task,
                                queue_ref) if args.task else None
   task_config = parsers.ParseCreateTaskArgs(
       args, constants.PULL_QUEUE, api.messages, is_alpha=True)
   create_response = tasks_client.Create(
       queue_ref, task_ref,
       schedule_time=task_config.scheduleTime,
       pull_message=task_config.pullMessage)
   log.CreatedResource(create_response.name, 'task')
   return create_response
Exemplo n.º 6
0
 def Run(self, args):
     api = GetApiAdapter(self.ReleaseTrack())
     tasks_client = api.tasks
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.task,
                                  queue_ref) if args.task else None
     task_config = parsers.ParseCreateTaskArgs(args, constants.PUSH_QUEUE,
                                               api.messages)
     create_response = tasks_client.Create(
         queue_ref,
         task_ref,
         schedule_time=task_config.scheduleTime,
         app_engine_http_request=task_config.appEngineHttpRequest)
     log.CreatedResource(create_response.name, 'task')
     return create_response
Exemplo n.º 7
0
  def Run(self, args):
    if self.ReleaseTrack() == base.ReleaseTrack.ALPHA:
      # This functionality doesn't exist in the alpha API so use beta.
      api_release_track = base.ReleaseTrack.BETA
    else:
      api_release_track = self.ReleaseTrack()

    api = GetApiAdapter(api_release_track)
    tasks_client = api.tasks
    queue_ref = parsers.ParseQueue(args.queue, args.location)
    task_ref = parsers.ParseTask(args.task,
                                 queue_ref) if args.task else None
    task_config = parsers.ParseCreateTaskArgs(
        args, constants.HTTP_TASK, api.messages,
        release_track=api_release_track)
    create_response = tasks_client.Create(
        queue_ref, task_ref,
        schedule_time=task_config.scheduleTime,
        http_request=task_config.httpRequest)
    log.CreatedResource(create_response.name, 'task')
    return create_response
Exemplo n.º 8
0
 def Run(self, args):
   tasks_client = tasks.Tasks()
   queue_ref = parsers.ParseQueue(args.queue, args.location)
   task_ref = parsers.ParseTask(args.task, queue_ref)
   return tasks_client.Run(task_ref)
 def Run(self, args):
   tasks_client = tasks.Tasks()
   queue_ref = parsers.ParseQueue(args.queue, args.location)
   task_ref = parsers.ParseTask(args.task, queue_ref)
   duration = parsers.FormatLeaseDuration(args.lease_duration)
   return tasks_client.RenewLease(task_ref, args.schedule_time, duration)
Exemplo n.º 10
0
 def Run(self, args):
     tasks_client = GetApiAdapter(self.ReleaseTrack()).tasks
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.task, queue_ref)
     return tasks_client.CancelLease(task_ref, args.schedule_time)
 def Run(self, args):
     tasks_client = tasks.Tasks()
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.task, queue_ref)
     tasks_client.Delete(task_ref)
     log.DeletedResource(task_ref.Name(), kind='task')
Exemplo n.º 12
0
 def Run(self, args):
   tasks_client = GetApiAdapter(self.ReleaseTrack()).tasks
   queue_ref = parsers.ParseQueue(args.queue, args.location)
   task_ref = parsers.ParseTask(args.task, queue_ref)
   duration = parsers.FormatLeaseDuration(args.lease_duration)
   return tasks_client.RenewLease(task_ref, args.schedule_time, duration)
Exemplo n.º 13
0
 def Run(self, args):
     tasks_client = tasks.Tasks()
     task_ref = parsers.ParseTask(args.task, args.queue)
     return tasks_client.Run(task_ref)
Exemplo n.º 14
0
 def testParseTaskId_NoQueueRef(self):
     with self.assertRaises(parsers.FullTaskUnspecifiedError):
         parsers.ParseTask('my-task')
Exemplo n.º 15
0
 def Run(self, args):
   tasks_client = tasks.Tasks()
   queue_ref = parsers.ParseQueue(args.queue)
   task_ref = parsers.ParseTask(args.task, queue_ref)
   return tasks_client.CancelLease(task_ref, args.schedule_time)
 def Run(self, args):
     tasks_client = tasks.Tasks()
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.task, queue_ref)
     return tasks_client.Acknowledge(task_ref, args.schedule_time)
Exemplo n.º 17
0
 def Run(self, args):
     tasks_client = GetApiAdapter(self.ReleaseTrack()).tasks
     queue_ref = parsers.ParseQueue(args.queue, args.location)
     task_ref = parsers.ParseTask(args.task, queue_ref)
     tasks_client.Delete(task_ref)
     log.DeletedResource(task_ref.Name(), kind='task')