def test_with_next_previous_prefix(self):
"""When next or previous prefixes are used, the host should be modified"""
host = "foo.com"
self.assertEqual(blue_green_host(host, "previous"), "previous.{}".format(host))
self.assertEqual(blue_green_host(host, "next"), "next.{}".format(host))
def test_with_empty_prefix(self):
"""When the prefix is empty, the host should be left unmodified"""
host = "foo.com"
self.assertEqual(blue_green_host(host), host)
self.assertEqual(blue_green_host(host, prefix=""), host)
def test_with_not_allowed_prefix(self):
"""If prefix is not allowed, we should raise an error"""
host = "foo.com"
with self.assertRaises(AnsibleFilterError) as context_manager:
blue_green_host(host, prefix="spam")
self.assertEqual(
context_manager.exception.message,
"prefix 'spam' is not allowed (must be in ['previous', 'current', 'next'])",
)
def test_with_current_prefix(self):
"""When the prefix is "current", the host should be left unmodified"""
host = "foo.com"
self.assertEqual(blue_green_host(host, "current"), host)
