added g_octave.checksum module, with tests

1 files changed, 40 insertions, 0 deletions

diff --git a/tests/test_checksum.py b/tests/test_checksum.py
new file mode 100644
index 0000000..66cd96b
--- /dev/null
+++ b/tests/test_checksum.py
@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+    test_checksum.py
+    ~~~~~~~~~~~~~~~~
+    
+    test suite for the *g_octave.checksum* module
+    
+    :copyright: (c) 2010 by Rafael Goncalves Martins
+    :license: GPL-2, see LICENSE for more details.
+"""
+
+import os
+import tempfile
+import unittest
+
+from g_octave import checksum
+
+
+class TestChecksum(unittest.TestCase):
+    
+    def setUp(self):
+        self._tempfile = tempfile.mkstemp()[1]
+        with open(self._tempfile, 'w') as fp:
+            # SHA1 checksum: 8aa49f56d049193b183cb2918f8fb59e0caf1283
+            fp.write("I'm the walrus\n")
+    
+    def test_checksum(self):
+        my_checksum = checksum.sha1_compute(self._tempfile)
+        self.assertEqual(my_checksum, '8aa49f56d049193b183cb2918f8fb59e0caf1283')
+        self.assertTrue(checksum.sha1_check(self._tempfile, my_checksum))
+
+    def tearDown(self):
+        os.unlink(self._tempfile)
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(TestChecksum('test_checksum'))
+    return suite