DownloadRequest,
EncryptRequest,
FileInfoRequest,
InitializeRequest,
NonceRequest,
SimpleFlowConsumeRequest,
)
from ocean_utils.agreements.service_types import ServiceTypes
from ocean_utils.did import did_to_id
from ocean_utils.http_requests.requests_session import get_requests_session
from . import services
setup_logging()
setup_network()
provider_wallet = get_provider_wallet()
requests_session = get_requests_session()
requests_session.mount("file://", LocalFileAdapter())
logger = logging.getLogger(__name__)
@services.route("/nonce", methods=["GET"])
@validate(NonceRequest)
def nonce():
data = get_request_data(request)
address = data.get("userAddress")
nonce = get_nonce(address)
logger.info(f"nonce for user {address} is {nonce}")
return Response(json.dumps({"nonce": nonce}),
200,
def test_fails(client):
provider_wallet = get_provider_wallet()
consumer_address = get_consumer_wallet().address
pub_wallet = get_publisher_wallet()
(
dataset,
did,
tx_id,
sa,
_,
alg_ddo,
alg_data_token,
_,
alg_tx_id,
) = build_and_send_ddo_with_compute_service(client)
# output key is invalid
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": "this can not be decoded",
"algorithmDid": alg_ddo.did,
"algorithmDataToken": alg_data_token,
"algorithmTransferTxId": alg_tx_id,
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert validator.error == "Output is invalid or can not be decoded."
# algorithmDid is not actually an algorithm
data = {
"documentId":
did,
"transferTxId":
tx_id,
"serviceId":
sa.index,
"output":
build_stage_output_dict(dict(), sa.service_endpoint, consumer_address,
pub_wallet),
"algorithmDid":
did,
"algorithmDataToken":
alg_data_token,
"algorithmTransferTxId":
alg_tx_id,
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert validator.error == f"DID {did} is not a valid algorithm"
valid_output = build_stage_output_dict(dict(), sa.service_endpoint,
consumer_address, pub_wallet)
# algorithmMeta doesn't contain 'url' or 'rawcode'
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": valid_output,
"algorithmMeta": {},
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert (
validator.error ==
"algorithmMeta must define one of `url` or `rawcode` or `remote`, but all seem missing."
)
# algorithmMeta container is empty
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": valid_output,
"algorithmMeta": {
"rawcode": "console.log('Hello world'!)",
"format": "docker-image",
"version": "0.1",
"container": {},
},
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert (
validator.error ==
"algorithm `container` must specify values for all of entrypoint, image and tag."
)
# algorithmMeta container is missing image
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": valid_output,
"algorithmMeta": {
"rawcode": "console.log('Hello world'!)",
"format": "docker-image",
"version": "0.1",
"container": {
"entrypoint": "node $ALGO",
"tag": "10"
},
},
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert (
validator.error ==
"algorithm `container` must specify values for all of entrypoint, image and tag."
)
# Additional Input validations ###
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": valid_output,
"algorithmDid": alg_ddo.did,
"algorithmDataToken": alg_data_token,
"algorithmTransferTxId": alg_tx_id,
"additionalInputs": "",
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is True
# additional input is invalid
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": valid_output,
"algorithmDid": alg_ddo.did,
"algorithmDataToken": alg_data_token,
"algorithmTransferTxId": alg_tx_id,
"additionalInputs": "i can not be decoded in json!",
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert validator.error == "Additional input is invalid or can not be decoded."
# Missing did in additional input
data = {
"documentId": did,
"transferTxId": tx_id,
"serviceId": sa.index,
"output": valid_output,
"algorithmDid": alg_ddo.did,
"algorithmDataToken": alg_data_token,
"algorithmTransferTxId": alg_tx_id,
"additionalInputs": [{
"transferTxId": tx_id,
"serviceId": sa.index
}],
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert validator.error == "Error in input at index 1: No documentId in input item."
# Did is not valid
data = {
"documentId":
did,
"transferTxId":
tx_id,
"serviceId":
sa.index,
"output":
valid_output,
"algorithmDid":
alg_ddo.did,
"algorithmDataToken":
alg_data_token,
"algorithmTransferTxId":
alg_tx_id,
"additionalInputs": [{
"documentId": "i am not a did",
"transferTxId": tx_id,
"serviceId": sa.index,
}],
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert (
validator.error ==
"Error in input at index 1: Asset for did i am not a did not found.")
# Service is not compute, nor access
other_service = [
s for s in dataset.services if s.type not in
[ServiceTypes.CLOUD_COMPUTE, ServiceTypes.ASSET_ACCESS]
][0]
data = {
"documentId":
did,
"transferTxId":
tx_id,
"serviceId":
sa.index,
"output":
valid_output,
"algorithmDid":
alg_ddo.did,
"algorithmDataToken":
alg_data_token,
"algorithmTransferTxId":
alg_tx_id,
"additionalInputs": [{
"documentId": did,
"transferTxId": tx_id,
"serviceId": other_service.index
}],
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert (
validator.error ==
"Error in input at index 1: Services in input can only be access or compute."
)
# Additional input has other trusted algs
(
_,
trust_did,
trust_tx_id,
trust_sa,
_,
_,
_,
_,
_,
) = build_and_send_ddo_with_compute_service(
client, asset_type="specific_algo_dids")
data = {
"documentId":
did,
"transferTxId":
tx_id,
"serviceId":
sa.index,
"output":
valid_output,
"algorithmDid":
alg_ddo.did,
"algorithmDataToken":
alg_data_token,
"algorithmTransferTxId":
alg_tx_id,
"additionalInputs": [{
"documentId": trust_did,
"transferTxId": trust_tx_id,
"serviceId": trust_sa.index,
}],
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is False
assert (
validator.error ==
f"Error in input at index 1: this algorithm did {alg_ddo.did} is not trusted."
)
def get_provider_address():
"""Gets the provider wallet address."""
provider_address = get_provider_wallet().address
return provider_address
def test_passes(client):
provider_wallet = get_provider_wallet()
consumer_address = get_consumer_wallet().address
pub_wallet = get_publisher_wallet()
(
_,
did,
tx_id,
sa,
_,
alg_ddo,
alg_data_token,
_,
alg_tx_id,
) = build_and_send_ddo_with_compute_service(client)
data = {
"documentId":
did,
"serviceId":
sa.index,
"transferTxId":
tx_id,
"output":
build_stage_output_dict(dict(), sa.service_endpoint, consumer_address,
pub_wallet),
"algorithmDid":
alg_ddo.did,
"algorithmDataToken":
alg_data_token,
"algorithmTransferTxId":
alg_tx_id,
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is True
data = {
"documentId":
did,
"serviceId":
sa.index,
"transferTxId":
tx_id,
"output":
build_stage_output_dict(dict(), sa.service_endpoint, consumer_address,
pub_wallet),
"algorithmMeta": {
"rawcode": "console.log('Hello world'!)",
"format": "docker-image",
"version": "0.1",
"container": {
"entrypoint": "node $ALGO",
"image": "node",
"tag": "10"
},
},
}
validator = WorkflowValidator(consumer_address, provider_wallet, data)
assert validator.validate() is True