def test_url_protocol_fix(): bad_url = 'flickr.com' tls_support_cache = {} result = CleanupFunctions.cleanup_url(bad_url, tls_support_cache) expected = "'https://flickr.com'" bad_http = 'neverssl.com' result_http = CleanupFunctions.cleanup_url(bad_http, tls_support_cache) expected_http = "'http://neverssl.com'" assert result == expected assert result_http == expected_http
def test_url_protocol_fix(): bad_url = "flickr.com" tls_support_cache = {} result = CleanupFunctions.cleanup_url(bad_url, tls_support_cache) expected = "'https://flickr.com'" bad_http = "neverssl.com" TlsTest.test_tls_supported = MagicMock(return_value=False) result_http = CleanupFunctions.cleanup_url(bad_http, tls_support_cache) expected_http = "'http://neverssl.com'" assert result == expected assert result_http == expected_http