def test_get_broker(self):
os.environ["POLYAXON_ENVIRONMENT"] = "testing"
os.environ.pop("POLYAXON_BROKER_BACKEND", None)
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json"
])
assert config.broker_backend == "rabbitmq"
assert config.is_redis_broker is False
assert config.is_rabbitmq_broker is True
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_BROKER_BACKEND": "rabbitmq"
},
])
assert config.broker_backend == "rabbitmq"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_BROKER_BACKEND": "redis"
},
])
assert config.broker_backend == "redis"
assert config.is_redis_broker is True
assert config.is_rabbitmq_broker is False
def test_get_from_os_env(self):
os.environ["POLYAXON_ENVIRONMENT"] = "testing"
os.environ["FOO_BAR_KEY"] = "foo_bar"
config = ConfigManager.read_configs(
[os.environ, "./tests/fixtures_static/configs/non_opt_config_tests.json"]
)
assert config.get_string("POLYAXON_ENVIRONMENT") == "testing"
assert config.get_string("FOO_BAR_KEY") == "foo_bar"
def test_get_broker_url(self):
os.environ["POLYAXON_ENVIRONMENT"] = "testing"
os.environ.pop("POLYAXON_RABBITMQ_USER", None)
os.environ.pop("POLYAXON_RABBITMQ_PASSWORD", None)
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_BROKER_BACKEND": "redis",
"POLYAXON_REDIS_CELERY_BROKER_URL": "foo",
},
])
assert config.get_broker_url() == "redis://foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_REDIS_PROTOCOL": "rediss",
"POLYAXON_BROKER_BACKEND": "redis",
"POLYAXON_REDIS_CELERY_BROKER_URL": "foo",
},
])
assert config.get_broker_url() == "rediss://foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_BROKER_BACKEND": "redis",
"POLYAXON_REDIS_CELERY_BROKER_URL": "foo",
"POLYAXON_REDIS_PASSWORD": "******",
},
])
assert config.get_broker_url() == "redis://:pass@foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_AMQP_URL": "foo",
"POLYAXON_BROKER_BACKEND": "rabbitmq",
"POLYAXON_REDIS_CELERY_BROKER_URL": "foo",
},
])
assert config.get_broker_url() == "amqp://foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_AMQP_URL": "foo",
"POLYAXON_BROKER_BACKEND": "rabbitmq",
"POLYAXON_RABBITMQ_PASSWORD": "",
"POLYAXON_REDIS_CELERY_BROKER_URL": "foo",
},
])
assert config.get_broker_url() == "amqp://foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_AMQP_URL": "foo",
"POLYAXON_BROKER_BACKEND": "rabbitmq",
"POLYAXON_RABBITMQ_PASSWORD": "",
"POLYAXON_RABBITMQ_USER": "******",
},
])
assert config.get_broker_url() == "amqp://foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_AMQP_URL": "foo",
"POLYAXON_BROKER_BACKEND": "rabbitmq",
"POLYAXON_RABBITMQ_USER": "",
"POLYAXON_RABBITMQ_PASSWORD": "******",
},
])
assert config.get_broker_url() == "amqp://foo"
config = ConfigManager.read_configs([
os.environ,
"./tests/fixtures_static/configs/non_opt_config_tests.json",
{
"POLYAXON_AMQP_URL": "foo",
"POLYAXON_BROKER_BACKEND": "rabbitmq",
"POLYAXON_RABBITMQ_USER": "******",
"POLYAXON_RABBITMQ_PASSWORD": "******",
},
])
assert config.get_broker_url() == "amqp://*****:*****@foo"
def test_raises_for_non_optional_env_vars(self):
with self.assertRaises(PolyaxonSchemaError):
ConfigManager.read_configs([os.environ])
