def test_dev(self):
collated_service_yamls = {
"devx": {
"devxhost": 1,
},
"xdev": {
"should_not_be_included": 1,
},
}
habitats = autosuggest.discover_habitats(collated_service_yamls)
assert "devx" in habitats
assert "xdev" not in habitats
def test_stage(self):
collated_service_yamls = {
"stagex": {
"stagexhost": 1,
},
"xstage": {
"should_not_be_included": 1,
},
}
habitats = autosuggest.discover_habitats(collated_service_yamls)
assert "stagex" in habitats
assert "xstage" not in habitats
def test_prod(self):
# Prod values are hardcoded, so even with no discovered habitats they
# should appear.
collated_service_yamls = {}
habitats = autosuggest.discover_habitats(collated_service_yamls)
assert "iad1" in habitats