pumpfun-cli/tests/test_core/test_launch.py at 57a20402365fbaf0ea30ac8f92a329aa97cd01f5 · chainstacklabs/pumpfun-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
"""Tests for core/launch.py — cashback flag pass-through."""

from unittest.mock import AsyncMock, patch

import pytest


@pytest.mark.asyncio
async def test_launch_passes_cashback_false(tmp_keystore):
    """launch_token passes is_cashback=False through to build_create_instructions."""
    with (
        patch(
            "pumpfun_cli.core.launch.upload_metadata",
            new_callable=AsyncMock,
            return_value="https://ipfs.example.com/metadata.json",
        ),
        patch("pumpfun_cli.core.launch.build_create_instructions", wraps=None) as mock_build,
        patch("pumpfun_cli.core.launch.build_extend_account_instruction"),
        patch(
            "pumpfun_cli.core.launch.RpcClient",
        ) as mock_rpc_cls,
    ):
        mock_client = AsyncMock()
        mock_client.send_tx = AsyncMock(return_value="fakesig123")
        mock_rpc_cls.return_value = mock_client

        # build_create_instructions needs to return a list of instructions
        from unittest.mock import MagicMock

        mock_ix = MagicMock()
        mock_build.return_value = [mock_ix]

        from pumpfun_cli.core.launch import launch_token

        result = await launch_token(
            rpc_url="https://fake.rpc",
            keystore_path=tmp_keystore,
            password="testpass",
            name="TestToken",
            ticker="TST",
            description="A test token",
            is_cashback=False,
        )

        mock_build.assert_called_once()
        call_kwargs = mock_build.call_args
        assert call_kwargs.kwargs.get("is_cashback") is False or (
            not call_kwargs.kwargs.get("is_cashback", True)
        )
        assert result["is_cashback"] is False


@pytest.mark.asyncio
async def test_launch_passes_cashback_true(tmp_keystore):
    """launch_token passes is_cashback=True through to build_create_instructions."""
    with (
        patch(
            "pumpfun_cli.core.launch.upload_metadata",
            new_callable=AsyncMock,
            return_value="https://ipfs.example.com/metadata.json",
        ),
        patch("pumpfun_cli.core.launch.build_create_instructions", wraps=None) as mock_build,
        patch("pumpfun_cli.core.launch.build_extend_account_instruction"),
        patch(
            "pumpfun_cli.core.launch.RpcClient",
        ) as mock_rpc_cls,
    ):
        mock_client = AsyncMock()
        mock_client.send_tx = AsyncMock(return_value="fakesig456")
        mock_rpc_cls.return_value = mock_client

        from unittest.mock import MagicMock

        mock_ix = MagicMock()
        mock_build.return_value = [mock_ix]

        from pumpfun_cli.core.launch import launch_token

        result = await launch_token(
            rpc_url="https://fake.rpc",
            keystore_path=tmp_keystore,
            password="testpass",
            name="TestToken",
            ticker="TST",
            description="A test token",
            is_cashback=True,
        )

        mock_build.assert_called_once()
        assert mock_build.call_args.kwargs["is_cashback"] is True
        assert result["is_cashback"] is True