diff options
Diffstat (limited to 'obnamlib/checksummer_tests.py')
-rw-r--r-- | obnamlib/checksummer_tests.py | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/obnamlib/checksummer_tests.py b/obnamlib/checksummer_tests.py new file mode 100644 index 00000000..10473740 --- /dev/null +++ b/obnamlib/checksummer_tests.py @@ -0,0 +1,48 @@ +# Copyright 2016 Lars Wirzenius +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. +# +# =*= License: GPL-3+ =*= + + +import unittest + +import obnamlib + + +class TestGetChecksummer(unittest.TestCase): + + def test_knows_some_algorithms(self): + self.assertEqual(type(obnamlib.checksum_algorithms), list) + self.assertNotEqual(obnamlib.checksum_algorithms, []) + + def test_raises_error_if_algorithm_is_unknown(self): + self.assertRaises( + obnamlib.ObnamError, obnamlib.get_checksum_algorithm, 'unknown') + + def test_returns_working_sha512(self): + summer = obnamlib.get_checksum_algorithm('sha512') + summer.update('hello, world') + self.assertEqual( + summer.hexdigest(), + '8710339dcb6814d0d9d2290ef422285c9322b7163951f9a0ca8f883d3305286f' + '44139aa374848e4174f5aada663027e4548637b6d19894aec4fb6c46a139fbf9') + + def test_every_algorithm_has_right_api(self): + for name in obnamlib.checksum_algorithms: + summer = obnamlib.get_checksum_algorithm(name) + summer.update('hello, world') + checksum = summer.hexdigest() + self.assertEqual(type(checksum), str) + self.assertNotEqual(checksum, '') |