def test_address_from_parts(self):
env = ArmEnvironment(host='test')
self.assertEqual(env.address, 'https://test')
env = ArmEnvironment(host='test', port=32455)
self.assertEqual(env.address, 'https://test:32455')
env = ArmEnvironment(host='test', port=32455, protocol='http')
self.assertEqual(env.address, 'http://test:32455')
env = ArmEnvironment(host='test', port=None, protocol='http')
self.assertEqual(env.address, 'http://test')
def test_init_fails_when_address_and_host_are_none(self):
with self.assertRaises(ValidationError) as context:
config = ArmEnvironment(port=80)
self.assertEqual(
str(context.exception),
'1 validation error for ArmEnvironment\n__root__\n AnsibleRM environment cannot be configured without "address" property or "host" property (type=value_error)'
)
def test_arm_driver(self, arm_driver_init):
session = ArmSession(
ArmSessionConfig(
ArmEnvironment('lm', host='test', port=80, protocol='https')))
driver = session.arm_driver
arm_driver_init.assert_called_once_with('https://test:80')
self.assertEqual(driver, arm_driver_init.return_value)
def test_init_with_all_as_parts(self):
config = ArmEnvironment(host='test',
port=31080,
protocol='http',
onboarding_addr='http://arm:80')
self.assertEqual(config.address, 'http://test:31080')
self.assertEqual(config.onboarding_addr, 'http://arm:80')
def test_config_to_file(self):
config = Config(
environments={
'test':
EnvironmentGroup(
name='test',
description='Just testing',
tnco=TNCOEnvironment(address='https://localhost:80',
secure=True,
username='******',
client_id='client'),
arms={
'defaultrm': ArmEnvironment(
address='https://localhost:81')
})
})
target_path = os.path.join(self.tmp_dir, 'write-config.yaml')
ConfigIO().config_to_file(config, target_path)
self.assertTrue(os.path.exists(target_path))
config_dict = self.test_helper.read_workspace_yaml_file(
'write-config.yaml')
self.assertEqual(
config_dict, {
'environments': {
'test': {
'description': 'Just testing',
'tnco': {
'address': 'https://localhost:80',
'secure': True,
'username': '******',
'client_id': 'client'
},
'arms': {
'defaultrm': {
'address': 'https://localhost:81'
}
}
}
}
})
def test_config_to_dict(self):
config = Config(
environments={
'test':
EnvironmentGroup(
name='test',
description='Just testing',
tnco=TNCOEnvironment(address='https://localhost:80',
secure=True,
username='******',
client_id='client'),
arms={
'defaultrm': ArmEnvironment(
address='https://localhost:81')
})
})
config_dict = ConfigIO().config_to_dict(config)
self.assertEqual(
config_dict, {
'environments': {
'test': {
'description': 'Just testing',
'tnco': {
'address': 'https://localhost:80',
'secure': True,
'username': '******',
'client_id': 'client'
},
'arms': {
'defaultrm': {
'address': 'https://localhost:81'
}
}
}
}
})
def test_init(self):
env = ArmEnvironment(host='test', port=80)
session_config = env.create_session_config()
session = ArmSession(session_config)
self.assertEqual(session.env, env)
def test_create(self):
env = ArmEnvironment(host='test', port=80, protocol='https')
session_config = env.create_session_config()
session = session_config.create()
self.assertIsInstance(session, ArmSession)
self.assertEqual(session.env, env)
def test_create_session_config(self):
env = ArmEnvironment(host='test')
session_config = env.create_session_config()
self.assertIsInstance(session_config, ArmSessionConfig)
self.assertEqual(session_config.env, env)
def test_init_with_address(self):
config = ArmEnvironment(address='https://test:8080/api')
self.assertEqual(config.address, 'https://test:8080/api')