26 lines
740 B
Python
26 lines
740 B
Python
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):
|
|
kojihub.get_verify_class('not_a_real_value')
|
|
|
|
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
|
|
|