From 00f5de52a8fb944c968213501ccdf1bbc9e751d0 Mon Sep 17 00:00:00 2001 From: Hsiaoming Yang Date: Fri, 31 May 2013 17:22:54 +0800 Subject: add test case for is_secure_transport --- tests/oauth2/rfc6749/test_utils.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'tests/oauth2/rfc6749/test_utils.py') diff --git a/tests/oauth2/rfc6749/test_utils.py b/tests/oauth2/rfc6749/test_utils.py index 9d25229..6e713a7 100644 --- a/tests/oauth2/rfc6749/test_utils.py +++ b/tests/oauth2/rfc6749/test_utils.py @@ -1,7 +1,9 @@ from __future__ import absolute_import, unicode_literals +import os from ...unittest import TestCase from oauthlib.oauth2.rfc6749.utils import escape, host_from_uri +from oauthlib.oauth2.rfc6749.utils import is_secure_transport class UtilsTests(TestCase): @@ -21,3 +23,15 @@ class UtilsTests(TestCase): self.assertEqual(host_from_uri('https://a.b.com:8080'), ('a.b.com', '8080')) self.assertEqual(host_from_uri('http://www.example.com'), ('www.example.com', '80')) self.assertEqual(host_from_uri('https://www.example.com'), ('www.example.com', '443')) + + def test_is_secure_transport(self): + """Test check secure uri.""" + if 'DEBUG' in os.environ: + del os.environ['DEBUG'] + + self.assertTrue(is_secure_transport('https://example.com')) + self.assertFalse(is_secure_transport('http://example.com')) + + os.environ['DEBUG'] = '1' + self.assertTrue(is_secure_transport('http://example.com')) + del os.environ['DEBUG'] -- cgit v1.2.1