Skip to content

Commit e9a1d8c

Browse files
Add back test settings.py
1 parent 722fe1a commit e9a1d8c

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

tests/settings.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
"""Defines all test wide settings and variables"""
2+
from os import environ
3+
from yarl import URL
4+
5+
from domaintools import API
6+
from vcr import VCR
7+
8+
9+
def remove_server(response):
10+
response.get('headers', {}).pop('server', None)
11+
if 'url' in response:
12+
url = URL(response['url'])
13+
query = dict(url.query)
14+
if 'api_username' in query:
15+
query.update(api_username='test')
16+
if 'api_key' in query:
17+
query.update(api_key='test')
18+
if 'signature' in query:
19+
query.update(signature='test')
20+
response['url'] = str(url.with_query(query))
21+
return response
22+
23+
24+
vcr = VCR(before_record_response=remove_server, filter_query_parameters=['timestamp', 'signature', 'api_username', 'api_key'], filter_post_data_parameters=['timestamp', 'signature', 'api_username', 'api_key'],
25+
cassette_library_dir='tests/fixtures/vcr/', path_transformer=VCR.ensure_suffix('.yaml'),
26+
record_mode='new_episodes')
27+
with vcr.use_cassette('init_user_account'):
28+
api = API(environ.get('TEST_USER', 'test'), environ.get('TEST_KEY', 'test'), always_sign_api_key=True)

0 commit comments

Comments
 (0)