Provider web certificate will always be bytestring now, requests complains otherwise

1 files changed, 14 insertions, 9 deletions

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)