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)