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'