from twisted.trial import unittest
from mock import Mock

from iro.validate import vBool, vInteger, vHash, validate
from iro.error import ValidateException

class testValidators(unittest.TestCase):
    '''test for simple validators'''

    def testBool(self):
        self.assertEqual(vBool(True,None),True)
        self.assertEqual(vBool(1,None),True)
        self.assertEqual(vBool("true",None),True)
        self.assertEqual(vBool("True",None),True)
        self.assertEqual(vBool("TRUE",None),True)

        self.assertEqual(vBool(False,None),False)
        self.assertEqual(vBool(0,None),False)
        self.assertEqual(vBool("false",None),False)
        self.assertEqual(vBool("False",None),False)
        self.assertEqual(vBool("FALSE",None),False)

        e = self.assertRaises(ValidateException, vBool, "TRue","test")
        self.assertEqual(e.msg,"test is not boolean")

    def testInteger(self):
        self.assertEqual(vInteger(123,None),123)
        self.assertEqual(vInteger("123",None),123)
        self.assertEqual(vInteger("-123",None),-123)

        self.assertRaises(ValidateException, vInteger, "a123",None)

    def testIntegerLimits(self):
        self.assertEqual(vInteger(10,None,maxv=10),10)
        self.assertRaises(ValidateException, vInteger, 11, None, maxv=10)

        self.assertEqual(vInteger(4,None,minv=4),4)
        self.assertRaises(ValidateException, vInteger, 3, None, minv=4)

        self.assertRaises(ValidateException, vInteger, -1, None, minv=0)
        self.assertRaises(ValidateException, vInteger, 1, None, maxv=0)

    def testIntegerNoneAllowed(self):
        self.assertEqual(vInteger(None,None,none_allowed=True),None)
        self.assertEqual(vInteger('',None,none_allowed=True),None)

        self.assertRaises(ValidateException, vInteger, "", None)
        self.assertRaises(ValidateException, vInteger, None, None)

    def testHash(self):
        self.assertEqual(vHash("0123456789abcdef",None),"0123456789abcdef")
        self.assertEqual(vHash("0123456789ABCDEF",None),"0123456789abcdef")
        self.assertEqual(vHash("F",None),"f")
        self.assertEqual(vHash("",None),'')

        self.assertRaises(ValidateException, vHash, "GHIJKL", None)
        
    def testHashLimits(self):
        self.assertEqual(vHash("F",None,minlength=1),"f")
        self.assertRaises(ValidateException, vHash, "", None, minlength=1)

        self.assertEqual(vHash("Fa",None,maxlength=2),"fa")
        self.assertRaises(ValidateException, vHash, "123", None, maxlength=1)

    
    def testValidate(self):
        f = Mock()
        f.return_value = "valid"
        @validate("t",f,True,1,2,3,4,k=5)
        def g(u=False, t="bla"):
            return t
        d = g(t="uhuhu")
        def r(t):
            f.called_once_with("uhuhu","t",1,2,3,4,k=5)
            self.assertEqual(t,"valid")
        d.addCallback(r)
        return d

    def testValidateMissingNeed(self):
        f = Mock()
        @validate("t",f,True,1,2,3,4,k=5)
        def g(u, t="buuh"):
            return t
        e = self.assertRaises(ValidateException, g, u="uhuhu", t = None)
        self.assertEqual(str(e),'700: t is nessasary')

    def testValidateMissingNeedNonExplicit(self):
        f = Mock()
        @validate("t",f,True,1,2,3,4,k=5)
        def g(u, **k):
            return k["t"]
        e = self.assertRaises(ValidateException, g, u="uhuhu")
        self.assertEqual(str(e),'700: t is nessasary')


    def testValidateMissingNeed2(self):
        f = Mock()
        f.return_value = "valid"
        @validate("t",f,True,1,2,3,4,k=5)
        def g(u, t="buuh"):
            return t

        d = g(True)

        def r(t):
            f.called_once_with("buuh","t",1,2,3,4,k=5)
            self.assertEqual(t,"valid")
        d.addCallback(r)
        return d

    def testvalidateNoNeed(self):
        f = Mock()
        f.return_value = "valid"
        @validate("t",f,False,1,2,3,4,k=5)
        def g(u, t="buuh"):
            return t
        d = g("uhu")
        def r(t):
            self.assertEqual(f.called,True)
            self.assertEqual(t,"valid")
        d.addCallback(r)
        return d

    def testvalidateNoNeed2(self):
        f = Mock()
        f.return_value = "valid"
        @validate("t",f,False,1,2,3,4,k=5)
        def g(u, **k):
            return k["t"]
        d = g("uhu")
        def r(t):
            self.assertEqual(f.called,False)
            self.assertEqual(t,None)
        d.addCallback(r)
        return d