def test_node_destroy_by_instance_name(
self,
VPCMock,
ConnectionMock,
InstancesMock,
):
_instances_obj_mock = mock.Mock()
InstancesMock.get = mock.Mock(return_value=_instances_obj_mock)
InstancesMock.get_hostnames_by_roles = mock.Mock(
return_value={'NODE': 'node-instance-name.domain'})
node = Node(vpc_id='vpc-id', name='node-instance-name')
node.destroy()
InstancesMock.get.assert_called_once_with(filters=[
{
'Name': 'tag-key',
'Values': ['Name']
},
{
'Name': 'tag-value',
'Values': ['node-instance-name']
},
])
_instances_obj_mock.terminate.assert_called_once_with()
def test_hostnames_for_none(self, ConnectionMock, InstancesMock):
InstancesMock.get_hostnames_by_roles = mock.Mock(return_value={})
node = Node(vpc_id='vpc-id', name='node')
self.assertEqual(node.hostnames_for(roles=['IPA']), [None])
self.assertEqual(node.hostnames_for(roles=[]), [])
def test_hostnames_for_single(self, ConnectionMock, InstancesMock):
InstancesMock.get_hostnames_by_roles = mock.Mock(return_value={
'IPA': 'ipa-hostname',
})
node = Node(vpc_id='vpc-id', name='node')
_ipa_hostname, = node.hostnames_for(roles=['IPA'])
self.assertEqual(_ipa_hostname, 'ipa-hostname')
def test_node_destroy_without_identifier_should_do_nothing(
self,
VPCMock,
ConnectionMock,
InstancesMock,
):
InstancesMock.get = mock.Mock()
node = Node(vpc_id='vpc-id', name=None)
node.destroy()
InstancesMock.get.assert_not_called()
def test_hostnames_for_multiple(self, ConnectionMock, InstancesMock):
InstancesMock.get_hostnames_by_roles = mock.Mock(
return_value={
'IPA': 'ipa-hostname',
'LDAP': 'ldap-hostname',
})
node = Node(vpc_id='vpc-id', name='node')
_ldap_hostname, _ipa_hostname = node.hostnames_for(
roles=['LDAP', 'IPA'])
self.assertEqual(_ldap_hostname, 'ldap-hostname')
self.assertEqual(_ipa_hostname, 'ipa-hostname')
def test_node_destroy_by_instance_id(
self,
VPCMock,
ConnectionMock,
InstancesMock,
):
_instances_obj_mock = mock.Mock()
InstancesMock.get = mock.Mock(return_value=_instances_obj_mock)
InstancesMock.get_hostnames_by_roles = mock.Mock(
return_value={'NODE': 'node1-1000.domain'})
node = Node(vpc_id='vpc-id', name='node')
node.destroy(instance_id='instance-id')
InstancesMock.get.assert_called_once_with(ids=['instance-id'])
_instances_obj_mock.terminate.assert_called_once_with()
def test_zk_url_standalone(self, ConnectionMock):
node = Node(vpc_id='vpc-id', name='node')
_zk_url = node._zk_url(hostname='zk1')
self.assertEqual(_zk_url, 'zookeeper://foo@zk1:2181')
def test_zk_url_cluster(self, ConnectionMock):
node = Node(vpc_id='vpc-id', name='node')
_zk_url = node._zk_url(hostname='zk1,zk2,zk3')
self.assertEqual(_zk_url, 'zookeeper://foo@zk1:2181,zk2:2181,zk3:2181')