Adds option to restrict certificate expiration dates to weekdays. (#453)

* Adding ability to restrict certificate creation to weekdays.

* Ensuring that we test for weekends.

lemur/tests/test_validators.py

@@ -1,29 +1,32 @@
-from marshmallow.exceptions import ValidationError
+import pytest
+from datetime import datetime
 from .vectors import PRIVATE_KEY_STR
+from marshmallow.exceptions import ValidationError
 
 
-def test_private_key():
+def test_private_key(session):
     from lemur.common.validators import private_key
-    try:
-        private_key(PRIVATE_KEY_STR)
-        assert True
-    except ValidationError:
-        assert False, "failed to validate private key as a bytes object"
 
+    private_key(PRIVATE_KEY_STR)
+    private_key(PRIVATE_KEY_STR.decode('utf-8'))
 
-def test_private_key_str_object():
-    from lemur.common.validators import private_key
-    try:
-        private_key(PRIVATE_KEY_STR.decode('utf-8'))
-        assert True
-    except ValidationError:
-        assert False, "failed to validate private key as a str object"
-
-
-def test_private_key_invalid():
-    from lemur.common.validators import private_key
-    try:
+    with pytest.raises(ValidationError):
         private_key('invalid_private_key')
-        assert False, "invalid private key should have raised an exception"
-    except ValidationError:
-        assert True
+
+
+def test_dates(session):
+    from lemur.common.validators import dates
+
+    dates(dict(validity_start=datetime(2016, 1, 1), validity_end=datetime(2016, 1, 5)))
+
+    with pytest.raises(ValidationError):
+        dates(dict(validity_start=datetime(2016, 1, 1)))
+
+    with pytest.raises(ValidationError):
+        dates(dict(validity_end=datetime(2016, 1, 1)))
+
+    with pytest.raises(ValidationError):
+        dates(dict(validity_start=datetime(2016, 1, 5), validity_end=datetime(2016, 1, 1)))
+
+    with pytest.raises(ValidationError):
+        dates(dict(validity_start=datetime(2016, 1, 1), validity_end=datetime(2016, 1, 10)))