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')])