def test_get_digest_algorithm_specified(self): ''' Getting a specific digest algorithm should yield that algorithm if it exists. ''' self.assertEqual(utils._get_digest_algorithm('md5'), hashlib.md5) self.assertEqual(utils._get_digest_algorithm('MD5'), hashlib.md5) with self.assertRaises(exceptions.AlgorithmNotSupported): utils._get_digest_algorithm('abcdef')
def test_get_digest_algorithm_none(self): ''' Get the digest algorithm when no valid ones are specified. An error should be raised. ''' with self.assertRaises(exceptions.AlgorithmNotSupported): with mock.patch.object(settings, 'ALLOWED_DIGEST_ALGORITHMS', ['ABCDE']): utils._get_digest_algorithm() with self.assertRaises(exceptions.AlgorithmNotSupported): with mock.patch.object(settings, 'ALLOWED_DIGEST_ALGORITHMS', []): utils._get_digest_algorithm()
def test_get_digest_algorithm(self): ''' Getting a digest algorithm should yield the first one from the possible options. ''' self.assertEqual(utils._get_digest_algorithm(), hashlib.sha256) self.assertEqual(utils._get_digest_algorithm(), hashlib.sha256) with mock.patch.object(settings, 'ALLOWED_DIGEST_ALGORITHMS', ['ABCDEF', 'SHA256']): self.assertEqual(utils._get_digest_algorithm(), hashlib.sha256) with mock.patch.object(settings, 'ALLOWED_DIGEST_ALGORITHMS', ['MD5']): self.assertEqual(utils._get_digest_algorithm(), hashlib.md5)