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"
Exemplo n.º 2
0
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"