class CertificateRabbitMQTest(TestCase):
def setUp(self):
self.host = '127.0.0.1'
self.connection = connection(self.host)
self._certificate_rabbitmq = CertificateRabbitMQ(self.connection)
@mock.patch.object(ResourceClient, 'create')
def test_generate_called_once_with_defaults(self, mock_create):
information = {"commonName": "default", "type": "RabbitMqClientCertV2"}
self._certificate_rabbitmq.generate(information)
mock_create.assert_called_once_with(information, timeout=-1)
@mock.patch.object(ResourceClient, 'get')
def test_get_by_alias_name_called_once(self, mock_get):
alias_name = 'default'
self._certificate_rabbitmq.get(alias_name)
mock_get.assert_called_once_with(alias_name)
@mock.patch.object(ResourceClient, 'get')
def test_get_key_pair_called_once(self, mock_get):
alias_name = 'default'
self._certificate_rabbitmq.get_key_pair(alias_name)
uri = "/rest/certificates/client/rabbitmq/keypair/" + alias_name
mock_get.assert_called_once_with(uri)
@mock.patch.object(ResourceClient, 'get')
def test_get_keys_called_once(self, mock_get):
alias_name = 'default'
key_format = 'Base64'
self._certificate_rabbitmq.get_keys(alias_name, key_format)
uri = "/rest/certificates/client/rabbitmq/keys/" + alias_name + "?format=" + key_format
mock_get.assert_called_once_with(uri)
def certificate_rabbitmq(self):
"""
Gets the Certificate RabbitMQ API client.
Returns:
CertificateRabbitMQ:
"""
if not self.__certificate_rabbitmq:
self.__certificate_rabbitmq = CertificateRabbitMQ(self.__connection)
return self.__certificate_rabbitmq
def setUp(self):
self.host = '127.0.0.1'
self.connection = connection(self.host)
self._certificate_rabbitmq = CertificateRabbitMQ(self.connection)