def test_state_refresh_disable_cache():
'''Disables caching via a scheduler envvar'''
check_running(FOLDERED_SERVICE_NAME)
task_ids = tasks.get_task_ids(FOLDERED_SERVICE_NAME, '')
# caching enabled by default:
stdout = cmd.run_cli('hello-world --name={} state refresh_cache'.format(
FOLDERED_SERVICE_NAME))
assert "Received cmd: refresh" in stdout
config = marathon.get_config(FOLDERED_SERVICE_NAME)
config['env']['DISABLE_STATE_CACHE'] = 'any-text-here'
marathon.update_app(FOLDERED_SERVICE_NAME, config)
tasks.check_tasks_not_updated(FOLDERED_SERVICE_NAME, '', task_ids)
check_running(FOLDERED_SERVICE_NAME)
# caching disabled, refresh_cache should fail with a 409 error (eventually, once scheduler is up):
def check_cache_refresh_fails_409conflict():
try:
cmd.run_cli('hello-world --name={} state refresh_cache'.format(
FOLDERED_SERVICE_NAME))
except Exception as e:
if "failed: 409 Conflict" in e.args[0]:
return True
return False
shakedown.wait_for(lambda: check_cache_refresh_fails_409conflict(),
timeout_seconds=120.)
config = marathon.get_config(FOLDERED_SERVICE_NAME)
del config['env']['DISABLE_STATE_CACHE']
marathon.update_app(FOLDERED_SERVICE_NAME, config)
tasks.check_tasks_not_updated(FOLDERED_SERVICE_NAME, '', task_ids)
check_running(FOLDERED_SERVICE_NAME)
shakedown.deployment_wait(
) # ensure marathon thinks the deployment is complete too
# caching reenabled, refresh_cache should succeed (eventually, once scheduler is up):
def check_cache_refresh():
return cmd.run_cli('hello-world --name={} state refresh_cache'.format(
FOLDERED_SERVICE_NAME))
stdout = shakedown.wait_for(lambda: check_cache_refresh(),
timeout_seconds=120.)
assert "Received cmd: refresh" in stdout
def test_state_refresh_disable_cache():
'''Disables caching via a scheduler envvar'''
check_running()
task_ids = tasks.get_task_ids(PACKAGE_NAME, '')
# caching enabled by default:
stdout = cmd.run_cli('hello-world state refresh_cache')
assert "Received cmd: refresh" in stdout
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['HELLO_CPUS'])
config['env']['DISABLE_STATE_CACHE'] = 'any-text-here'
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_not_updated(PACKAGE_NAME, '', task_ids)
check_running()
# caching disabled, refresh_cache should fail with a 409 error (eventually, once scheduler is up):
def check_cache_refresh_fails_409conflict():
try:
cmd.run_cli('hello-world state refresh_cache')
except Exception as e:
if "failed: 409 Conflict" in e.args[0]:
return True
return False
spin.time_wait_noisy(lambda: check_cache_refresh_fails_409conflict(),
timeout_seconds=120.)
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['HELLO_CPUS'])
del config['env']['DISABLE_STATE_CACHE']
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_not_updated(PACKAGE_NAME, '', task_ids)
check_running()
# caching reenabled, refresh_cache should succeed (eventually, once scheduler is up):
def check_cache_refresh():
return cmd.run_cli('hello-world state refresh_cache')
stdout = spin.time_wait_return(lambda: check_cache_refresh(),
timeout_seconds=120.)
assert "Received cmd: refresh" in stdout
def test_bump_world_cpus():
check_running()
world_ids = tasks.get_task_ids(PACKAGE_NAME, 'world')
print('world ids: ' + str(world_ids))
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['WORLD_CPUS'])
config['env']['WORLD_CPUS'] = str(cpus + 0.1)
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_updated(PACKAGE_NAME, 'world', world_ids)
check_running()
def test_bump_hello_cpus():
check_running()
hello_ids = tasks.get_task_ids(PACKAGE_NAME, 'hello')
print('hello ids: ' + str(hello_ids))
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['HELLO_CPUS'])
config['env']['HELLO_CPUS'] = str(cpus + 0.1)
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_updated(PACKAGE_NAME, 'hello', hello_ids)
check_running()
def test_bump_hello_cpus():
check_running()
hello_ids = tasks.get_task_ids(PACKAGE_NAME, 'hello')
print('hello ids: ' + str(hello_ids))
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['HELLO_CPUS'])
config['env']['HELLO_CPUS'] = str(cpus + 0.1)
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_updated(PACKAGE_NAME, 'hello', hello_ids)
check_running()
def test_state_refresh_disable_cache():
'''Disables caching via a scheduler envvar'''
check_running()
task_ids = tasks.get_task_ids(PACKAGE_NAME, '')
# caching enabled by default:
stdout = cmd.run_cli('hello-world state refresh_cache')
assert "Received cmd: refresh" in stdout
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['HELLO_CPUS'])
config['env']['DISABLE_STATE_CACHE'] = 'any-text-here'
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_not_updated(PACKAGE_NAME, '', task_ids)
check_running()
# caching disabled, refresh_cache should fail with a 409 error (eventually, once scheduler is up):
def check_cache_refresh_fails_409conflict():
try:
cmd.run_cli('hello-world state refresh_cache')
except Exception as e:
if "failed: 409 Conflict" in e.args[0]:
return True
return False
spin.time_wait_noisy(lambda: check_cache_refresh_fails_409conflict(), timeout_seconds=120.)
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['HELLO_CPUS'])
del config['env']['DISABLE_STATE_CACHE']
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
tasks.check_tasks_not_updated(PACKAGE_NAME, '', task_ids)
check_running()
# caching reenabled, refresh_cache should succeed (eventually, once scheduler is up):
def check_cache_refresh():
return cmd.run_cli('hello-world state refresh_cache')
stdout = spin.time_wait_return(lambda: check_cache_refresh(), timeout_seconds=120.)
assert "Received cmd: refresh" in stdout
def test_bump_hello_nodes():
check_running()
hello_ids = tasks.get_task_ids(PACKAGE_NAME, 'hello')
print('hello ids: ' + str(hello_ids))
config = marathon.get_config(PACKAGE_NAME)
node_count = int(config['env']['HELLO_COUNT']) + 1
config['env']['HELLO_COUNT'] = str(node_count)
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
check_running()
tasks.check_tasks_not_updated(PACKAGE_NAME, 'hello', hello_ids)
def test_bump_hello_nodes():
check_running()
hello_ids = tasks.get_task_ids(PACKAGE_NAME, 'hello')
print('hello ids: ' + str(hello_ids))
config = marathon.get_config(PACKAGE_NAME)
node_count = int(config['env']['HELLO_COUNT']) + 1
config['env']['HELLO_COUNT'] = str(node_count)
marathon.update_app(PACKAGE_NAME, config)
check_running()
tasks.check_tasks_not_updated(PACKAGE_NAME, 'hello', hello_ids)
def test_bump_hello_nodes():
check_running()
hello_ids = tasks.get_task_ids(PACKAGE_NAME, 'hello')
print('hello ids: ' + str(hello_ids))
config = marathon.get_config(PACKAGE_NAME)
node_count = int(config['env']['HELLO_COUNT']) + 1
config['env']['HELLO_COUNT'] = str(node_count)
cmd.request('put', marathon.api_url('apps/' + PACKAGE_NAME), json=config)
check_running()
tasks.check_tasks_not_updated(PACKAGE_NAME, 'hello', hello_ids)
def test_bump_hello_nodes():
check_running()
hello_ids = tasks.get_task_ids(PACKAGE_NAME, 'hello')
print('hello ids: ' + str(hello_ids))
config = marathon.get_config(PACKAGE_NAME)
node_count = int(config['env']['HELLO_COUNT']) + 1
config['env']['HELLO_COUNT'] = str(node_count)
marathon.update_app(PACKAGE_NAME, config)
check_running()
tasks.check_tasks_not_updated(PACKAGE_NAME, 'hello', hello_ids)
def test_bump_world_cpus():
check_running()
world_ids = tasks.get_task_ids(PACKAGE_NAME, 'world')
print('world ids: ' + str(world_ids))
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['WORLD_CPUS'])
updated_cpus = cpus + 0.1
config['env']['WORLD_CPUS'] = str(updated_cpus)
marathon.update_app(PACKAGE_NAME, config)
tasks.check_tasks_updated(PACKAGE_NAME, 'world', world_ids)
check_running()
all_tasks = shakedown.get_service_tasks(PACKAGE_NAME)
running_tasks = [t for t in all_tasks if t['name'].startswith('world') and t['state'] == "TASK_RUNNING"]
assert len(running_tasks) == world_task_count()
for t in running_tasks:
assert close_enough(t['resources']['cpus'], updated_cpus)
def test_bump_world_cpus():
check_running()
world_ids = tasks.get_task_ids(PACKAGE_NAME, 'world')
print('world ids: ' + str(world_ids))
config = marathon.get_config(PACKAGE_NAME)
cpus = float(config['env']['WORLD_CPUS'])
updated_cpus = cpus + 0.1
config['env']['WORLD_CPUS'] = str(updated_cpus)
marathon.update_app(PACKAGE_NAME, config)
tasks.check_tasks_updated(PACKAGE_NAME, 'world', world_ids)
check_running()
all_tasks = shakedown.get_service_tasks(PACKAGE_NAME)
running_tasks = [
t for t in all_tasks
if t['name'].startswith('world') and t['state'] == "TASK_RUNNING"
]
assert len(running_tasks) == world_task_count()
for t in running_tasks:
assert close_enough(t['resources']['cpus'], updated_cpus)
