mirror/github-octodns
1
0
Fork 0
mirror of https://github.com/github/octodns.git synced 2024-05-11 05:55:00 +00:00
Code Issues Releases Activity

made hetzner.HetznerProvider._do Mock-able for testing purposes making it a wrapper for _do_raw

Browse Source
This commit is contained in:
Ricard Bejarano
2021-04-26 19:02:44 +02:00
parent ab436af92d
commit fe74c46213
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
octodns/provider/hetzner.py
View File

@@ -46,7 +46,7 @@ class HetznerClient(object):
session.headers.update({'Auth-API-Token': token})
self._session = session
def _do(self, method, path, params=None, data=None):
def _do_raw(self, method, path, params=None, data=None):
url = self.BASE_URL + path
response = self._session.request(method, url, params=params, json=data)
if response.status_code == 401:
@@ -54,7 +54,10 @@ class HetznerClient(object):
if response.status_code == 404:
raise HetznerClientNotFound()
response.raise_for_status()
return response.json()
return response
def _do(self, method, path, params=None, data=None):
return self._do_raw(method, path, params, data).json()
def _do_with_pagination(self, method, path, key, params=None, data=None,
per_page=100):