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')
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()
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'}, ]})
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' }, ]})
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")
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")
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')