class TestUrlFormatter(unittest.TestCase):
def setUp(self):
url_validator = mock.create_autospec(UrlValidator).return_value
self.url_formatter = UrlFormatter(url_validator)
@mock.patch.object(UrlFormatter, 'add_protocol_domain_separator')
def test_format_to_parsable_url(self, mock_add_protocol_domain_separator):
mock_url = "test.com/link/to/example_page.html"
mock_add_protocol_domain_separator.return_value = "//test.com/link/to/example_page.html"
result = self.url_formatter.format_to_parsable_url(mock_url)
self.assertEquals(result, "//test.com/link/to/example_page.html")
def test_add_protocol_domain_separator(self):
mock_url = "test.com/link/to/example_page.html"
self.url_formatter.url_validator.is_domain.return_value = True
result = self.url_formatter.add_protocol_domain_separator(mock_url)
self.assertEquals(result, "//test.com/link/to/example_page.html")
def __init__(self,
url_formatter=UrlFormatter(),
url_splitter=UrlSplitter()):
self.url_formatter = url_formatter
self.url_splitter = url_splitter
def setUp(self):
url_validator = mock.create_autospec(UrlValidator).return_value
self.url_formatter = UrlFormatter(url_validator)
