tests/test_checksum.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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