def test_get_smartstack_replication_for_attribute():
fake_namespace = 'fake_main'
fake_service = 'fake_service'
mock_filtered_slaves = [
{
'hostname': 'hostone',
'attributes': {
'fake_attribute': 'foo',
},
},
{
'hostname': 'hostone',
'attributes': {
'fake_attribute': 'bar',
},
},
]
fake_system_paasta_config = SystemPaastaConfig({}, '/fake/config')
with mock.patch(
'paasta_tools.mesos_tools.get_all_slaves_for_blacklist_whitelist',
return_value=mock_filtered_slaves,
autospec=True,
) as mock_get_all_slaves_for_blacklist_whitelist, mock.patch(
'paasta_tools.smartstack_tools.get_replication_for_services',
return_value={},
autospec=True,
) as mock_get_replication_for_services:
expected = {
'foo': {},
'bar': {},
}
actual = smartstack_tools.get_smartstack_replication_for_attribute(
attribute='fake_attribute',
service=fake_service,
namespace=fake_namespace,
blacklist=[],
system_paasta_config=fake_system_paasta_config,
)
mock_get_all_slaves_for_blacklist_whitelist.assert_called_once_with(
blacklist=[],
whitelist=None,
)
assert actual == expected
assert mock_get_replication_for_services.call_count == 2
mock_get_replication_for_services.assert_any_call(
synapse_host='hostone',
synapse_port=fake_system_paasta_config.get_synapse_port(),
synapse_haproxy_url_format=fake_system_paasta_config.
get_synapse_haproxy_url_format(),
services=['fake_service.fake_main'],
)
def test_get_smartstack_replication_for_attribute():
fake_namespace = 'fake_main'
fake_service = 'fake_service'
mock_filtered_slaves = [
{
'hostname': 'hostone',
'attributes': {
'fake_attribute': 'foo'
}
},
{
'hostname': 'hostone',
'attributes': {
'fake_attribute': 'bar'
}
}
]
fake_system_paasta_config = SystemPaastaConfig({}, '/fake/config')
with contextlib.nested(
mock.patch('paasta_tools.mesos_tools.get_all_slaves_for_blacklist_whitelist',
return_value=mock_filtered_slaves, autospec=True),
mock.patch('paasta_tools.smartstack_tools.get_replication_for_services',
return_value={}, autospec=True),
) as (
mock_get_all_slaves_for_blacklist_whitelist,
mock_get_replication_for_services,
):
expected = {
'foo': {},
'bar': {}
}
actual = smartstack_tools.get_smartstack_replication_for_attribute(
attribute='fake_attribute',
service=fake_service,
namespace=fake_namespace,
blacklist=[],
system_paasta_config=fake_system_paasta_config,
)
mock_get_all_slaves_for_blacklist_whitelist.assert_called_once_with(
blacklist=[],
whitelist=[]
)
assert actual == expected
assert mock_get_replication_for_services.call_count == 2
mock_get_replication_for_services.assert_any_call(
synapse_host='hostone',
synapse_port=fake_system_paasta_config.get_synapse_port(),
synapse_haproxy_url_format=fake_system_paasta_config.get_synapse_haproxy_url_format(),
services=['fake_service.fake_main'],
)
def test_get_smartstack_replication_for_attribute(system_paasta_config):
fake_namespace = "fake_main"
fake_service = "fake_service"
mock_filtered_slaves = [
{
"hostname": "hostone",
"attributes": {
"fake_attribute": "foo"
}
},
{
"hostname": "hostone",
"attributes": {
"fake_attribute": "bar"
}
},
]
with mock.patch(
"paasta_tools.mesos_tools.get_all_slaves_for_blacklist_whitelist",
return_value=mock_filtered_slaves,
autospec=True,
) as mock_get_all_slaves_for_blacklist_whitelist, mock.patch(
"paasta_tools.smartstack_tools.get_replication_for_services",
return_value={},
autospec=True,
) as mock_get_replication_for_services:
expected = {"foo": {}, "bar": {}}
actual = smartstack_tools.get_smartstack_replication_for_attribute(
attribute="fake_attribute",
service=fake_service,
namespace=fake_namespace,
blacklist=[],
system_paasta_config=system_paasta_config,
)
mock_get_all_slaves_for_blacklist_whitelist.assert_called_once_with(
blacklist=[], whitelist=None)
assert actual == expected
assert mock_get_replication_for_services.call_count == 2
mock_get_replication_for_services.assert_any_call(
synapse_host="hostone",
synapse_port=system_paasta_config.get_synapse_port(),
synapse_haproxy_url_format=system_paasta_config.
get_synapse_haproxy_url_format(),
services=["fake_service.fake_main"],
)