From 031ebd58fd97bebae81e4e17cd7c4a4ed5a493d0 Mon Sep 17 00:00:00 2001 From: Bruno Wagner Date: Wed, 10 Jun 2015 08:56:39 -0300 Subject: Provider web certificate will always be bytestring now, requests complains otherwise --- service/test/unit/bitmask_libraries/test_certs.py | 23 ++++++++++++++--------- 1 file changed, 14 insertions(+), 9 deletions(-) (limited to 'service/test/unit/bitmask_libraries/test_certs.py') diff --git a/service/test/unit/bitmask_libraries/test_certs.py b/service/test/unit/bitmask_libraries/test_certs.py index 150a1f14..f1e643c4 100644 --- a/service/test/unit/bitmask_libraries/test_certs.py +++ b/service/test/unit/bitmask_libraries/test_certs.py @@ -6,22 +6,27 @@ from mock import MagicMock, patch class CertsTest(unittest.TestCase): + def setUp(self): + config = MagicMock(leap_home='/some/leap/home') + self.provider = MagicMock(server_name=u'test.leap.net', config=config) + def test_set_cert_and_fingerprint_sets_cert(self): LeapCertificate.set_cert_and_fingerprint('some cert', None) - self.assertIsNone(LeapCertificate.LEAP_FINGERPRINT) - self.assertEqual('some cert', LeapCertificate.LEAP_CERT) + certs = LeapCertificate(self.provider) + + self.assertIsNone(certs.LEAP_FINGERPRINT) + self.assertEqual('some cert', certs.provider_web_cert) def test_set_cert_and_fingerprint_sets_fingerprint(self): LeapCertificate.set_cert_and_fingerprint(None, 'fingerprint') - self.assertEqual('fingerprint', LeapCertificate.LEAP_FINGERPRINT) - self.assertFalse(LeapCertificate.LEAP_CERT) + certs = LeapCertificate(self.provider) - def test_api_ca_bundle(self): - config = MagicMock(leap_home='/some/leap/home') - provider = MagicMock(server_name=u'test.leap.net', config=config) + self.assertEqual('fingerprint', LeapCertificate.LEAP_FINGERPRINT) + self.assertFalse(certs.provider_web_cert) - cert = LeapCertificate(provider).api_ca_bundle + def test_provider_api_cert(self): + certs = LeapCertificate(self.provider).provider_api_cert - self.assertEqual('/some/leap/home/providers/test.leap.net/keys/client/api.pem', cert) + self.assertEqual('/some/leap/home/providers/test.leap.net/keys/client/api.pem', certs) -- cgit v1.2.3