import hashlib
import unittest

import kojihub
from koji import GenericError
from koji.util import adler32_constructor


class TestGetVerifyClass(unittest.TestCase):

    def test_get_verify_class_generic_error(self):
        with self.assertRaises(GenericError) as ex:
            kojihub.get_verify_class('not_a_real_value')
        self.assertEqual('Unsupported verify type: not_a_real_value', str(ex.exception))

    def test_get_verify_class_is_none(self):
        kojihub.get_verify_class(None) is None

    def test_get_verify_class_is_md5(self):
        kojihub.get_verify_class('md5') is hashlib.md5

    def test_get_verify_class_is_adler32(self):
        kojihub.get_verify_class('adler32') is adler32_constructor

    def test_get_verify_class_is_sha256(self):
        kojihub.get_verify_class('sha256') is hashlib.sha256