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