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)