conftest.py

#
# Copyright 2023 Ocean Protocol Foundation
# SPDX-License-Identifier: Apache-2.0
#

import os

import pytest
from eth_account import Account
from ocean_provider.run import app
from ocean_provider.utils.basics import get_provider_private_key, get_web3, send_ether
from ocean_provider.utils.provider_fees import get_c2d_environments

app = app


@pytest.fixture
def client():
    client = app.test_client()
    yield client


@pytest.fixture
def publisher_wallet():
    return Account.from_key(os.getenv("TEST_PRIVATE_KEY1"))


@pytest.fixture
def publisher_address(publisher_wallet):
    return publisher_wallet.address


@pytest.fixture
def consumer_wallet():
    return Account.from_key(os.getenv("TEST_PRIVATE_KEY2"))


@pytest.fixture
def consumer_address(consumer_wallet):
    return consumer_wallet.address


@pytest.fixture
def ganache_wallet():
    web3 = get_web3(8996)
    if (
        web3.eth.accounts
        and web3.eth.accounts[0].lower()
        == "0xe2DD09d719Da89e5a3D0F2549c7E24566e947260".lower()
    ):
        return Account.from_key(
            "0xfd5c1ccea015b6d663618850824154a3b3fb2882c46cefb05b9a93fea8c3d215"
        )

    return None


@pytest.fixture
def provider_wallet():
    pk = get_provider_private_key(8996)
    return Account.from_key(pk)


@pytest.fixture
def provider_address(provider_wallet):
    return provider_wallet.address


@pytest.fixture(autouse=True)
def setup_all(provider_address, consumer_address, ganache_wallet):
    web3 = get_web3(8996)
    if ganache_wallet:
        if (
            web3.fromWei(
                web3.eth.get_balance(provider_address, block_identifier="latest"),
                "ether",
            )
            < 10
        ):
            send_ether(web3, ganache_wallet, provider_address, 25)

        if (
            web3.fromWei(
                web3.eth.get_balance(provider_address, block_identifier="latest"),
                "ether",
            )
            < 10
        ):
            send_ether(web3, ganache_wallet, consumer_address, 25)


@pytest.fixture
def web3():
    return get_web3(8996)


@pytest.fixture
def free_c2d_env():
    environments = get_c2d_environments(flat=True)

    return next(env for env in environments if float(env["priceMin"]) == float(0))


@pytest.fixture
def paid_c2d_env():
    environments = get_c2d_environments(flat=True)

    return next(env for env in environments if env["id"] == "ocean-compute-env2")