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()
