1 files changed, 96 insertions, 0 deletions
diff --git a/keystone-moon/keystone/tests/unit/filtering.py b/keystone-moon/keystone/tests/unit/filtering.py
new file mode 100644
index 00000000..1a31a23f
--- /dev/null
+++ b/keystone-moon/keystone/tests/unit/filtering.py
@@ -0,0 +1,96 @@
+# Copyright 2013 OpenStack Foundation
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+import uuid
+
+from oslo_config import cfg
+
+
+CONF = cfg.CONF
+
+
+class FilterTests(object):
+
+    # Provide support for checking if a batch of list items all
+    # exist within a contiguous range in a total list
+    def _match_with_list(self, this_batch, total_list,
+                         batch_size=None,
+                         list_start=None, list_end=None):
+        if batch_size is None:
+            batch_size = len(this_batch)
+        if list_start is None:
+            list_start = 0
+        if list_end is None:
+            list_end = len(total_list)
+        for batch_item in range(0, batch_size):
+            found = False
+            for list_item in range(list_start, list_end):
+                if this_batch[batch_item]['id'] == total_list[list_item]['id']:
+                    found = True
+            self.assertTrue(found)
+
+    def _create_entity(self, entity_type):
+        f = getattr(self.identity_api, 'create_%s' % entity_type, None)
+        if f is None:
+            f = getattr(self.assignment_api, 'create_%s' % entity_type)
+        return f
+
+    def _delete_entity(self, entity_type):
+        f = getattr(self.identity_api, 'delete_%s' % entity_type, None)
+        if f is None:
+            f = getattr(self.assignment_api, 'delete_%s' % entity_type)
+        return f
+
+    def _list_entities(self, entity_type):
+        f = getattr(self.identity_api, 'list_%ss' % entity_type, None)
+        if f is None:
+            f = getattr(self.assignment_api, 'list_%ss' % entity_type)
+        return f
+
+    def _create_one_entity(self, entity_type, domain_id, name):
+        new_entity = {'name': name,
+                      'domain_id': domain_id}
+        if entity_type in ['user', 'group']:
+            # The manager layer creates the ID for users and groups
+            new_entity = self._create_entity(entity_type)(new_entity)
+        else:
+            new_entity['id'] = '0000' + uuid.uuid4().hex
+            self._create_entity(entity_type)(new_entity['id'], new_entity)
+        return new_entity
+
+    def _create_test_data(self, entity_type, number, domain_id=None,
+                          name_dict=None):
+        """Create entity test data
+
+        :param entity_type: type of entity to create, e.g. 'user', group' etc.
+        :param number: number of entities to create,
+        :param domain_id: if not defined, all users will be created in the
+                          default domain.
+        :param name_dict: optional dict containing entity number and name pairs
+
+        """
+        entity_list = []
+        if domain_id is None:
+            domain_id = CONF.identity.default_domain_id
+        name_dict = name_dict or {}
+        for x in range(number):
+            # If this index has a name defined in the name_dict, then use it
+            name = name_dict.get(x, uuid.uuid4().hex)
+            new_entity = self._create_one_entity(entity_type, domain_id, name)
+            entity_list.append(new_entity)
+        return entity_list
+
+    def _delete_test_data(self, entity_type, entity_list):
+        for entity in entity_list:
+            self._delete_entity(entity_type)(entity['id'])