From 3698dca8e350febf21f830d61630e12d40d51392 Mon Sep 17 00:00:00 2001
From: Rob Crittenden <rcritten@redhat.com>
Date: Fri, 30 Apr 2010 12:02:28 -0400
Subject: Add test cases for AccessTime param and fix some problems in
 AccessTime

---
 ipalib/parameters.py                 | 14 +++++++++----
 tests/test_ipalib/test_parameters.py | 40 ++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+), 4 deletions(-)

diff --git a/ipalib/parameters.py b/ipalib/parameters.py
index 606a57483..612ec3297 100644
--- a/ipalib/parameters.py
+++ b/ipalib/parameters.py
@@ -1451,7 +1451,7 @@ class AccessTime(Str):
             for v in values:
                 check_func(v)
             if len(values) == 2:
-                if int(v[0]) > int(v[1]):
+                if int(values[0]) > int(values[1]):
                     raise ValueError('invalid time range')
 
     def _check_W_spec(self, ts, index):
@@ -1477,7 +1477,7 @@ class AccessTime(Str):
             index += 1
             self._check_interval(ts[index], self._check_month_num)
             month_num = int(ts[index])
-            index = self._check_M_spec(ts, index + 1, month_num)
+            index = self._check_M_spec(ts, index + 1)
         elif ts[index] == 'week':
             self._check_interval(ts[index + 1], self._check_woty)
             index = self._check_W_spec(ts, index + 2)
@@ -1489,6 +1489,7 @@ class AccessTime(Str):
         return index
 
     def _check_generalized(self, t):
+        assert type(t) is unicode
         if len(t) not in (10, 12, 14):
             raise ValueError('incomplete generalized time')
         if not t.isnumeric():
@@ -1510,6 +1511,8 @@ class AccessTime(Str):
     def _check(self, time):
         ts = time.split()
         if ts[0] == 'absolute':
+            if len(ts) != 4:
+                raise ValueError('invalid format, must be \'absolute generalizedTime ~ generalizedTime\'')
             self._check_generalized(ts[1])
             if ts[2] != '~':
                 raise ValueError('invalid time range separator')
@@ -1517,12 +1520,15 @@ class AccessTime(Str):
             if int(ts[1]) >= int(ts[3]):
                 raise ValueError('invalid time range')
         elif ts[0] == 'periodic':
+            index = None
             if ts[1] == 'yearly':
                 index = self._check_Y_spec(ts, 2)
             elif ts[1] == 'monthly':
                 index = self._check_M_spec(ts, 2)
             elif ts[1] == 'daily':
                 index = 1
+            if index is None:
+                raise ValueError('period must be yearly, monthy or daily, got \'%s\'' % ts[1])
             self._check_interval(ts[index + 1], self._check_HHMM)
         else:
             raise ValueError('time neither absolute or periodic')
@@ -1531,10 +1537,10 @@ class AccessTime(Str):
         try:
             self._check(value)
         except ValueError, e:
-            raise ValidationError(name=self.cli_name, error=e.message)
+            raise ValidationError(name=self.cli_name, error=e.args[0])
         except IndexError:
             raise ValidationError(
-                name=self.cli_name, errors='incomplete time value'
+                name=self.cli_name, error='incomplete time value'
             )
         return None
 
diff --git a/tests/test_ipalib/test_parameters.py b/tests/test_ipalib/test_parameters.py
index 038941b80..e14538c50 100644
--- a/tests/test_ipalib/test_parameters.py
+++ b/tests/test_ipalib/test_parameters.py
@@ -1348,6 +1348,46 @@ class test_List(ClassChecker):
         # the output w/o skipspace is ['a',' "b','c"',' d']
         assert len(n) is 4
 
+class test_AccessTime(ClassChecker):
+    """
+    Test the `ipalib.parameters.AccessTime` class.
+    """
+    _cls = parameters.AccessTime
+
+    def test_init(self):
+        """
+        Test the `ipalib.parameters.AccessTime.__init__` method.
+        """
+        # Test with no kwargs:
+        o = self.cls('my_time')
+        assert o.type is unicode
+        assert isinstance(o, parameters.AccessTime)
+        assert o.multivalue is False
+        translation = u'length=%(length)r'
+        dummy = dummy_ugettext(translation)
+        assert dummy.translation is translation
+        rule = o._rule_required
+
+        # Check some good rules
+        for value in (u'absolute 201012161032 ~ 201012161033',
+                      u'periodic monthly week 2 day Sat,Sun 0900-1300',
+                      u'periodic yearly month 4 day 1-31 0800-1400',
+                      u'periodic daily 0800-1400',
+            ):
+            assert rule(dummy, value) is None
+            assert dummy.called() is False
+
+        # FIXME, weekly is not implemented in AccessTime
+#                      u'periodic weekly day 8 0800-1400',
+
+        # And some bad ones
+        for value in (u'absolute 201012161032 - 201012161033',
+                      u'absolute 201012161032 ~',
+                      u'periodic monthly day Sat,Sun 0900-1300',
+                      u'periodical yearly month 4 day 1-31 0800-1400',
+            ):
+            e = raises(ValidationError, o._rule_required, None, value)
+
 def test_create_param():
     """
     Test the `ipalib.parameters.create_param` function.
-- 
cgit