def test_success(self, mock_destroy): self.mock_runner.run.return_value = ("tool output", "", 0) lib.client_setup(self.mock_runner, "certificate data".encode("utf-8")) self.assertEqual("certificate data".encode("utf-8"), open(self.ca_file_path, "rb").read()) self.mock_runner.run.assert_called_once_with( [_client_cert_tool, "-i", "-c", self.ca_file_path]) mock_destroy.assert_called_once_with()
def test_success(self, mock_destroy): self.mock_runner.run.return_value = ("tool output", 0) lib.client_setup(self.mock_runner, "certificate data".encode("utf-8")) self.assertEqual( "certificate data".encode("utf-8"), open(self.ca_file_path, "rb").read() ) self.mock_runner.run.assert_called_once_with([ _client_cert_tool, "-i", "-c", self.ca_file_path ]) mock_destroy.assert_called_once_with()
def client_net_setup(lib_env, ca_certificate): """ Intialize qdevice net client on local host ca_certificate base64 encoded qnetd CA certificate """ try: ca_certificate_data = base64.b64decode(ca_certificate) except (TypeError, binascii.Error): raise LibraryError( reports.invalid_option_value("qnetd CA certificate", ca_certificate, ["base64 encoded certificate"])) qdevice_net.client_setup(lib_env.cmd_runner(), ca_certificate_data)
def client_net_setup(lib_env, ca_certificate): """ Intialize qdevice net client on local host ca_certificate base64 encoded qnetd CA certificate """ try: ca_certificate_data = base64.b64decode(ca_certificate) except (TypeError, binascii.Error): raise LibraryError(reports.invalid_option_value( "qnetd CA certificate", ca_certificate, ["base64 encoded certificate"] )) qdevice_net.client_setup(lib_env.cmd_runner(), ca_certificate_data)
def client_net_setup(lib_env: LibraryEnvironment, ca_certificate): """ Intialize qdevice net client on local host ca_certificate base64 encoded qnetd CA certificate """ try: ca_certificate_data = base64.b64decode(ca_certificate) except (TypeError, binascii.Error): raise LibraryError( ReportItem.error( reports.messages.InvalidOptionValue( "qnetd CA certificate", ca_certificate, ["base64 encoded certificate"], ))) qdevice_net.client_setup(lib_env.cmd_runner(), ca_certificate_data)
def test_init_error(self, mock_destroy): self.mock_runner.run.return_value = ("stdout", "tool output error", 1) assert_raise_library_error( lambda: lib.client_setup( self.mock_runner, "qnetd CA certificate".encode("utf-8") ), ( severity.ERROR, report_codes.QDEVICE_INITIALIZATION_ERROR, { "model": "net", "reason": "tool output error\nstdout", } ) ) self.assertEqual( "qnetd CA certificate".encode("utf-8"), open(self.ca_file_path, "rb").read() ) self.mock_runner.run.assert_called_once_with([ _client_cert_tool, "-i", "-c", self.ca_file_path ]) mock_destroy.assert_called_once_with()