def test_bootstrap_context_key_pair(self, *_): """tests finding a key file via bootstrap context """ name = 'test_bootstrap_context_key_pair' private_key_dir = tempfile.mkdtemp() private_key_path = '{0}/{1}.pem' \ .format(private_key_dir, name) open(private_key_path, 'w').close() cloudify_agent = MockContext() cloudify_agent['agent_key_path'] = private_key_path bootstrap_ctx = BootstrapContext({ 'cloudify_agent': cloudify_agent }) ctx = MockCloudifyContext(bootstrap_context=bootstrap_ctx, node_name=name) current_ctx.set(ctx=ctx) with mock.patch('ec2.utils.get_single_connected_node_by_type') \ as mock_get_connected: mock_get_connected.return_value = None output = instance._get_private_key('') self.assertEqual(private_key_path, output)
def test_relationship_key_pair(self, *_): """tests finding a key file via relationship """ name = 'test_relationship_key_pair' ctx = self.mock_ctx(name) current_ctx.set(ctx=ctx) ctx.node.properties['use_password'] = True private_key_dir = tempfile.mkdtemp() private_key_path = '{0}/{1}.pem' \ .format(private_key_dir, name) open(private_key_path, 'w').close() key_pair_node = MockNodeContext( properties={'private_key_path': private_key_path}) with mock.patch('ec2.utils.get_single_connected_node_by_type') \ as mock_get_single_connected_node_by_type: mock_get_single_connected_node_by_type.return_value = key_pair_node output = instance._get_private_key('') self.assertEqual(private_key_path, output)
def test_relationship_key_pair(self, *_): """tests finding a key file via relationship """ name = 'test_relationship_key_pair' ctx = self.mock_ctx(name) current_ctx.set(ctx=ctx) ctx.node.properties['use_password'] = True private_key_dir = tempfile.mkdtemp() private_key_path = '{0}/{1}.pem' \ .format(private_key_dir, name) open(private_key_path, 'w').close() key_pair_node = MockNodeContext(properties={ 'private_key_path': private_key_path }) with mock.patch('ec2.utils.get_single_connected_node_by_type') \ as mock_get_single_connected_node_by_type: mock_get_single_connected_node_by_type.return_value = key_pair_node output = instance._get_private_key('') self.assertEqual(private_key_path, output)
def test_bootstrap_context_key_pair(self, *_): """tests finding a key file via bootstrap context """ name = 'test_bootstrap_context_key_pair' private_key_dir = tempfile.mkdtemp() private_key_path = '{0}/{1}.pem' \ .format(private_key_dir, name) open(private_key_path, 'w').close() cloudify_agent = MockContext() cloudify_agent['agent_key_path'] = private_key_path bootstrap_ctx = BootstrapContext({'cloudify_agent': cloudify_agent}) ctx = MockCloudifyContext(bootstrap_context=bootstrap_ctx, node_name=name) current_ctx.set(ctx=ctx) with mock.patch('ec2.utils.get_single_connected_node_by_type') \ as mock_get_connected: mock_get_connected.return_value = None output = instance._get_private_key('') self.assertEqual(private_key_path, output)