def test_read_context_from_url_file(parser_class_map):
content = 'my_content'
class_ = MagicMock()
class_().load.return_value = content
parser_class_map.get.return_value = class_
with patch('pygen.parsers.open', mock_open(read_data="foo")):
result = read_context_from_url('file://foo.txt')
parser_class_map.get.assert_called_once_with('text/plain')
assert result == 'my_content'
def test_read_context_from_url_http(parser_class_map):
content = 'my_content'
class_ = MagicMock()
class_().loads.return_value = content
parser_class_map.get.return_value = class_
request = MagicMock()
request.headers = {'Content-Type': 'text/plain'}
with patch('pygen.parsers.requests.get', return_value=request):
result = read_context_from_url('http://foo.txt')
parser_class_map.get.assert_called_once_with('text/plain')
assert result == 'my_content'
