summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorkali <kali@leap.se>2012-12-07 02:22:33 +0900
committerkali <kali@leap.se>2012-12-07 02:22:33 +0900
commit4c2f68b1158f3840f33a38a81a5fc03495d28466 (patch)
treefec0a56c64f1120aa064d824bc2427aa8ca91899
parente7dbf89f31711271e61f653e1cc7fb2c2b57cc6e (diff)
pep8
-rw-r--r--src/leap/base/tests/test_auth.py13
-rw-r--r--src/leap/base/tests/test_checks.py8
2 files changed, 11 insertions, 10 deletions
diff --git a/src/leap/base/tests/test_auth.py b/src/leap/base/tests/test_auth.py
index a6f2ceb9..17b84b52 100644
--- a/src/leap/base/tests/test_auth.py
+++ b/src/leap/base/tests/test_auth.py
@@ -1,14 +1,15 @@
from BaseHTTPServer import BaseHTTPRequestHandler
+import urlparse
try:
import unittest2 as unittest
except ImportError:
import unittest
import requests
-from mock import Mock
+#from mock import Mock
from leap.base import auth
-from leap.base import exceptions
+#from leap.base import exceptions
from leap.eip.tests.test_checks import NoLogRequestHandler
from leap.testing.basetest import BaseLeapTest
from leap.testing.https_server import BaseHTTPSServerTestCase
@@ -20,8 +21,7 @@ class LeapSRPRegisterTests(BaseHTTPSServerTestCase, BaseLeapTest):
class request_handler(NoLogRequestHandler, BaseHTTPRequestHandler):
responses = {
- '/': ['OK', ''],
- }
+ '/': ['OK', '']}
def do_GET(self):
path = urlparse.urlparse(self.path)
@@ -53,5 +53,6 @@ class LeapSRPRegisterTests(BaseHTTPSServerTestCase, BaseLeapTest):
srp_auth = auth.LeapSRPRegister(provider=SERVER, verify=False)
self.assertIsInstance(srp_auth.session, requests.sessions.Session)
- self.assertEqual(srp_auth.get_registration_uri(),
- "https://localhost:8443/1/users.json")
+ self.assertEqual(
+ srp_auth.get_registration_uri(),
+ "https://localhost:8443/1/users.json")
diff --git a/src/leap/base/tests/test_checks.py b/src/leap/base/tests/test_checks.py
index 272e7053..7a694f89 100644
--- a/src/leap/base/tests/test_checks.py
+++ b/src/leap/base/tests/test_checks.py
@@ -119,16 +119,16 @@ class LeapNetworkCheckTest(BaseLeapTest):
checker.check_internet_connection()
with patch.object(requests, "get") as mocked_get:
- mocked_get.side_effect = \
- requests.ConnectionError("[Errno 113] No route to host")
+ mocked_get.side_effect = requests.ConnectionError(
+ "[Errno 113] No route to host")
with self.assertRaises(exceptions.NoInternetConnection):
with patch.object(checker, "ping_gateway") as mock_ping:
mock_ping.return_value = True
checker.check_internet_connection()
with patch.object(requests, "get") as mocked_get:
- mocked_get.side_effect = \
- requests.ConnectionError("[Errno 113] No route to host")
+ mocked_get.side_effect = requests.ConnectionError(
+ "[Errno 113] No route to host")
with self.assertRaises(exceptions.NoInternetConnection):
with patch.object(checker, "ping_gateway") as mock_ping:
mock_ping.side_effect = exceptions.NoConnectionToGateway