def test_receiver_get_all_sorting(self): values = [{'id': '001', 'name': 'receiver1'}, {'id': '002', 'name': 'receiver3'}, {'id': '003', 'name': 'receiver2'}] obj_ids = {'receiver1': 'id3', 'receiver2': 'id2', 'receiver3': 'id1'} for v in values: self._create_receiver(self.ctx, cluster_id=obj_ids[v['name']], **v) receivers = db_api.receiver_get_all(self.ctx, sort='name,cluster_id') self.assertEqual(3, len(receivers)) # Sorted by name (ascending) self.assertEqual('001', receivers[0].id) self.assertEqual('003', receivers[1].id) self.assertEqual('002', receivers[2].id) receivers = db_api.receiver_get_all(self.ctx, sort='cluster_id,name') self.assertEqual(3, len(receivers)) # Sorted by obj_id (ascending) self.assertEqual('002', receivers[0].id) self.assertEqual('003', receivers[1].id) self.assertEqual('001', receivers[2].id) receivers = db_api.receiver_get_all(self.ctx, sort='cluster_id:desc,name:desc') self.assertEqual(3, len(receivers)) # Sorted by obj_id (descending) self.assertEqual('001', receivers[0].id) self.assertEqual('003', receivers[1].id) self.assertEqual('002', receivers[2].id)
def test_receiver_get_all_used_sort_keys(self, mock_paginate): receiver_ids = ['receiver1', 'receiver2', 'receiver3'] for v in receiver_ids: self._create_receiver(self.ctx, id=v) sort_keys = consts.RECEIVER_SORT_KEYS db_api.receiver_get_all(self.ctx, sort=','.join(sort_keys)) args = mock_paginate.call_args[0] sort_keys.append('id') self.assertEqual(set(sort_keys), set(args[3]))
def test_receiver_get_all_used_sort_keys(self, mock_paginate): receiver_ids = ['receiver1', 'receiver2', 'receiver3'] for v in receiver_ids: self._create_receiver(self.ctx, id=v) sort_keys = consts.RECEIVER_SORT_KEYS db_api.receiver_get_all(self.ctx, sort=','.join(sort_keys)) args = mock_paginate.call_args[0] sort_keys.append('id') self.assertEqual(set(sort_keys), set(args[3]))
def test_receiver_get_all_with_filters(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') filters = {'name': ['receiver1', 'receiverx']} results = db_api.receiver_get_all(self.ctx, filters=filters) self.assertEqual(1, len(results)) self.assertEqual('receiver1', results[0]['name']) filters = {'name': 'receiver1'} results = db_api.receiver_get_all(self.ctx, filters=filters) self.assertEqual(1, len(results)) self.assertEqual('receiver1', results[0]['name'])
def test_receiver_get_all_with_filters(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') filters = {'name': ['receiver1', 'receiverx']} results = db_api.receiver_get_all(self.ctx, filters=filters) self.assertEqual(1, len(results)) self.assertEqual('receiver1', results[0]['name']) filters = {'name': 'receiver1'} results = db_api.receiver_get_all(self.ctx, filters=filters) self.assertEqual(1, len(results)) self.assertEqual('receiver1', results[0]['name'])
def test_receiver_get_all_with_project_safe(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') self.ctx.project = 'a-different-project' results = db_api.receiver_get_all(self.ctx, project_safe=False) self.assertEqual(2, len(results)) self.ctx.project = 'a-different-project' results = db_api.receiver_get_all(self.ctx) self.assertEqual(0, len(results)) results = db_api.receiver_get_all(self.ctx, project_safe=True) self.assertEqual(0, len(results))
def test_receiver_get_all_with_project_safe(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') self.ctx.project_id = 'a-different-project' results = db_api.receiver_get_all(self.ctx, project_safe=False) self.assertEqual(2, len(results)) self.ctx.project_id = 'a-different-project' results = db_api.receiver_get_all(self.ctx) self.assertEqual(0, len(results)) results = db_api.receiver_get_all(self.ctx, project_safe=True) self.assertEqual(0, len(results))
def test_receiver_get_all_with_empty_filters(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') filters = None results = db_api.receiver_get_all(self.ctx, filters=filters) self.assertEqual(2, len(results))
def test_receiver_get_all_with_empty_filters(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') filters = None results = db_api.receiver_get_all(self.ctx, filters=filters) self.assertEqual(2, len(results))
def test_receiver_get_all_with_admin_context(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') admin_ctx = utils.dummy_context(project='a-different-project', is_admin=True) results = db_api.receiver_get_all(admin_ctx, project_safe=True) self.assertEqual(2, len(results))
def test_receiver_get_all_with_admin_context(self): self._create_receiver(self.ctx, name='receiver1') self._create_receiver(self.ctx, name='receiver2') admin_ctx = utils.dummy_context(project='a-different-project', is_admin=True) results = db_api.receiver_get_all(admin_ctx, project_safe=True) self.assertEqual(2, len(results))
def test_receiver_get_all_with_limit_marker(self): receiver_ids = ['receiver1', 'receiver2', 'receiver3'] for v in receiver_ids: self._create_receiver(self.ctx, id=v, created_at=tu.utcnow(True)) receivers = db_api.receiver_get_all(self.ctx, limit=1) self.assertEqual(1, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, limit=2) self.assertEqual(2, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, limit=5) self.assertEqual(3, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, marker='receiver1') self.assertEqual(2, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, marker='receiver2') self.assertEqual(1, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, marker='receiver3') self.assertEqual(0, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, limit=1, marker='receiver1') self.assertEqual(1, len(receivers))
def test_receiver_get_all_with_limit_marker(self): receiver_ids = ['receiver1', 'receiver2', 'receiver3'] for v in receiver_ids: self._create_receiver(self.ctx, id=v, created_at=tu.utcnow(True)) receivers = db_api.receiver_get_all(self.ctx, limit=1) self.assertEqual(1, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, limit=2) self.assertEqual(2, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, limit=5) self.assertEqual(3, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, marker='receiver1') self.assertEqual(2, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, marker='receiver2') self.assertEqual(1, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, marker='receiver3') self.assertEqual(0, len(receivers)) receivers = db_api.receiver_get_all(self.ctx, limit=1, marker='receiver1') self.assertEqual(1, len(receivers))
def test_receiver_get_all(self): values = [{'name': 'receiver1'}, {'name': 'receiver2'}, {'name': 'receiver3'}] [self._create_receiver(self.ctx, **v) for v in values] receivers = db_api.receiver_get_all(self.ctx) self.assertEqual(3, len(receivers)) names = [receiver.name for receiver in receivers] for val in values: self.assertIn(val['name'], names)
def test_receiver_get_all(self): values = [{'name': 'receiver1'}, {'name': 'receiver2'}, {'name': 'receiver3'}] [self._create_receiver(self.ctx, **v) for v in values] receivers = db_api.receiver_get_all(self.ctx) self.assertEqual(3, len(receivers)) names = [receiver.name for receiver in receivers] for val in values: self.assertIn(val['name'], names)
def test_receiver_get_all_sorting(self): values = [{ 'id': '001', 'name': 'receiver1' }, { 'id': '002', 'name': 'receiver3' }, { 'id': '003', 'name': 'receiver2' }] obj_ids = {'receiver1': 'id3', 'receiver2': 'id2', 'receiver3': 'id1'} for v in values: self._create_receiver(self.ctx, cluster_id=obj_ids[v['name']], **v) receivers = db_api.receiver_get_all(self.ctx, sort='name,cluster_id') self.assertEqual(3, len(receivers)) # Sorted by name (ascending) self.assertEqual('001', receivers[0].id) self.assertEqual('003', receivers[1].id) self.assertEqual('002', receivers[2].id) receivers = db_api.receiver_get_all(self.ctx, sort='cluster_id,name') self.assertEqual(3, len(receivers)) # Sorted by obj_id (ascending) self.assertEqual('002', receivers[0].id) self.assertEqual('003', receivers[1].id) self.assertEqual('001', receivers[2].id) receivers = db_api.receiver_get_all(self.ctx, sort='cluster_id:desc,name:desc') self.assertEqual(3, len(receivers)) # Sorted by obj_id (descending) self.assertEqual('001', receivers[0].id) self.assertEqual('003', receivers[1].id) self.assertEqual('002', receivers[2].id)
def test_receiver_get_all_sorting_default(self): values = [{'id': '001', 'name': 'receiver1'}, {'id': '002', 'name': 'receiver2'}, {'id': '003', 'name': 'receiver3'}] obj_ids = {'receiver1': 'id3', 'receiver2': 'id2', 'receiver3': 'id1'} for v in values: self._create_receiver(self.ctx, cluster_id=obj_ids[v['name']], **v) receivers = db_api.receiver_get_all(self.ctx) self.assertEqual(3, len(receivers)) self.assertEqual(values[0]['id'], receivers[0].id) self.assertEqual(values[1]['id'], receivers[1].id) self.assertEqual(values[2]['id'], receivers[2].id)
def test_receiver_get_all_sorting_default(self): values = [{'id': '001', 'name': 'receiver1'}, {'id': '002', 'name': 'receiver2'}, {'id': '003', 'name': 'receiver3'}] obj_ids = {'receiver1': 'id3', 'receiver2': 'id2', 'receiver3': 'id1'} for v in values: self._create_receiver(self.ctx, cluster_id=obj_ids[v['name']], **v) receivers = db_api.receiver_get_all(self.ctx) self.assertEqual(3, len(receivers)) self.assertEqual(values[0]['id'], receivers[0].id) self.assertEqual(values[1]['id'], receivers[1].id) self.assertEqual(values[2]['id'], receivers[2].id)