def test_perform_request_good(self):
with mocks.PatchUrlOpen(Response(1), 0, None) as opener:
lines = freedb.perform_request(
"http://freedb", "query", "hello", "proto"
)
self.assertEqual(lines,
['', '# Comment 1 ' + a_tiddle, '',
'# Comment 2', '', 'name=value'])
def test_perform_request_503(self):
expected_lines = ['', '# Comment 1 ' + a_tiddle, '', '# Comment 2', '',
'name=value']
with mocks.PatchUrlOpen(Response(1), 1, 503) as opener:
lines = freedb.perform_request(
"http://freedb", "query", "hello", "proto"
)
self.assertEqual(lines, expected_lines)
with mocks.PatchUrlOpen(Response(1), 3, 503) as opener:
lines = freedb.perform_request(
"http://freedb", "query", "hello", "proto"
)
self.assertEqual(lines, expected_lines)
with mocks.PatchUrlOpen(None, 4, 503) as opener:
lines = freedb.perform_request(
"http://freedb", "query", "hello", "proto"
)
self.assertEqual(lines, None)
def test_perform_request_bad_url(self):
with mocks.PatchUrlOpen(None, 1, "bad_url") as opener:
lines = freedb.perform_request(
"http://freedb", "query", "hello", "proto"
)
self.assertEqual(lines, None)
