Python TaskQueueConfig.load_queues_from_db Exemples

Exemple #1

    def test_create_celery_worker_scripts(self):
        flexmock(file_io).should_receive("read").and_return(sample_queue_yaml2)
        flexmock(file_io).should_receive("write").and_return(None)
        flexmock(file_io).should_receive("mkdir").and_return(None)

        flexmock(datastore).should_receive("Get").\
             and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
        tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 'myapp')
        flexmock(file_io) \
           .should_receive("exists").and_return(True)
        queue_info = tqc.load_queues_from_file('app_id')
        queue_info = tqc.load_queues_from_db()
        FILE1 = open(
            os.path.dirname(os.path.realpath(__file__)) +
            '/../../templates/header.py', 'r')
        file1 = FILE1.read()
        FILE1.close()
        FILE2 = open(
            os.path.dirname(os.path.realpath(__file__)) +
            '/../../templates/task.py', 'r')
        file2 = FILE2.read()
        FILE2.close()

        flexmock(file_io).should_receive('write').and_return(None)
        flexmock(file_io).should_receive("read").and_return(file1).and_return(
            file2)
        self.assertEquals(
            tqc.create_celery_worker_scripts(TaskQueueConfig.QUEUE_INFO_DB),
            TaskQueueConfig.CELERY_WORKER_DIR + 'app___myapp.py')
        self.assertEquals(
            tqc.create_celery_worker_scripts(TaskQueueConfig.QUEUE_INFO_FILE),
            TaskQueueConfig.CELERY_WORKER_DIR + 'app___myapp.py')

Exemple #2

 def test_load_queues(self):
     flexmock(file_io) \
        .should_receive("read").and_return(sample_queue_yaml2)
     flexmock(file_io) \
        .should_receive("exists").and_return(True)
     flexmock(file_io) \
        .should_receive("write").and_return(None)
     flexmock(file_io) \
        .should_receive("mkdir").and_return(None)
     flexmock(datastore).should_receive("Get").\
          and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
     tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 'myapp')
     queue_info = tqc.load_queues_from_file('app_id')
     queue_info = tqc.load_queues_from_db()

Exemple #3

Fichier : test_tq_config.py Projet : GavinHwa/appscale

 def test_load_queues(self):
   flexmock(file_io) \
      .should_receive("read").and_return(sample_queue_yaml2)
   flexmock(file_io) \
      .should_receive("exists").and_return(True)
   flexmock(file_io) \
      .should_receive("write").and_return(None)
   flexmock(file_io) \
      .should_receive("mkdir").and_return(None)
   flexmock(datastore).should_receive("Get").\
        and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
   tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 
                         'myapp')
   queue_info = tqc.load_queues_from_file('app_id')
   queue_info = tqc.load_queues_from_db()

Exemple #4

Fichier : test_tq_config.py Projet : GavinHwa/appscale

 def test_load_queues_from_db(self):
   flexmock(file_io) \
      .should_receive("read").and_return(sample_queue_yaml2)
   flexmock(file_io) \
      .should_receive("write").and_return(None)
   flexmock(file_io) \
      .should_receive("mkdir").and_return(None)
   flexmock(datastore).should_receive("Get").\
        and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
   tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 
                         'myapp')
   queue_info = tqc.load_queues_from_db()
   self.assertEquals(queue_info, {'queue':[{'name': 'foo',
                                            'rate': '10/m'},
                                           ]})

Exemple #5

 def test_load_queues_from_db(self):
     flexmock(file_io) \
        .should_receive("read").and_return(sample_queue_yaml2)
     flexmock(file_io) \
        .should_receive("write").and_return(None)
     flexmock(file_io) \
        .should_receive("mkdir").and_return(None)
     flexmock(datastore).should_receive("Get").\
          and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
     tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 'myapp')
     queue_info = tqc.load_queues_from_db()
     self.assertEquals(queue_info,
                       {'queue': [
                           {
                               'name': 'foo',
                               'rate': '10/m'
                           },
                       ]})

Exemple #6

