17
17
18
18
19
19
class MockResponse :
20
- def __init__ (self , content ):
21
- self .content = content
22
- self .text = content . decode ()
20
+ def __init__ (self , text ):
21
+ self .text = text
22
+ self .content = text . encode ()
23
23
24
24
25
25
def data_path (filename , output = False ):
@@ -44,7 +44,7 @@ def get_mockreturn(session, method, url, params=None, timeout=10, **kwargs):
44
44
45
45
filename = filename_for_request (url , params )
46
46
try :
47
- content = open (filename , "rb " ).read ()
47
+ content = open (filename , "rt " ).read ()
48
48
except FileNotFoundError :
49
49
log .error (
50
50
f"no stored mock data in { filename } for url=\" { url } \" and params=\" { params } \" "
@@ -58,20 +58,20 @@ def get_mockreturn(session, method, url, params=None, timeout=10, **kwargs):
58
58
59
59
def save_response_of_get (session , method , url , params = None , timeout = 10 , ** kwargs ):
60
60
61
- content = requests .Session ._original_request (
61
+ text = requests .Session ._original_request (
62
62
session , method , url , params = params , timeout = timeout
63
- ).content
63
+ ).text
64
64
65
65
filename = filename_for_request (url , params , output = True )
66
66
67
- with open (filename , "wb " ) as f :
67
+ with open (filename , "wt " ) as f :
68
68
log .info (f"saving output to { filename } for url=\" { url } \" and params=\" { params } \" " )
69
69
log .warning (
70
70
f"you may want to run `cp -fv { os .path .dirname (filename )} /* astroquery/heasarc/tests/data/; rm -rfv build`"
71
71
)
72
- f .write (content )
72
+ f .write (text )
73
73
74
- return MockResponse (content )
74
+ return MockResponse (text )
75
75
76
76
77
77
@pytest .fixture (autouse = True )
0 commit comments