From 36ec97fe56b02e18d33ba9dee49f3377b1f47431 Mon Sep 17 00:00:00 2001 From: John Peterson Date: Mon, 4 Nov 2024 07:19:28 -0500 Subject: [PATCH] Lint and Unit Test PR CI Checks --- .github/workflows/lint.yml | 71 +++++++++++++++++ .github/workflows/unit_tests.yml | 79 +++++++++++++++++++ cdp-langchain/Makefile | 4 + cdp-langchain/tests/tools/test_cdp_action.py | 2 +- .../tests/utils/test_cdp_agentkit_wrapper.py | 18 ++--- 5 files changed, 164 insertions(+), 10 deletions(-) create mode 100644 .github/workflows/lint.yml create mode 100644 .github/workflows/unit_tests.yml diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml new file mode 100644 index 00000000..456385ec --- /dev/null +++ b/.github/workflows/lint.yml @@ -0,0 +1,71 @@ +name: Lint +on: [pull_request] + +jobs: + lint-agentkit: + runs-on: ubuntu-latest + defaults: + run: + working-directory: ./cdp-agentkit-core + steps: + - uses: actions/checkout@v3 + + - name: Set up Python + uses: actions/setup-python@v4 + with: + python-version: '3.10' + + - name: Install Poetry + uses: snok/install-poetry@v1 + with: + version: latest + virtualenvs-create: true + virtualenvs-in-project: true + + - name: Load cached venv + id: cached-poetry-dependencies + uses: actions/cache@v3 + with: + path: ./cdp-agentkit-core/.venv + key: venv-agentkit-${{ runner.os }}-${{ hashFiles('cdp-agentkit-core/poetry.lock') }} + + - name: Install dependencies + if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' + run: poetry install --with dev + + - name: Run linters + run: poetry run make lint + + lint-langchain: + runs-on: ubuntu-latest + defaults: + run: + working-directory: ./cdp-langchain + steps: + - uses: actions/checkout@v3 + + - name: Set up Python + uses: actions/setup-python@v4 + with: + python-version: '3.10' + + - name: Install Poetry + uses: snok/install-poetry@v1 + with: + version: latest + virtualenvs-create: true + virtualenvs-in-project: true + + - name: Load cached venv + id: cached-poetry-dependencies + uses: actions/cache@v3 + with: + path: ./cdp-langchain/.venv + key: venv-langchain-${{ runner.os }}-${{ hashFiles('cdp-langchain/poetry.lock') }} + + - name: Install dependencies + if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' + run: poetry install --with dev + + - name: Run linters + run: poetry run make lint diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml new file mode 100644 index 00000000..96763bc7 --- /dev/null +++ b/.github/workflows/unit_tests.yml @@ -0,0 +1,79 @@ +name: Run Unit Tests +on: [pull_request] + +jobs: + test-agentkit: + runs-on: ubuntu-latest + defaults: + run: + working-directory: ./cdp-agentkit-core + strategy: + matrix: + python: ['3.10', '3.11', '3.12'] + + steps: + - uses: actions/checkout@v3 + + - name: Set up Python ${{ matrix.python }} + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python }} + + - name: Install Poetry + uses: snok/install-poetry@v1 + with: + version: latest + virtualenvs-create: true + virtualenvs-in-project: true + + - name: Load cached venv + id: cached-poetry-dependencies + uses: actions/cache@v3 + with: + path: ./cdp-agentkit-core/.venv + key: venv-agentkit-${{ runner.os }}-${{ matrix.python }}-${{ hashFiles('cdp-agentkit-core/poetry.lock') }} + + - name: Install dependencies + if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' + run: poetry install --with dev + + - name: Run tests + run: poetry run make test + + test-langchain: + runs-on: ubuntu-latest + defaults: + run: + working-directory: ./cdp-langchain + strategy: + matrix: + python: ['3.10', '3.11', '3.12'] + + steps: + - uses: actions/checkout@v3 + + - name: Set up Python ${{ matrix.python }} + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python }} + + - name: Install Poetry + uses: snok/install-poetry@v1 + with: + version: latest + virtualenvs-create: true + virtualenvs-in-project: true + + - name: Load cached venv + id: cached-poetry-dependencies + uses: actions/cache@v3 + with: + path: ./cdp-langchain/.venv + key: venv-langchain-${{ runner.os }}-${{ matrix.python }}-${{ hashFiles('cdp-langchain/poetry.lock') }} + + - name: Install dependencies + if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true' + run: poetry install --with dev + + - name: Run tests + run: poetry run make test diff --git a/cdp-langchain/Makefile b/cdp-langchain/Makefile index 6a467169..a8fcdde5 100644 --- a/cdp-langchain/Makefile +++ b/cdp-langchain/Makefile @@ -9,3 +9,7 @@ lint: .PHONY: lint-fix lint-fix: ruff check . --fix + +.PHONY: test +test: + pytest diff --git a/cdp-langchain/tests/tools/test_cdp_action.py b/cdp-langchain/tests/tools/test_cdp_action.py index 1c599689..afe3c7bb 100644 --- a/cdp-langchain/tests/tools/test_cdp_action.py +++ b/cdp-langchain/tests/tools/test_cdp_action.py @@ -20,7 +20,7 @@ class TestArgsSchema(BaseModel): @pytest.fixture def mock_cdp_agentkit_wrapper(): """Fixture for mocked CDP Agentkit wrapper.""" - with patch("langchain_cdp.tools.cdp_action.CdpAgentkitWrapper") as mock: + with patch("cdp_langchain.tools.cdp_action.CdpAgentkitWrapper") as mock: cdp_agentkit_wrapper = Mock(spec=CdpAgentkitWrapper) mock.return_value = cdp_agentkit_wrapper yield cdp_agentkit_wrapper diff --git a/cdp-langchain/tests/utils/test_cdp_agentkit_wrapper.py b/cdp-langchain/tests/utils/test_cdp_agentkit_wrapper.py index 22b52b23..77cc787f 100644 --- a/cdp-langchain/tests/utils/test_cdp_agentkit_wrapper.py +++ b/cdp-langchain/tests/utils/test_cdp_agentkit_wrapper.py @@ -143,19 +143,19 @@ def test_cdp_sdk_import_error(): ( "get_wallet_details", {}, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.get_wallet_details_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.get_wallet_details_wrapper", "Got wallet details successfully", ), ( "get_balance", {"asset_id": "usdc"}, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.get_balance_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.get_balance_wrapper", "Got balance successfully", ), ( "request_faucet_funds", {}, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.request_faucet_funds_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.request_faucet_funds_wrapper", "Requested assets from faucet successfully", ), ( @@ -166,7 +166,7 @@ def test_cdp_sdk_import_error(): "destination": "example.base.eth", "gasless": True, }, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.transfer_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.transfer_wrapper", "Transferred successfully", ), ( @@ -176,7 +176,7 @@ def test_cdp_sdk_import_error(): "from_asset_id": "usdc", "to_asset_id": "weth", }, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.trade_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.trade_wrapper", "Traded successfully", ), ( @@ -186,7 +186,7 @@ def test_cdp_sdk_import_error(): "symbol": "TOKEN", "total_supply": "1000000", }, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.deploy_token_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.deploy_token_wrapper", "Deployed token successfully", ), ( @@ -195,7 +195,7 @@ def test_cdp_sdk_import_error(): "contract_address": "0xvalidContractAddress", "destination": "0xvalidAddress", }, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.mint_nft_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.mint_nft_wrapper", "Minted NFT successfully", ), ( @@ -205,7 +205,7 @@ def test_cdp_sdk_import_error(): "symbol": "TEST", "base_uri": "https://www.test.xyz/metadata/", }, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.deploy_nft_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.deploy_nft_wrapper", "Minted NFT successfully", ), ( @@ -213,7 +213,7 @@ def test_cdp_sdk_import_error(): { "basename": "test-basename", }, - "langchain_cdp.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.register_basename_wrapper", + "cdp_langchain.utils.cdp_agentkit_wrapper.CdpAgentkitWrapper.register_basename_wrapper", "Registered basename successfully", ), ],