start writing some unit tests
[mevemon] / package / src / tests / test_validation.py
1 """" This module tests the functions in validation.py """
2 import unittest
3
4 import validation
5
6 INVALID_KEYS = [
7 "asdf", # too short
8 "12345678901234567890123456789012345678901234567890123456789012345", # too long
9 "asdfs#$7^(DGK", # invalid characters
10 ]
11
12 VALID_KEY = "1234567890123456789012345678901234567890123456789012345678901234"
13
14 INVALID_UIDS = ["12asd34",  # must be a number
15                 "", # can't be empty
16                 ]
17
18 VALID_UIDS = ["123456", "234523", "34", "2344566774756455645"]
19
20 class TestValidation(unittest.TestCase):
21     def test_validate_api_key(self):
22         # negative test
23         for key in INVALID_KEYS:
24             self.assertRaises(validation.ValidationError,
25                     validation.validate_api_key, key)
26         
27         # positive test    
28         validation.validate_api_key(VALID_KEY)
29         
30
31     def test_validate_uid(self):
32         # negative test
33         for key in INVALID_UIDS:
34             self.assertRaises(validation.ValidationError,
35                     validation.validate_uid, key)
36
37         # positive test
38         for uid in VALID_UIDS:
39             validation.validate_uid(uid)
40