1 files changed, 102 insertions, 0 deletions
diff --git a/ipatests/test_webui/task_range.py b/ipatests/test_webui/task_range.py
new file mode 100644
index 00000000..4775078e
--- /dev/null
+++ b/ipatests/test_webui/task_range.py
@@ -0,0 +1,102 @@
+# Authors:
+#   Petr Vobornik <pvoborni@redhat.com>
+#
+# Copyright (C) 2013  Red Hat
+# see file 'COPYING' for use and warranty information
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+"""
+Range tasks
+"""
+
+from ipatests.test_webui.ui_driver import UI_driver
+
+
+class range_tasks(UI_driver):
+
+    def get_shifts(self, idranges=None):
+
+        if not idranges:
+            result = self.execute_api_from_ui('idrange_find', [], {})
+            idranges = result['result']['result']
+
+        id_shift = 0
+        rid_shift = 0
+
+        for idrange in idranges:
+            size = int(idrange['ipaidrangesize'][0])
+            base_id = int(idrange['ipabaseid'][0])
+
+            id_end = base_id + size
+            rid_end = 0
+
+            if 'ipabaserid' in idrange:
+                base_rid = int(idrange['ipabaserid'][0])
+                rid_end = base_rid + size
+
+                if 'ipasecondarybaserid' in idrange:
+                    secondary_base_rid = int(idrange['ipasecondarybaserid'][0])
+                    rid_end = max(base_rid, secondary_base_rid) + size
+
+            if id_shift < id_end:
+                id_shift = id_end + 1000000
+
+            if rid_shift < rid_end:
+                rid_shift = rid_end + 1000000
+
+        self.id_shift = id_shift
+        self.rid_shift = rid_shift
+        self.sec_rid_shift = rid_shift + 1000
+        self.shift = 0
+
+    def get_sid(self):
+        result = self.execute_api_from_ui('trust_find', [], {})
+        trusts = result['result']['result']
+        sid = None
+        if trusts:
+            sid = trusts[0]['ipanttrusteddomainsid']
+        return sid
+
+    def get_data(self, pkey, size=50, add_data=None):
+
+        if not add_data:
+            add_data = self.get_add_data(pkey, size=size)
+
+        data = {
+            'pkey': pkey,
+            'add': add_data,
+            'mod': [
+                ('textbox', 'ipaidrangesize', str(size + 1)),
+            ],
+        }
+        return data
+
+    def get_add_data(self, pkey, range_type='ipa-local', size=50, shift=100, sid=None):
+
+        self.shift += shift
+        add = [
+            ('textbox', 'cn', pkey),
+            ('textbox', 'ipabaseid', str(self.id_shift + self.shift)),
+            ('textbox', 'ipaidrangesize', str(size)),
+            ('textbox', 'ipabaserid', str(self.rid_shift + self.shift)),
+            ('radio', 'iparangetype', range_type),
+        ]
+
+        if not sid:
+            add.append(('textbox', 'ipasecondarybaserid', str(self.sec_rid_shift + self.shift)))
+        if sid:
+            add.append(('textbox', 'ipanttrusteddomainsid', sid))
+
+        return add