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))