def test_request_includes_auth_headers(self):
def server_response(request, uri, response_headers):
headers = ['BB-ACCESS-KEY', 'BB-ACCESS-SIGN', 'BB-TIMESTAMP',
'Accept', 'Content-Type', 'User-Agent']
for header in headers:
self.assertIn(header, request.headers)
self.assertNotEqual(request.headers[header], '')
if 'BB-PARTNER-ID' in request.headers:
self.assertNotIn('BB-USER-ID', request.headers)
return 200, response_headers, '{}'
if 'BB-USER-ID' in request.headers:
self.assertNotIn('BB-PARTNER-ID', request.headers)
return 200, response_headers, '{}'
return 400, response_headers, '{}'
# Test a partner route
partner_client = Client(partner_api_key, partner_secret, base_uri,
partner_id=partner_id,
partnership_id=partnership_id)
url = base_uri + get_all_users_route
httpretty.register_uri(httpretty.GET, url, server_response)
response = partner_client.get_all_users()
self.assertEqual(response.status_code, 200)
# Test a user route
user_client = Client(user_api_key, user_secret, base_uri)
url = base_uri + get_user_balance_route
httpretty.register_uri(httpretty.GET, url, server_response)
response = user_client.get_user_balance(user_id)
self.assertEqual(response.status_code, 200)
def test_get_all_users(self):
client = Client(partner_api_key, partner_secret, base_uri,
partner_id=partner_id, partnership_id=partnership_id)
response = client.get_all_users()
self.assertEqual(response.json()['data'], mock_collection)