Add unit test

Tests ability to disable error message interpolation. Added parameter to
turn on/off error message interpolation.

Author: rubber-side-down

flask_validator/validator.py

@@ -19,7 +19,8 @@ class FlaskValidator(object):
     throw_exception = False
     message = None
 
-    def __init__(self, field, allow_null, throw_exception, message, parent):
+    def __init__(self, field, allow_null, throw_exception, message,
+                 interpolate_message, parent):
         """ Initialize a Validator object.
 
         :type throw_exception: Throw a ValidateError exception
@@ -30,6 +31,7 @@ def __init__(self, field, allow_null, throw_exception, message, parent):
         self.allow_null = allow_null
         self.throw_exception = throw_exception
         self.message = message
+        self.interpolate_message = interpolate_message
         self.__create_event()
 
     def __validate(self, target, value, oldvalue, initiator):
@@ -93,14 +95,16 @@ def start(self):
 class Validator(FlaskValidator):
     """ Main validotr class """
 
-    def __init__(self, field, allow_null=True, throw_exception=False, message=None):
+    def __init__(self, field, allow_null=True, throw_exception=False,
+                 message=None, interpolate_message=True):
         """
         Validator Interface initialization
 
         :param field:  Flask Column to validate
         """
 
-        FlaskValidator.__init__(self, field, allow_null, throw_exception, message, self)
+        FlaskValidator.__init__(self, field, allow_null, throw_exception,
+                                message, interpolate_message, self)
 
     @abc.abstractmethod
     def check_value(self, value):

test/test_message.py

@@ -0,0 +1,74 @@
+""" FlaskValidator tests """
+
+import unittest
+
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+
+from flask_validator import Validator
+
+
+class MessageTest(unittest.TestCase):
+    def setUp(self):
+        app = Flask(__name__)
+        app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
+        app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+        db = SQLAlchemy(app)
+
+        class DummyModel(db.Model):
+            """ SQLAlchemy Dummy Object """
+
+            id = db.Column(db.Integer, primary_key=True)
+            integer = db.Column(db.Integer)
+
+        db.create_all()
+
+        self.DummyModel = DummyModel
+        self.define_validators()
+
+        self.dummy = DummyModel()
+
+        self.app = app
+        self.db = db
+
+    def simple_validate(self, field, value):
+        """
+        Simple Validation
+
+        """
+        setattr(self.dummy, field, value)
+
+    def define_validators(self):
+        """
+        Define Validators
+
+        """
+        class FailValidator(Validator):
+            def check_value(self, value):
+                return False
+
+        FailValidator(self.DummyModel.integer, throw_exception=True,
+                      message="{do} {not} {interpolate} {this}",
+                      interpolate_message=False)
+
+    def test_disable_message_interpolation(self):
+        """
+        Testing FlaskValidator.interpolate_message
+
+        """
+        with self.assertRaises(Exception) as context:
+            self.simple_validate('integer', 1)
+
+        self.assertEqual("{do} {not} {interpolate} {this}",
+                         str(context.exception))
+
+
+def suite():
+    """ Test Suite """
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(MessageTest))
+    return suite
+
+
+if __name__ == '__main__':
+    unittest.main(defaultTest='suite')