start writing some unit tests

diff --git a/package/src/tests/__init__.py b/package/src/tests/__init__.py
new file mode 100644 (file)
index 0000000..e69de29

diff --git a/package/src/tests/test_util.py b/package/src/tests/test_util.py
new file mode 100644 (file)
index 0000000..24ca9ec
--- /dev/null
+++ b/package/src/tests/test_util.py
@@ -0,0 +1,14 @@
+""" this module tests the util.py module """
+import unittest
+
+import util
+
+NUMBERS = { 12345:'12,345', 12345.23:'12,345.23', 1234:'1,234'}
+
+class TestUtil(unittest.TestCase):
+    def test_comma(self):
+        for number in NUMBERS.keys():
+            self.assertEqual(util.comma(number), NUMBERS[number])
+
+
+

diff --git a/package/src/tests/test_validation.py b/package/src/tests/test_validation.py
new file mode 100644 (file)
index 0000000..0cf941c
--- /dev/null
+++ b/package/src/tests/test_validation.py
@@ -0,0 +1,40 @@
+"""" This module tests the functions in validation.py """
+import unittest
+
+import validation
+
+INVALID_KEYS = [
+"asdf", # too short
+"12345678901234567890123456789012345678901234567890123456789012345", # too long
+"asdfs#$7^(DGK", # invalid characters
+]
+
+VALID_KEY = "1234567890123456789012345678901234567890123456789012345678901234"
+
+INVALID_UIDS = ["12asd34",  # must be a number
+                "", # can't be empty
+                ]
+
+VALID_UIDS = ["123456", "234523", "34", "2344566774756455645"]
+
+class TestValidation(unittest.TestCase):
+    def test_validate_api_key(self):
+        # negative test
+        for key in INVALID_KEYS:
+            self.assertRaises(validation.ValidationError,
+                    validation.validate_api_key, key)
+        
+        # positive test    
+        validation.validate_api_key(VALID_KEY)
+        
+
+    def test_validate_uid(self):
+        # negative test
+        for key in INVALID_UIDS:
+            self.assertRaises(validation.ValidationError,
+                    validation.validate_uid, key)
+
+        # positive test
+        for uid in VALID_UIDS:
+            validation.validate_uid(uid)
+