def test_parse_brkt_env(self):
""" Test parsing of the command-line --brkt-env value.
"""
be = brkt_cli.parse_brkt_env(
'api.example.com:777,hsmproxy.example.com:888')
self.assertEqual('api.example.com', be.api_host)
self.assertEqual(777, be.api_port)
self.assertEqual('hsmproxy.example.com', be.hsmproxy_host)
self.assertEqual(888, be.hsmproxy_port)
with self.assertRaises(ValidationError):
brkt_cli.parse_brkt_env('a:7,b:8:9')
with self.assertRaises(ValidationError):
brkt_cli.parse_brkt_env('a:7,b?:8')
def test_parse_brkt_env(self):
""" Test parsing of the command-line --brkt-env value.
"""
be = brkt_cli.parse_brkt_env(
'api.example.com:777,hsmproxy.example.com:888')
self.assertEqual('api.example.com', be.api_host)
self.assertEqual(777, be.api_port)
self.assertEqual('hsmproxy.example.com', be.hsmproxy_host)
self.assertEqual(888, be.hsmproxy_port)
with self.assertRaises(ValidationError):
brkt_cli.parse_brkt_env('a:7,b:8:9')
with self.assertRaises(ValidationError):
brkt_cli.parse_brkt_env('a:7,b?:8')
def test_brkt_env_from_values(self):
""" Test parsing BracketEnvironment from the --service-domain and
--brkt-env command line options.
"""
# No values are set.
self.assertIsNone(brkt_cli.brkt_env_from_values(DummyValues()))
# Test --service-domain
values = DummyValues()
values.service_domain = 'example.com'
brkt_env = brkt_cli.brkt_env_from_values(values)
self.assertEqual(
str(brkt_cli.brkt_env_from_domain('example.com')),
str(brkt_env)
)
# Test --brkt-env
values = DummyValues()
values.brkt_env = 'yetiapi.example.com:443,' + \
'hsmproxy.example.com:443,network.example.com:443'
brkt_env = brkt_cli.brkt_env_from_values(values)
self.assertEqual(
str(brkt_cli.parse_brkt_env(values.brkt_env)),
str(brkt_env)
)
def test_add_brkt_env_to_user_data(self):
userdata = {}
api_host_port = 'api.example.com:777'
hsmproxy_host_port = 'hsmproxy.example.com:888'
expected_userdata = {
'api_host': api_host_port,
'hsmproxy_host': hsmproxy_host_port
}
brkt_env = brkt_cli.parse_brkt_env(
api_host_port + ',' + hsmproxy_host_port)
brkt_cli.add_brkt_env_to_brkt_config(brkt_env, userdata)
self.assertEqual(userdata, expected_userdata)
def test_add_brkt_env_to_user_data(self):
userdata = {}
api_host_port = 'api.example.com:777'
hsmproxy_host_port = 'hsmproxy.example.com:888'
expected_userdata = {
'api_host': api_host_port,
'hsmproxy_host': hsmproxy_host_port
}
brkt_env = brkt_cli.parse_brkt_env(api_host_port + ',' +
hsmproxy_host_port)
brkt_cli.add_brkt_env_to_brkt_config(brkt_env, userdata)
self.assertEqual(userdata, expected_userdata)
def test_brkt_env_from_values(self):
""" Test parsing BracketEnvironment from the --service-domain and
--brkt-env command line options.
"""
# No values are set.
self.assertIsNone(brkt_cli.brkt_env_from_values(DummyValues()))
# Test --service-domain
values = DummyValues()
values.service_domain = 'example.com'
brkt_env = brkt_cli.brkt_env_from_values(values)
self.assertEqual(str(brkt_cli.brkt_env_from_domain('example.com')),
str(brkt_env))
# Test --brkt-env
values = DummyValues()
values.brkt_env = 'yetiapi.example.com:443,hsmproxy.example.com:443'
brkt_env = brkt_cli.brkt_env_from_values(values)
self.assertEqual(str(brkt_cli.parse_brkt_env(values.brkt_env)),
str(brkt_env))
