@@ -17,13 +17,14 @@ def pytest_addoption(parser):
1717@pytest .fixture (scope = 'module' )
1818def manager ():
1919 import upcloud_api
20+
2021 return upcloud_api .CloudManager ("testuser" , "mock-api-password" )
2122
2223
2324def read_from_file (filename ):
2425 filename = filename .replace ("/" , "_" )
2526 cwd = os .path .dirname (__file__ )
26- f = open (cwd + '/json_data/' + filename )
27+ f = open (cwd + '/json_data/' + filename )
2728 return f .read ()
2829
2930
@@ -40,14 +41,19 @@ def mock_get(target, response_file=None):
4041 response_file = target + '.json'
4142
4243 data = Mock .read_from_file (response_file )
43- responses .add (responses .GET , Mock .base_url + '/' + target ,
44- body = data ,
45- status = 200 ,
46- content_type = 'application/json' )
44+ responses .add (
45+ responses .GET ,
46+ Mock .base_url + '/' + target ,
47+ body = data ,
48+ status = 200 ,
49+ content_type = 'application/json' ,
50+ )
4751 return data
4852
4953 @staticmethod
50- def __put_patch_post_callback (request , target , data , ignore_data_field = False , empty_payload = False ):
54+ def __put_patch_post_callback (
55+ request , target , data , ignore_data_field = False , empty_payload = False
56+ ):
5157 data_field = target .split ("/" )[0 ]
5258
5359 if not empty_payload :
@@ -57,61 +63,74 @@ def __put_patch_post_callback(request, target, data, ignore_data_field=False, em
5763 for field in data [data_field ]:
5864 if field in payload [data_field ]:
5965 data [data_field ][field ] = payload [data_field ][field ]
60- return (200 , {}, json .dumps (data ))
66+ return (200 , {}, json .dumps (data ))
6167
6268 @staticmethod
6369 def mock_post (target , empty_content = False , ignore_data_field = False , empty_payload = False ):
64-
6570 def callback (request ):
6671 if not empty_content :
6772 data = json .loads (Mock .read_from_file (target + '_post.json' ))
68- return Mock .__put_patch_post_callback (request , target , data , ignore_data_field , empty_payload )
73+ return Mock .__put_patch_post_callback (
74+ request , target , data , ignore_data_field , empty_payload
75+ )
6976 else :
70- return (200 , {}, '{}' )
77+ return (200 , {}, '{}' )
7178
72- responses .add_callback (responses .POST , Mock .base_url + '/' + target ,
73- callback = callback ,
74- content_type = 'application/json' )
79+ responses .add_callback (
80+ responses .POST ,
81+ Mock .base_url + '/' + target ,
82+ callback = callback ,
83+ content_type = 'application/json' ,
84+ )
7585
7686 @staticmethod
7787 def mock_put (target , ignore_data_field = False , empty_payload = False , call_api = True ):
7888 data = json .loads (Mock .read_from_file (target + '.json' ))
7989
8090 def callback (request ):
81- return Mock .__put_patch_post_callback (request , target , data , ignore_data_field , empty_payload )
91+ return Mock .__put_patch_post_callback (
92+ request , target , data , ignore_data_field , empty_payload
93+ )
8294
8395 url = Mock .base_url + '/' + target if call_api else target
84- responses .add_callback (responses . PUT , url ,
85- callback = callback ,
86- content_type = 'application/json' )
96+ responses .add_callback (
97+ responses . PUT , url , callback = callback , content_type = 'application/json'
98+ )
8799
88100 @staticmethod
89101 def mock_patch (target , ignore_data_field = False , empty_payload = False , call_api = True ):
90102 data = json .loads (Mock .read_from_file (target + '.json' ))
91103
92104 def callback (request ):
93- return Mock .__put_patch_post_callback (request , target , data , ignore_data_field , empty_payload )
105+ return Mock .__put_patch_post_callback (
106+ request , target , data , ignore_data_field , empty_payload
107+ )
94108
95109 url = Mock .base_url + '/' + target if call_api else target
96- responses .add_callback (responses . PATCH , url ,
97- callback = callback ,
98- content_type = 'application/json' )
110+ responses .add_callback (
111+ responses . PATCH , url , callback = callback , content_type = 'application/json'
112+ )
99113
100114 @staticmethod
101115 def mock_delete (target ):
102- responses .add (responses .DELETE , Mock .base_url + '/' + target ,
103- status = 204 )
116+ responses .add (responses .DELETE , Mock .base_url + '/' + target , status = 204 )
104117
105118 @staticmethod
106119 def mock_server_operation (target ):
107120 # drop third (last) part of a string divided by two slashes ("/"); e.g "this/is/string" -> "this/is"
108121 targetsplit = target .split ('/' )
109- targetfile = '/' .join ( targetsplit [:2 ] )
122+ targetfile = '/' .join (targetsplit [:2 ])
110123
111- data = json .loads ( Mock .read_from_file (targetfile + '.json' ) )
124+ data = json .loads (Mock .read_from_file (targetfile + '.json' ))
112125
113126 # API will always respond state: "started", see: Server.stop, Server.start, Server,restart
114127 data ['server' ]['state' ] = 'started'
115128
116- data = json .dumps ( data )
117- responses .add (responses .POST , Mock .base_url + "/" + target , status = 200 , body = data , content_type = 'application/json' )
129+ data = json .dumps (data )
130+ responses .add (
131+ responses .POST ,
132+ Mock .base_url + "/" + target ,
133+ status = 200 ,
134+ body = data ,
135+ content_type = 'application/json' ,
136+ )
0 commit comments