Skip to content

Commit 5419e51

Browse files
committed
fix NRAO monkeypatching
1 parent 83f3b5b commit 5419e51

File tree

1 file changed

+6
-15
lines changed

1 file changed

+6
-15
lines changed

astroquery/nrao/tests/test_nrao.py

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -28,29 +28,20 @@ def parse_coordinates_mock_return(c):
2828
return mp
2929

3030

31-
@pytest.fixture
32-
def patch_get(request):
33-
mp = request.getfuncargvalue("monkeypatch")
34-
mp.setattr(requests, 'get', get_mockreturn)
35-
return mp
36-
3731

3832
@pytest.fixture
3933
def patch_post(request):
4034
mp = request.getfuncargvalue("monkeypatch")
41-
mp.setattr(requests, 'post', post_mockreturn)
35+
mp.setattr(requests.Session, 'request', post_mockreturn)
4236
return mp
4337

4438

45-
def get_mockreturn(url, params=None, timeout=10, **kwargs):
46-
filename = data_path(DATA_FILES['votable'])
47-
content = open(filename, 'rb').read()
48-
return MockResponse(content, **kwargs)
49-
50-
51-
def post_mockreturn(url, data=None, timeout=10, **kwargs):
39+
def post_mockreturn(self, method, url, data=None, timeout=10, files=None,
40+
params=None, headers=None, **kwargs):
41+
if method != 'POST':
42+
raise ValueError("A 'post request' was made with method != POST")
5243
filename = data_path(DATA_FILES['votable'])
53-
content = open(filename, 'rb').read()
44+
content = open(filename, "rb").read()
5445
return MockResponse(content, **kwargs)
5546

5647

0 commit comments

Comments
 (0)