starred/certbot-certbot-1
1
0
Fork 0
mirror of https://github.com/certbot/certbot.git synced 2026-08-02 19:42:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use KEY in DVSNI tests

Browse Source
This commit is contained in:
Jakub Warmuz
2015-11-07 14:24:43 +00:00
parent dd92e95290
commit c805ebc2bf
1 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acme/acme/challenges_test.py
+7 -10
View File
@@ -571,8 +571,6 @@ class ProofOfPossessionResponseTest(unittest.TestCase):
class DNSTest(unittest.TestCase): class DNSTest(unittest.TestCase):
def setUp(self): def setUp(self):
self.account_key = jose.JWKRSA.load(
test_util.load_vector('rsa512_key.pem'))
from acme.challenges import DNS from acme.challenges import DNS
self.msg = DNS(token=jose.b64decode( self.msg = DNS(token=jose.b64decode(
b'evaGxfADs6pSRb2LAv9IZf17Dt3juxGJ-PCt92wr-oA')) b'evaGxfADs6pSRb2LAv9IZf17Dt3juxGJ-PCt92wr-oA'))
@@ -594,34 +592,33 @@ class DNSTest(unittest.TestCase):
def test_gen_check_validation(self): def test_gen_check_validation(self):
self.assertTrue(self.msg.check_validation( self.assertTrue(self.msg.check_validation(
self.msg.gen_validation(self.account_key), self.msg.gen_validation(KEY), KEY.public_key()))
self.account_key.public_key()))
def test_gen_check_validation_wrong_key(self): def test_gen_check_validation_wrong_key(self):
key2 = jose.JWKRSA.load(test_util.load_vector('rsa1024_key.pem')) key2 = jose.JWKRSA.load(test_util.load_vector('rsa1024_key.pem'))
self.assertFalse(self.msg.check_validation( self.assertFalse(self.msg.check_validation(
self.msg.gen_validation(self.account_key), key2.public_key())) self.msg.gen_validation(KEY), key2.public_key()))
def test_check_validation_wrong_payload(self): def test_check_validation_wrong_payload(self):
validations = tuple( validations = tuple(
jose.JWS.sign(payload=payload, alg=jose.RS256, key=self.account_key) jose.JWS.sign(payload=payload, alg=jose.RS256, key=KEY)
for payload in (b'', b'{}') for payload in (b'', b'{}')
) )
for validation in validations: for validation in validations:
self.assertFalse(self.msg.check_validation( self.assertFalse(self.msg.check_validation(
validation, self.account_key.public_key())) validation, KEY.public_key()))
def test_check_validation_wrong_fields(self): def test_check_validation_wrong_fields(self):
bad_validation = jose.JWS.sign( bad_validation = jose.JWS.sign(
payload=self.msg.update(token=b'x' * 20).json_dumps().encode('utf-8'), payload=self.msg.update(token=b'x' * 20).json_dumps().encode('utf-8'),
alg=jose.RS256, key=self.account_key) alg=jose.RS256, key=KEY)
self.assertFalse(self.msg.check_validation( self.assertFalse(self.msg.check_validation(
bad_validation, self.account_key.public_key())) bad_validation, KEY.public_key()))
def test_gen_response(self): def test_gen_response(self):
with mock.patch('acme.challenges.DNS.gen_validation') as mock_gen: with mock.patch('acme.challenges.DNS.gen_validation') as mock_gen:
mock_gen.return_value = mock.sentinel.validation mock_gen.return_value = mock.sentinel.validation
response = self.msg.gen_response(self.account_key) response = self.msg.gen_response(KEY)
from acme.challenges import DNSResponse from acme.challenges import DNSResponse
self.assertTrue(isinstance(response, DNSResponse)) self.assertTrue(isinstance(response, DNSResponse))
self.assertEqual(response.validation, mock.sentinel.validation) self.assertEqual(response.validation, mock.sentinel.validation)