Fichier : test_tq_config.py Projet : GavinHwa/appscale

  def test_create_celery_file(self):
    flexmock(file_io) \
       .should_receive("read").and_return(sample_queue_yaml2)
    flexmock(file_io) \
       .should_receive("exists").and_return(True)
    flexmock(file_io) \
       .should_receive("write").and_return(None)
    flexmock(file_io) \
       .should_receive("mkdir").and_return(None)
    flexmock(datastore).should_receive("Get").\
         and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
    tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 
                          'myapp')
    flexmock(file_io).should_receive("read").and_return(sample_queue_yaml2)
    queue_info = tqc.load_queues_from_file('app_id')
    queue_info = tqc.load_queues_from_db()

    # making sure it does not throw an exception
    self.assertEquals(tqc.create_celery_file(TaskQueueConfig.QUEUE_INFO_DB),
                      TaskQueueConfig.CELERY_CONFIG_DIR + "myapp" + ".py")
    self.assertEquals(tqc.create_celery_file(TaskQueueConfig.QUEUE_INFO_FILE),
                      TaskQueueConfig.CELERY_CONFIG_DIR + "myapp" + ".py")

Exemple #7

    def test_create_celery_file(self):
        flexmock(file_io) \
           .should_receive("read").and_return(sample_queue_yaml2)
        flexmock(file_io) \
           .should_receive("exists").and_return(True)
        flexmock(file_io) \
           .should_receive("write").and_return(None)
        flexmock(file_io) \
           .should_receive("mkdir").and_return(None)
        flexmock(datastore).should_receive("Get").\
             and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
        tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 'myapp')
        flexmock(file_io).should_receive("read").and_return(sample_queue_yaml2)
        queue_info = tqc.load_queues_from_file('app_id')
        queue_info = tqc.load_queues_from_db()

        # making sure it does not throw an exception
        self.assertEquals(
            tqc.create_celery_file(TaskQueueConfig.QUEUE_INFO_DB),
            TaskQueueConfig.CELERY_CONFIG_DIR + "myapp" + ".py")
        self.assertEquals(
            tqc.create_celery_file(TaskQueueConfig.QUEUE_INFO_FILE),
            TaskQueueConfig.CELERY_CONFIG_DIR + "myapp" + ".py")

Exemple #8

Fichier : test_tq_config.py Projet : GavinHwa/appscale

  def test_create_celery_worker_scripts(self):
    flexmock(file_io).should_receive("read").and_return(sample_queue_yaml2)
    flexmock(file_io).should_receive("write").and_return(None)
    flexmock(file_io).should_receive("mkdir").and_return(None)

    flexmock(datastore).should_receive("Get").\
         and_return({TaskQueueConfig.QUEUE_INFO: '{"queue":[{"name": "foo", "rate": "10/m"}]}'})
    tqc = TaskQueueConfig(TaskQueueConfig.RABBITMQ, 
                          'myapp')
    flexmock(file_io) \
       .should_receive("exists").and_return(True)
    queue_info = tqc.load_queues_from_file('app_id')
    queue_info = tqc.load_queues_from_db()
    FILE1 = open(os.path.dirname(os.path.realpath(__file__)) + '/../../templates/header.py', 'r')
    file1 = FILE1.read()
    FILE1.close()
    FILE2 = open(os.path.dirname(os.path.realpath(__file__)) + '/../../templates/task.py', 'r')
    file2 = FILE2.read()
    FILE2.close()

    flexmock(file_io).should_receive('write').and_return(None)
    flexmock(file_io).should_receive("read").and_return(file1).and_return(file2)
    self.assertEquals(tqc.create_celery_worker_scripts(TaskQueueConfig.QUEUE_INFO_DB), TaskQueueConfig.CELERY_WORKER_DIR + 'app___myapp.py')
    self.assertEquals(tqc.create_celery_worker_scripts(TaskQueueConfig.QUEUE_INFO_FILE), TaskQueueConfig.CELERY_WORKER_DIR + 'app___myapp.py')