def test_agents(self):
devs = devices.Devices(self._global_admin)
devs.devices = mock.MagicMock()
devs.agents()
devs.devices.assert_called_once_with(None, False, mock.ANY)
expected_filters = [
query.FilterBuilder(devices.Devices.type_attr).like('Agent')
]
self._verify_filters_list(expected_filters,
devs.devices.call_args[0][2])
def test_desktops(self):
devs = devices.Devices(self._global_admin)
devs.devices = mock.MagicMock()
devs.desktops()
devs.devices.assert_called_once_with(None, False, mock.ANY)
expected_filters = [
query.FilterBuilder(devices.Devices.type_attr).eq(
enum.DeviceType.WorkstationAgent)
]
self._verify_filters_list(expected_filters,
devs.devices.call_args[0][2])
def _test_filers(self, device_types, expected_device_types_filters):
devs = devices.Devices(self._global_admin)
devs.devices = mock.MagicMock()
devs.filers(deviceTypes=device_types)
devs.devices.assert_called_once_with(None, False, mock.ANY)
expected_filters = [
query.FilterBuilder(devices.Devices.type_attr).eq(deviceType)
for deviceType in expected_device_types_filters
]
self._verify_filters_list(expected_filters,
devs.devices.call_args[0][2])
def test_by_name(self):
devs = devices.Devices(self._global_admin)
devs.devices = mock.MagicMock()
names = ['dev1', 'dev2']
devs.by_name(names)
devs.devices.assert_called_once_with(None, False, mock.ANY)
expected_filters = [
query.FilterBuilder('name').eq(name) for name in names
]
self._verify_filters_list(expected_filters,
devs.devices.call_args[0][2])
def _get_list_folders_param(include=None,
include_deleted=False,
filter_deleted=False,
user_uid=None):
include = union(include or [], cloudfs.CloudFS.default)
builder = query.QueryParamBuilder().include(include)
if include_deleted:
builder.put('includeDeleted', True)
if filter_deleted:
query_filter = query.FilterBuilder('isDeleted').eq(True)
builder.addFilter(query_filter)
if user_uid:
builder.ownedBy(user_uid)
return builder.build()
def test_devices_with_filters(self):
self._test_devices([query.FilterBuilder('name').eq('test')])