def test_link_test(mock_urlopen):
"""
Read from an open, 'mocked', URL.
"""
class MockFile:
def __init__(self, content):
self.content = content
def read(self):
return self.content
def close(self):
return
expected = '<!doctype html><title>T</title>'
mock_urlopen.return_value = MockFile(expected)
assert link_test('http://www/google.com') == expected
def test_link_test(mock_urlopen):
"""
Read from an open, 'mocked', URL.
"""
class MockFile(object):
def __init__(self, content):
self.content = content
def read(self):
return self.content
def close(self):
return
expected = '<!doctype html><title>T</title>'
mock_urlopen.return_value = MockFile(expected)
assert link_test('http://www/google.com') == expected
def test_link_test_on_urlerror(mock_link_test):
"""
If `link_test` internally raises `URLError`, ensure it
returns `None`
"""
link_test('') is None
def test_link_test_on_urlerror(mock_link_test):
"""
If `link_test` internally raises `URLError`, ensure it
returns `None`
"""
link_test('') is None
def test_link_test():
assert b"# SunPy Sample Data" in link_test('http://data.sunpy.org/sunpy/README.md')
