def test_default_http_client_topup(
doggo_proxy,
default_http_client: DefaultHTTPClient,
http_doggo_url: str
):
# Create a channel that has just enough capacity for one transfer.
default_http_client.initial_deposit = lambda x: 0
check_response(default_http_client.get(http_doggo_url))
client = default_http_client.client
open_channels = client.get_open_channels()
assert len(open_channels) == 1
channel1 = open_channels[0]
assert channel1 == default_http_client.get_channel(http_doggo_url)
assert channel1.balance_sig
assert channel1.balance == channel1.deposit
# Do another payment. Topup should occur.
check_response(default_http_client.get(http_doggo_url))
open_channels = client.get_open_channels()
assert len(open_channels) == 1
channel2 = open_channels[0]
assert channel2 == default_http_client.get_channel(http_doggo_url)
assert channel2.balance_sig
assert channel2.balance < channel2.deposit
assert channel1 == channel2
def test_full_cycle_error_500(
default_http_client: DefaultHTTPClient,
api_endpoint_address: str,
token_contract_address,
channel_manager_contract_address,
receiver_address
):
default_http_client.initial_deposit = lambda x: x
with requests_mock.mock() as server_mock:
headers1 = Munch()
headers1.token_address = token_contract_address
headers1.contract_address = channel_manager_contract_address
headers1.receiver_address = receiver_address
headers1.price = '3'
headers2 = Munch()
headers2.cost = '3'
headers1 = HTTPHeaders.serialize(headers1)
headers2 = HTTPHeaders.serialize(headers2)
url = 'http://{}/something'.format(api_endpoint_address)
server_mock.get(url, [
{'status_code': 402, 'headers': headers1},
{'status_code': 500, 'headers': {}},
{'status_code': 200, 'headers': headers2, 'text': 'success'}
])
response = default_http_client.get(url)
# First cycle, request price.
request = server_mock.request_history[0]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_contract_address
# Second cycle, pay price but receive error.
request = server_mock.request_history[1]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_contract_address
assert request.headers['RDN-Balance'] == '3'
channel = default_http_client.get_channel(url)
assert channel.balance == 3
balance_sig_hex = encode_hex(channel.balance_sig)
assert request.headers['RDN-Balance-Signature'] == balance_sig_hex
# Third cycle, retry naively.
request = server_mock.request_history[2]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_contract_address
assert request.headers['RDN-Balance'] == '3'
assert channel.balance == 3
assert request.headers['RDN-Balance-Signature'] == balance_sig_hex
assert channel.balance_sig
assert response.text == 'success'
def test_custom_headers(default_http_client: DefaultHTTPClient,
api_endpoint_address: str, token_address: str,
channel_manager_address: str, receiver_address: str):
default_http_client.initial_deposit = lambda x: x
with requests_mock.mock() as server_mock:
headers1 = Munch()
headers1.token_address = token_address
headers1.contract_address = channel_manager_address
headers1.receiver_address = receiver_address
headers1.price = '7'
headers2 = Munch()
headers2.cost = '7'
headers1 = HTTPHeaders.serialize(headers1)
headers2 = HTTPHeaders.serialize(headers2)
url = 'http://{}/something'.format(api_endpoint_address)
server_mock.get(url, [{
'status_code': 402,
'headers': headers1
}, {
'status_code': 200,
'headers': headers2,
'text': 'success'
}])
response = default_http_client.get(
url,
headers={
'someheader': 'somevalue',
# This should override the actual balance but doesn't actually make sense.
'RDN-Balance': '5'
})
# First cycle, request price.
request = server_mock.request_history[0]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_address
assert request.headers['RDN-Balance'] == '5'
assert request.headers['someheader'] == 'somevalue'
# Second cycle, pay price.
request = server_mock.request_history[1]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_address
assert request.headers['RDN-Balance'] == '5'
assert request.headers['someheader'] == 'somevalue'
channel = default_http_client.get_channel(url)
assert channel.balance == 7
balance_sig_hex = encode_hex(channel.balance_sig)
assert request.headers['RDN-Balance-Signature'] == balance_sig_hex
assert channel.balance_sig
assert response.text == 'success'
def test_default_http_client(doggo_proxy,
default_http_client: DefaultHTTPClient,
sender_address: str, receiver_address: str,
http_doggo_url: str):
check_response(default_http_client.get(http_doggo_url))
client = default_http_client.client
open_channels = client.get_open_channels()
assert len(open_channels) == 1
channel = open_channels[0]
assert channel == default_http_client.get_channel(http_doggo_url)
assert channel.balance_sig
assert channel.balance < channel.deposit
assert channel.sender == sender_address
assert channel.receiver == receiver_address
def test_full_cycle_adapt_balance(
default_http_client: DefaultHTTPClient,
api_endpoint_address: str,
token_contract_address,
channel_manager_contract_address,
receiver_address
):
# Simulate a lost balance signature.
client = default_http_client.client
channel = client.get_suitable_channel(receiver_address, 10, initial_deposit=lambda x: 2 * x)
channel.create_transfer(3)
lost_balance_sig = channel.balance_sig
channel.update_balance(0)
with requests_mock.mock() as server_mock:
headers1 = Munch()
headers1.token_address = token_contract_address
headers1.contract_address = channel_manager_contract_address
headers1.receiver_address = receiver_address
headers1.price = '7'
headers2 = headers1.copy()
headers2.invalid_amount = '1'
headers2.sender_balance = '3'
headers2.balance_signature = encode_hex(lost_balance_sig)
headers3 = Munch()
headers3.cost = '7'
headers1 = HTTPHeaders.serialize(headers1)
headers2 = HTTPHeaders.serialize(headers2)
headers3 = HTTPHeaders.serialize(headers3)
url = 'http://{}/something'.format(api_endpoint_address)
server_mock.get(url, [
{'status_code': 402, 'headers': headers1},
{'status_code': 402, 'headers': headers2},
{'status_code': 200, 'headers': headers3, 'text': 'success'}
])
response = default_http_client.get(url)
# First cycle, request price.
request = server_mock.request_history[0]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_contract_address
# Second cycle, pay price based on outdated balance.
request = server_mock.request_history[1]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_contract_address
assert request.headers['RDN-Balance'] == '7'
assert request.headers['RDN-Balance-Signature']
# Third cycle, adapt new balance and pay price again.
request = server_mock.request_history[2]
assert request.path == '/something'
assert request.method == 'GET'
assert request.headers['RDN-Contract-Address'] == channel_manager_contract_address
assert request.headers['RDN-Balance'] == '10'
channel = default_http_client.get_channel(url)
assert channel.balance == 10
balance_sig_hex = encode_hex(channel.balance_sig)
assert request.headers['RDN-Balance-Signature'] == balance_sig_hex
assert channel.balance_sig
assert response.text == 'success'
