def test_not_prod(self):
"""All non-prod habitats have the same workflow, so just test one."""
expected = "stagexservices1"
collated_service_yamls = {
"stagex": {
expected: 5,
"stagexservices2": 10,
"stagex-ineligibile-non-services-box3": 1,
},
}
actual = autosuggest.suggest_hosts_for_habitat(collated_service_yamls, "stagex")
assert expected == actual
def test_prod(self):
expected = "host1,host2"
collated_service_yamls = {
"hab1": {
"host1": 99,
"host2": 99,
"host3": 1,
"host4": 1,
},
# All these machines will be filtered out since they're not general service machines.
"hab2": {
"host5": 10,
"host6": 1,
"host7": 1,
},
}
actual = autosuggest.suggest_hosts_for_habitat(collated_service_yamls, "hab1")
assert expected == actual
def test_habitat_not_in_collated_service_yamls(self):
collated_service_yamls = {}
actual = autosuggest.suggest_hosts_for_habitat(collated_service_yamls, "nonexistent")
assert "" == actual
