CHIA-1730: port chia plotnft to @chia_commands framework

chia/_tests/cmds/test_cmd_framework.py

@@ -16,7 +16,7 @@
 from chia.types.blockchain_format.sized_bytes import bytes32
 
 
-def check_click_parsing(cmd: ChiaCommand, *args: str) -> None:
+def check_click_parsing(cmd: ChiaCommand, *args: str, obj: Optional[Any] = None) -> None:
     @click.group()
     def _cmd() -> None:
         pass
@@ -40,7 +40,7 @@ def new_run(self: Any) -> None:
     chia_command(_cmd, "_", "")(mock_type)
 
     runner = CliRunner()
-    result = runner.invoke(_cmd, ["_", *args], catch_exceptions=False)
+    result = runner.invoke(_cmd, ["_", *args], catch_exceptions=False, obj=obj)
     assert result.output == ""
 
 

chia/_tests/pools/conftest.py

@@ -0,0 +1,3 @@
+from __future__ import annotations
+
+from chia._tests.wallet.conftest import trusted_full_node, tx_config, wallet_environments  # noqa: F401

chia/_tests/pools/test_pool_cli_parsing.py

@@ -0,0 +1,126 @@
+from __future__ import annotations
+
+from chia._tests.cmds.test_cmd_framework import check_click_parsing
+from chia.cmds.cmd_classes import NeedsWalletRPC
+from chia.cmds.param_types import CliAddress
+from chia.cmds.plotnft import (
+    ChangePayoutInstructionsPlotNFTCMD,
+    ClaimPlotNFTCMD,
+    CreatePlotNFTCMD,
+    GetLoginLinkCMD,
+    InspectPlotNFTCMD,
+    JoinPlotNFTCMD,
+    LeavePlotNFTCMD,
+    ShowPlotNFTCMD,
+)
+from chia.types.blockchain_format.sized_bytes import bytes32
+from chia.util.bech32m import encode_puzzle_hash
+from chia.util.ints import uint64
+from chia.wallet.util.address_type import AddressType
+
+
+def test_plotnft_command_default_parsing() -> None:
+    launcher_id = bytes32([1] * 32)
+    check_click_parsing(
+        GetLoginLinkCMD(context=dict(), launcher_id=launcher_id),
+        "--launcher_id",
+        launcher_id.hex(),
+    )
+
+    burn_ph = bytes32.from_hexstr("0x000000000000000000000000000000000000000000000000000000000000dead")
+    burn_address = encode_puzzle_hash(burn_ph, "xch")
+    check_click_parsing(
+        ChangePayoutInstructionsPlotNFTCMD(
+            launcher_id=launcher_id, address=CliAddress(burn_ph, burn_address, AddressType.XCH)
+        ),
+        "--launcher_id",
+        launcher_id.hex(),
+        "--address",
+        burn_address,
+        obj={"expected_prefix": "xch"},  # Needed for AddressParamType to work correctly without config
+    )
+
+    check_click_parsing(
+        ClaimPlotNFTCMD(
+            rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None), fee=uint64(1), id=5
+        ),
+        "--id",
+        "5",
+        "--fee",
+        "0.000000000001",
+    )
+
+    check_click_parsing(
+        CreatePlotNFTCMD(
+            rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None),
+            pool_url="http://localhost:1234",
+            state="pool",
+            fee=uint64(0),
+            dont_prompt=False,
+        ),
+        "--state",
+        "pool",
+        "--pool-url",
+        "http://localhost:1234",
+        "--fee",
+        "0.0",
+    )
+
+    check_click_parsing(
+        CreatePlotNFTCMD(
+            rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None),
+            pool_url=None,
+            state="local",
+            fee=uint64(0),
+            dont_prompt=True,
+        ),
+        "--state",
+        "local",
+        "-y",
+    )
+
+    check_click_parsing(
+        InspectPlotNFTCMD(
+            rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None),
+            id=5,
+        ),
+        "--id",
+        "5",
+    )
+
+    check_click_parsing(
+        JoinPlotNFTCMD(
+            rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None),
+            id=5,
+            fee=uint64(3),
+            pool_url="http://localhost:1234",
+            dont_prompt=True,
+        ),
+        "--id",
+        "5",
+        "--fee",
+        "0.000000000003",
+        "--pool-url",
+        "http://localhost:1234",
+        "-y",
+    )
+
+    check_click_parsing(
+        LeavePlotNFTCMD(
+            rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None),
+            id=5,
+            fee=uint64(3),
+            dont_prompt=True,
+        ),
+        "--id",
+        "5",
+        "--fee",
+        "0.000000000003",
+        "-y",
+    )
+
+    check_click_parsing(
+        ShowPlotNFTCMD(rpc_info=NeedsWalletRPC(client_info=None, wallet_rpc_port=None, fingerprint=None), id=5),
+        "--id",
+        "5",
+    )