def test_modify_billing_request(self, put_mock):
modify_billing_request_response = {
'provider': {
'external_id': '321-123'
}
}
put_mock.return_value = Response(True, modify_billing_request_response,
200)
body = modify_billing_request_response
pk = 'BRP-6750-9514-7931-0001'
request = BillingRequestResource(config=self.config)
billing_request = request.update_billing_request(pk, body)
url = ('http://localhost:8080/api/public/v1'
'/subscriptions/requests/BRP-6750-9514-7931-0001/attributes')
put_mock.assert_called_with(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
json=body,
timeout=300,
url=url)
assert billing_request == ({
'provider': {
'external_id': '321-123'
}
}, 200)
class Subscription(object):
""" This class allows manage billing request and recurring asset,
:param Config config: Config object or ``None`` to use environment config (default).
"""
_config = None # type: Config
def __init__(self, config=None):
self._config = config or Config.get_instance()
self._billing_request = BillingRequestResource(config=self._config)
self._recurring_asset = RecurringAssetResource(config=self._config)
def create_billing_request(self, obj):
return self._billing_request.create(obj)
def update_billing_request(self, id_br, body):
return self._billing_request.update(id_br, body)
def test_billing_request_empty(self, get_mock):
# type: (Mock) -> None
get_mock.return_value = Response(True, '[]', 200)
request = BillingRequestResource(config=self.config)
billing_request = request.get('BPR-0000-0000-0000-0000')
assert get_mock.call_count == 1
url = 'http://localhost:8080/api/public/v1/subscriptions/requests/BPR-0000-0000-0000-0000'
get_mock.assert_has_calls([
call(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
timeout=300,
url=url)
])
self.assertEqual(billing_request, [], msg=None)
def test_add_billing_request(self, post_mock):
# type: (Mock) -> None
post_mock.return_value = Response(True, add_billing_request_response,
200)
body = add_billing_request_body
request = BillingRequestResource(config=self.config)
billing_request = request.create(body)
post_mock.assert_called_with(
headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
json=body,
timeout=300,
url='http://localhost:8080/api/public/v1/subscriptions/requests')
assert isinstance(billing_request, BillingRequest)
assert billing_request.id == 'BRP-6750-9514-7931-0001'
def test_list_billing_request_ok(self, get_mock):
# type: (Mock) -> None
get_mock.side_effect = [
Response(True, list_billing_request_contents, 200)
]
request = BillingRequestResource(config=self.config)
billing_request = request.list()
assert get_mock.call_count == 1
get_mock.assert_has_calls([
call(
headers={
'Authorization': 'ApiKey XXXX:YYYYY',
'Content-Type': 'application/json'
},
timeout=300,
url=
'http://localhost:8080/api/public/v1/subscriptions/requests?limit=100'
)
])
self.assertEqual(len(billing_request), 2, msg=None)
def test_get_billing_request_ok(self, get_mock):
# type: (Mock) -> None
get_mock.side_effect = [
Response(True, '[' + get_billing_request_contents + ']', 200),
Response(True, get_billing_request_contents, 200)
]
request = BillingRequestResource(config=self.config)
billing_request = request.get('BRP-3110-7077-0368-0001')
assert get_mock.call_count == 1
self.assertEqual(billing_request.type, 'provider', msg=None)
# 054922da-ceae-47de-8e5d-7a2950acbfe1
get_mock.assert_has_calls([
call(headers={
'Content-Type': 'application/json',
'Authorization': 'ApiKey XXXX:YYYYY'
},
timeout=300,
url=('http://localhost:8080/api/public/v1/'
'subscriptions/requests/BRP-3110-7077-0368-0001'))
])
assert isinstance(billing_request, BillingRequest)
def __init__(self, config=None):
self._config = config or Config.get_instance()
self._billing_request = BillingRequestResource(config=self._config)
self._recurring_asset = RecurringAssetResource(config=self._config)
def test_update_billing_request_bad(self):
request = BillingRequestResource(config=self.config)
with pytest.raises(ValueError) as e:
request.update_billing_request(None, None)
assert str(e.value) == 'Invalid ID'