def test_approval_bulk_create_csv(cbcsdk_mock):
"""Test the bulk_create_csv function."""
req = "vendor_id,product_id,serial_number,approval_name,notes\n" \
"0x0781,0x5581,4C531001331122115172,Example Approval,A few notes\n" \
"0x0666,0x6969,4Q2123456789,Example Approval2,Whatever"
def post_validate(url, body, **kwargs):
assert body == req
headers = kwargs.pop('headers')
assert headers['Content-Type'] == 'text/csv'
return USBDEVICE_APPROVAL_BULK_CREATE_RESP
cbcsdk_mock.mock_request("POST",
"/device_control/v3/orgs/test/approvals/_bulk",
post_validate)
api = cbcsdk_mock.api
results = USBDeviceApproval.bulk_create_csv(api, req)
assert len(results) == 2
approval = results[0]
assert approval._model_unique_id == "10373"
assert approval.vendor_id == "0x0781"
assert approval.vendor_name == "SanDisk"
assert approval.product_id == "0x5581"
assert approval.product_name == "Ultra"
assert approval.approval_name == "Example Approval"
approval = results[1]
assert approval._model_unique_id == "10444"
assert approval.vendor_id == "0x0666"
assert approval.vendor_name == "Sirius Cybernetics Corp."
assert approval.product_id == "0x6969"
assert approval.product_name == "Happy Hard Drive"
assert approval.approval_name == "Example Approval2"
def test_approval_create_from_usb_device(cbcsdk_mock):
"""Test the create_from_usb_device function."""
api = cbcsdk_mock.api
usb = USBDevice(api, USBDEVICE_GET_RESP["id"], USBDEVICE_GET_RESP)
approval = USBDeviceApproval.create_from_usb_device(usb)
assert approval.serial_number == "4C531001331122115172"
assert approval.vendor_id == "0x0781"
assert approval.product_id == "0x5581"
def test_approval_get_and_modify(cbcsdk_mock):
"""Tests a simple load, modify, and store of an approval object."""
def put_validate(url, body, **kwargs):
assert body["approval_name"] == "Altered Approval"
assert body["notes"] == "Altered state"
assert body["vendor_id"] == "0x0781"
assert body["product_id"] == "0x5581"
return USBDEVICE_APPROVAL_PUT_RESP
cbcsdk_mock.mock_request("GET",
"/device_control/v3/orgs/test/approvals/10373",
USBDEVICE_APPROVAL_GET_RESP)
cbcsdk_mock.mock_request("PUT",
"/device_control/v3/orgs/test/approvals/10373",
put_validate)
api = cbcsdk_mock.api
approval = USBDeviceApproval(api, "10373")
assert approval._model_unique_id == "10373"
assert approval.vendor_id == "0x0781"
assert approval.vendor_name == "SanDisk"
assert approval.product_id == "0x5581"
assert approval.product_name == "Ultra"
assert approval.approval_name == "Example Approval"
approval.notes = "Altered state"
approval.approval_name = "Altered Approval"
approval.save()
assert approval.vendor_id == "0x0781"
assert approval.vendor_name == "SanDisk"
assert approval.product_id == "0x5581"
assert approval.product_name == "Ultra"
assert approval.notes == "Altered state"
assert approval.approval_name == "Altered Approval"
def test_approval_create_and_save(cbcsdk_mock):
"""Tests a new approval being created and saved."""
cbcsdk_mock.mock_request("POST", "/device_control/v3/orgs/test/approvals/_bulk",
USBDEVICE_APPROVAL_BULK_CREATE_RESP)
api = cbcsdk_mock.api
approval = USBDeviceApproval(api, None)
approval.vendor_id = "0x0781"
approval.product_id = "0x5581"
approval.serial_number = "4C531001331122115172"
approval.notes = "A few notes"
approval.approval_name = "Example Approval"
approval.save()
assert approval._model_unique_id == "10373"
assert approval.vendor_id == "0x0781"
assert approval.vendor_name == "SanDisk"
assert approval.product_id == "0x5581"
assert approval.product_name == "Ultra"
assert approval.approval_name == "Example Approval"
assert approval.notes == "A few notes"
def test_approval_bulk_create(cbcsdk_mock):
"""Test the bulk_create function."""
cbcsdk_mock.mock_request("POST", "/device_control/v3/orgs/test/approvals/_bulk",
USBDEVICE_APPROVAL_BULK_CREATE_RESP)
api = cbcsdk_mock.api
results = USBDeviceApproval.bulk_create(api, USBDEVICE_APPROVAL_BULK_CREATE_REQ)
assert len(results) == 2
approval = results[0]
assert approval._model_unique_id == "10373"
assert approval.vendor_id == "0x0781"
assert approval.vendor_name == "SanDisk"
assert approval.product_id == "0x5581"
assert approval.product_name == "Ultra"
assert approval.approval_name == "Example Approval"
approval = results[1]
assert approval._model_unique_id == "10444"
assert approval.vendor_id == "0x0666"
assert approval.vendor_name == "Sirius Cybernetics Corp."
assert approval.product_id == "0x6969"
assert approval.product_name == "Happy Hard Drive"
assert approval.approval_name == "Example Approval2"