def test_ssl_cert_create_revert(self): mock_ssl_cert = cert_tasks.SSLCertCreate() mock_ssl_cert.axapi_client = self.client_mock mock_ssl_cert.revert(CERT_DATA, VTHUNDER) self.client_mock.file.ssl_cert.delete.assert_called_with( private_key=a10_test_constants.MOCK_KEY_FILENAME, cert_name=a10_test_constants.MOCK_CERT_FILENAME)
def test_ssl_cert_barbican_exception(self, barbican_class, cert_data): mock_ssl_cert = cert_tasks.SSLCertCreate() mock_ssl_cert.axapi_client = self.client_mock barbican_class.side_effect = Exception self.assertRaises( Exception, lambda: mock_ssl_cert.execute(LB, self.listener, VTHUNDER)) self.client_mock.file.ssl_cert.exists.assert_not_called()
def test_ssl_cert_create_new(self, barbican_class, cert_data): mock_ssl_cert = cert_tasks.SSLCertCreate() mock_ssl_cert.axapi_client = self.client_mock self.client_mock.file.ssl_cert.exists.return_value = False mock_ssl_cert.execute(LB, self.listener, VTHUNDER) self.client_mock.file.ssl_cert.create.assert_called_with( file=a10_test_constants.MOCK_CERT_FILENAME, cert=a10_test_constants.MOCK_CERT_CONTENT, size=len(a10_test_constants.MOCK_CERT_CONTENT), action=a10_test_constants.MOCK_CERT_ACTION, certificate_type=a10_test_constants.MOCK_CERT_TYPE) self.client_mock.file.ssl_cert.update.assert_not_called()
def get_ssl_certificate_create_flow(self): create_ssl_cert_flow = linear_flow.Flow( a10constants.CREATE_SSL_CERT_FLOW) create_ssl_cert_flow.add(cert_tasks.GetSSLCertData( requires=[constants.LOADBALANCER, constants.LISTENER], provides=a10constants.CERT_DATA)) create_ssl_cert_flow.add(cert_tasks.SSLCertCreate( requires=[a10constants.CERT_DATA, a10constants.VTHUNDER])) create_ssl_cert_flow.add(cert_tasks.SSLKeyCreate( requires=[a10constants.CERT_DATA, a10constants.VTHUNDER])) create_ssl_cert_flow.add(cert_tasks.ClientSSLTemplateCreate( requires=[a10constants.CERT_DATA, a10constants.VTHUNDER])) return create_ssl_cert_flow
def test_ssl_cert_create_update(self): mock_ssl_cert = cert_tasks.SSLCertCreate() mock_ssl_cert.axapi_client = self.client_mock mock_ssl_cert.execute(CERT_DATA, VTHUNDER) self.client_mock.file.ssl_cert.exists.assert_called_with( file=a10_test_constants.MOCK_CERT_FILENAME) self.client_mock.file.ssl_cert.update.assert_called_with( file=a10_test_constants.MOCK_CERT_FILENAME, cert=a10_test_constants.MOCK_CERT_CONTENT, size=len(a10_test_constants.MOCK_CERT_CONTENT), action=a10_test_constants.MOCK_CERT_ACTION, certificate_type=a10_test_constants.MOCK_CERT_TYPE) self.client_mock.file.ssl_cert.create.assert_not_called()
def test_ssl_cert_create_revert(self, barbican_class, cert_data): mock_ssl_cert = cert_tasks.SSLCertCreate() mock_ssl_cert.axapi_client = self.client_mock mock_ssl_cert.revert(LB, self.listener, VTHUNDER) self.client_mock.file.ssl_cert.delete.assert_called_with( file=a10_test_constants.MOCK_CERT_FILENAME)
def test_GetSSLCertData_barbican_exception(self, barbican_class): mock_ssl_cert = cert_tasks.SSLCertCreate() mock_ssl_cert.axapi_client = self.client_mock barbican_class.side_effect = Exception self.assertRaises(Exception, lambda: mock_ssl_cert.execute(LB, LISTENER)) self.client_mock.file.ssl_cert.exists.assert_not_called()