Skip to content

Commit ec21ca9

Browse files
authored
[CHIA-3390] Port remaining test_trades.py tests to parametrized with R-CATs (#19831)
* Port `test_trade_cancellation` * Port `test_trade_conflict` * Port `test_trade_bad_spend` * Port `test_trade_high_fee` * Port `test_aggregated_trade_state`
1 parent e8e060b commit ec21ca9

File tree

1 file changed

+55
-55
lines changed

1 file changed

+55
-55
lines changed

chia/_tests/wallet/cat_wallet/test_trades.py

Lines changed: 55 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -1629,8 +1629,9 @@ async def assert_trade_tx_number(wallet_node: WalletNode, trade_id: bytes32, num
16291629
indirect=True,
16301630
)
16311631
@pytest.mark.limit_consensus_modes(reason="irrelevant")
1632+
@pytest.mark.parametrize("wallet_type", [CATWallet, RCATWallet])
16321633
@pytest.mark.anyio
1633-
async def test_trade_cancellation(wallet_environments: WalletTestFramework) -> None:
1634+
async def test_trade_cancellation(wallet_environments: WalletTestFramework, wallet_type: type[CATWallet]) -> None:
16341635
env_maker = wallet_environments.environments[0]
16351636
env_taker = wallet_environments.environments[1]
16361637

@@ -1645,16 +1646,15 @@ async def test_trade_cancellation(wallet_environments: WalletTestFramework) -> N
16451646

16461647
xch_to_cat_amount = uint64(100)
16471648

1648-
async with env_maker.wallet_state_manager.new_action_scope(
1649-
wallet_environments.tx_config, push=True
1650-
) as action_scope:
1651-
cat_wallet_maker = await CATWallet.create_new_cat_wallet(
1652-
env_maker.wallet_state_manager,
1653-
env_maker.xch_wallet,
1654-
{"identifier": "genesis_by_id"},
1655-
xch_to_cat_amount,
1656-
action_scope,
1657-
)
1649+
cat_wallet_maker = await mint_cat(
1650+
wallet_environments,
1651+
env_maker,
1652+
"xch",
1653+
"cat",
1654+
xch_to_cat_amount,
1655+
wallet_type,
1656+
"cat",
1657+
)
16581658

16591659
await wallet_environments.process_pending_states(
16601660
[
@@ -1941,8 +1941,9 @@ async def test_trade_cancellation(wallet_environments: WalletTestFramework) -> N
19411941
indirect=True,
19421942
)
19431943
@pytest.mark.limit_consensus_modes(reason="irrelevant")
1944+
@pytest.mark.parametrize("wallet_type", [CATWallet, RCATWallet])
19441945
@pytest.mark.anyio
1945-
async def test_trade_conflict(wallet_environments: WalletTestFramework) -> None:
1946+
async def test_trade_conflict(wallet_environments: WalletTestFramework, wallet_type: type[CATWallet]) -> None:
19461947
env_maker = wallet_environments.environments[0]
19471948
env_taker = wallet_environments.environments[1]
19481949
env_trader = wallet_environments.environments[2]
@@ -1963,16 +1964,15 @@ async def test_trade_conflict(wallet_environments: WalletTestFramework) -> None:
19631964
xch_to_cat_amount = uint64(100)
19641965
fee = uint64(10)
19651966

1966-
async with env_maker.wallet_state_manager.new_action_scope(
1967-
wallet_environments.tx_config, push=True
1968-
) as action_scope:
1969-
await CATWallet.create_new_cat_wallet(
1970-
env_maker.wallet_state_manager,
1971-
env_maker.xch_wallet,
1972-
{"identifier": "genesis_by_id"},
1973-
xch_to_cat_amount,
1974-
action_scope,
1975-
)
1967+
await mint_cat(
1968+
wallet_environments,
1969+
env_maker,
1970+
"xch",
1971+
"cat",
1972+
xch_to_cat_amount,
1973+
wallet_type,
1974+
"cat",
1975+
)
19761976

19771977
await wallet_environments.process_pending_states(
19781978
[
@@ -2138,8 +2138,9 @@ async def test_trade_conflict(wallet_environments: WalletTestFramework) -> None:
21382138
indirect=True,
21392139
)
21402140
@pytest.mark.limit_consensus_modes(reason="irrelevant")
2141+
@pytest.mark.parametrize("wallet_type", [CATWallet, RCATWallet])
21412142
@pytest.mark.anyio
2142-
async def test_trade_bad_spend(wallet_environments: WalletTestFramework) -> None:
2143+
async def test_trade_bad_spend(wallet_environments: WalletTestFramework, wallet_type: type[CATWallet]) -> None:
21432144
env_maker = wallet_environments.environments[0]
21442145
env_taker = wallet_environments.environments[1]
21452146

@@ -2154,16 +2155,15 @@ async def test_trade_bad_spend(wallet_environments: WalletTestFramework) -> None
21542155

21552156
xch_to_cat_amount = uint64(100)
21562157

2157-
async with env_maker.wallet_state_manager.new_action_scope(
2158-
wallet_environments.tx_config, push=True
2159-
) as action_scope:
2160-
await CATWallet.create_new_cat_wallet(
2161-
env_maker.wallet_state_manager,
2162-
env_maker.xch_wallet,
2163-
{"identifier": "genesis_by_id"},
2164-
xch_to_cat_amount,
2165-
action_scope,
2166-
)
2158+
await mint_cat(
2159+
wallet_environments,
2160+
env_maker,
2161+
"xch",
2162+
"cat",
2163+
xch_to_cat_amount,
2164+
wallet_type,
2165+
"cat",
2166+
)
21672167

21682168
await wallet_environments.process_pending_states(
21692169
[
@@ -2260,8 +2260,9 @@ def check_wallet_cache_empty() -> bool:
22602260
indirect=True,
22612261
)
22622262
@pytest.mark.limit_consensus_modes(reason="irrelevant")
2263+
@pytest.mark.parametrize("wallet_type", [CATWallet, RCATWallet])
22632264
@pytest.mark.anyio
2264-
async def test_trade_high_fee(wallet_environments: WalletTestFramework) -> None:
2265+
async def test_trade_high_fee(wallet_environments: WalletTestFramework, wallet_type: type[CATWallet]) -> None:
22652266
env_maker = wallet_environments.environments[0]
22662267
env_taker = wallet_environments.environments[1]
22672268

@@ -2276,16 +2277,15 @@ async def test_trade_high_fee(wallet_environments: WalletTestFramework) -> None:
22762277

22772278
xch_to_cat_amount = uint64(100)
22782279

2279-
async with env_maker.wallet_state_manager.new_action_scope(
2280-
wallet_environments.tx_config, push=True
2281-
) as action_scope:
2282-
await CATWallet.create_new_cat_wallet(
2283-
env_maker.wallet_state_manager,
2284-
env_maker.xch_wallet,
2285-
{"identifier": "genesis_by_id"},
2286-
xch_to_cat_amount,
2287-
action_scope,
2288-
)
2280+
await mint_cat(
2281+
wallet_environments,
2282+
env_maker,
2283+
"xch",
2284+
"cat",
2285+
xch_to_cat_amount,
2286+
wallet_type,
2287+
"cat",
2288+
)
22892289

22902290
await wallet_environments.process_pending_states(
22912291
[
@@ -2405,8 +2405,9 @@ async def test_trade_high_fee(wallet_environments: WalletTestFramework) -> None:
24052405
indirect=True,
24062406
)
24072407
@pytest.mark.limit_consensus_modes(reason="irrelevant")
2408+
@pytest.mark.parametrize("wallet_type", [CATWallet, RCATWallet])
24082409
@pytest.mark.anyio
2409-
async def test_aggregated_trade_state(wallet_environments: WalletTestFramework) -> None:
2410+
async def test_aggregated_trade_state(wallet_environments: WalletTestFramework, wallet_type: type[CATWallet]) -> None:
24102411
env_maker = wallet_environments.environments[0]
24112412
env_taker = wallet_environments.environments[1]
24122413

@@ -2421,16 +2422,15 @@ async def test_aggregated_trade_state(wallet_environments: WalletTestFramework)
24212422

24222423
xch_to_cat_amount = uint64(100)
24232424

2424-
async with env_maker.wallet_state_manager.new_action_scope(
2425-
wallet_environments.tx_config, push=True
2426-
) as action_scope:
2427-
await CATWallet.create_new_cat_wallet(
2428-
env_maker.wallet_state_manager,
2429-
env_maker.xch_wallet,
2430-
{"identifier": "genesis_by_id"},
2431-
xch_to_cat_amount,
2432-
action_scope,
2433-
)
2425+
await mint_cat(
2426+
wallet_environments,
2427+
env_maker,
2428+
"xch",
2429+
"cat",
2430+
xch_to_cat_amount,
2431+
wallet_type,
2432+
"cat",
2433+
)
24342434

24352435
await wallet_environments.process_pending_states(
24362436
[

0 commit comments

Comments
 (0)