def test_get_cmd_default(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={},
)
assert fake_conf.get_cmd() is None
def test_get_cpus_in_config_float(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'cpus': .66},
branch_dict={},
)
assert fake_conf.get_cpus() == .66
def test_get_mem_in_config(self):
fake_conf = utils.InstanceConfig(
service='',
instance='',
cluster='',
config_dict={'mem': -999},
branch_dict={},
)
assert fake_conf.get_mem() == -999
def test_get_env_with_config(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'env': {'SPECIAL_ENV': 'TRUE'}},
branch_dict={},
)
assert fake_conf.get_env() == {'SPECIAL_ENV': 'TRUE'}
def test_get_monitoring(self):
fake_info = 'fake_info'
assert utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'monitoring': fake_info},
branch_dict={},
).get_monitoring() == fake_info
def test_deploy_group_string_interpolation(self):
fake_conf = utils.InstanceConfig(
service='',
instance='',
cluster='fake_cluster',
config_dict={'deploy_group': 'cluster_is_{cluster}'},
branch_dict={},
)
assert fake_conf.get_deploy_group() == 'cluster_is_fake_cluster'
def test_deploy_group_if_config(self):
fake_conf = utils.InstanceConfig(
service='',
instance='',
cluster='',
config_dict={'deploy_group': 'fake_deploy_group'},
branch_dict={},
)
assert fake_conf.get_deploy_group() == 'fake_deploy_group'
def test_deploy_group_default(self):
fake_conf = utils.InstanceConfig(
service='',
instance='fake_instance',
cluster='fake_cluster',
config_dict={},
branch_dict={},
)
assert fake_conf.get_deploy_group() == 'fake_cluster.fake_instance'
def test_extra_volumes_default(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={},
)
assert fake_conf.get_extra_volumes() == []
def test_get_cmd_in_config(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'cmd': 'FAKECMD'},
branch_dict={},
)
assert fake_conf.get_cmd() == 'FAKECMD'
def test_get_pool_default(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={},
)
assert fake_conf.get_pool() == 'default'
def test_get_desired_state(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={'desired_state': 'stop'},
)
assert fake_conf.get_desired_state() == 'stop'
def test_deploy_blacklist_default(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={},
)
assert fake_conf.get_deploy_blacklist() == []
def test_get_args_in_config(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'args': ['arg1', 'arg2']},
branch_dict={},
)
assert fake_conf.get_args() == ['arg1', 'arg2']
def test_get_force_bounce(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={'force_bounce': 'blurp'},
)
assert fake_conf.get_force_bounce() == 'blurp'
def test_get_args_default_with_cmd(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'cmd': 'FAKECMD'},
branch_dict={},
)
assert fake_conf.get_args() is None
def test_get_args_default_no_cmd(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={},
branch_dict={},
)
assert fake_conf.get_args() == []
def test_deploy_blacklist_reads_blacklist(self):
fake_deploy_blacklist = [["region", "fake_region"]]
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'deploy_blacklist': fake_deploy_blacklist},
branch_dict={},
)
assert fake_conf.get_deploy_blacklist() == fake_deploy_blacklist
def test_get_pool(self):
pool = "poolname"
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'pool': pool},
branch_dict={},
)
assert fake_conf.get_pool() == pool
def test_get_args_in_config_with_cmd(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'args': ['A'], 'cmd': 'C'},
branch_dict={},
)
fake_conf.get_cmd()
with raises(utils.InvalidInstanceConfig):
fake_conf.get_args()
def test_get_mem_swap_float_rounds_up(self):
fake_conf = utils.InstanceConfig(
service='',
instance='',
cluster='',
config_dict={
'mem': 50.4
},
branch_dict={},
)
assert fake_conf.get_mem_swap() == "51m"
def test_get_mem_swap_int(self):
fake_conf = utils.InstanceConfig(
service='',
instance='',
cluster='',
config_dict={
'mem': 50
},
branch_dict={},
)
assert fake_conf.get_mem_swap() == "50m"
def test_get_env_default(self):
fake_conf = utils.InstanceConfig(
service='fake_service',
cluster='fake_cluster',
instance='fake_instance',
config_dict={},
branch_dict={},
)
assert fake_conf.get_env() == {
'PAASTA_SERVICE': 'fake_service',
'PAASTA_INSTANCE': 'fake_instance',
'PAASTA_CLUSTER': 'fake_cluster',
}
def test_get_env_with_config(self):
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'env': {'SPECIAL_ENV': 'TRUE'}},
branch_dict={},
)
assert fake_conf.get_env() == {
'SPECIAL_ENV': 'TRUE',
'PAASTA_SERVICE': '',
'PAASTA_INSTANCE': '',
'PAASTA_CLUSTER': '',
}
def test_extra_volumes_normal(self):
fake_extra_volumes = [
{
"containerPath": "/etc/a",
"hostPath": "/var/data/a",
"mode": "RO"
},
]
fake_conf = utils.InstanceConfig(
service='',
cluster='',
instance='',
config_dict={'extra_volumes': fake_extra_volumes},
branch_dict={},
)
assert fake_conf.get_extra_volumes() == fake_extra_volumes
