import smartpy as sp

from templates import fa2_lib as fa2
from templates import fa2_lib_testing as testing

administrator = sp.test_account("Administrator")
alice = sp.test_account("Alice")
tok0_md = fa2.make_metadata(name="Token Zero", decimals=1, symbol="Tok0")
tok1_md = fa2.make_metadata(name="Token One", decimals=1, symbol="Tok1")
tok2_md = fa2.make_metadata(name="Token Two", decimals=1, symbol="Tok2")
TOKEN_METADATA = [tok0_md, tok1_md, tok2_md]
METADATA = sp.utils.metadata_of_url("ipfs://example")

main = fa2.main


@sp.module
def m():
    # Order of inheritance: [Admin], [<policy>], <base class>, [<mixins>]

    class SingleAssetTestNoTransfer(main.NoTransfer, main.SingleAsset):
        def __init__(self, metadata, ledger, token_metadata):
            main.SingleAsset.__init__(self, metadata, ledger, token_metadata)
            main.NoTransfer.__init__(self)

    class SingleAssetTestOwnerTransfer(main.OwnerTransfer, main.SingleAsset):
        def __init__(self, metadata, ledger, token_metadata):
            main.SingleAsset.__init__(self, metadata, ledger, token_metadata)
            main.OwnerTransfer.__init__(self)

    class SingleAssetTestPauseOwnerOrOperatorTransfer(
        main.Admin, main.PauseOwnerOrOperatorTransfer, main.SingleAsset
    ):
        def __init__(self, administrator, metadata, ledger, token_metadata):
            main.SingleAsset.__init__(self, metadata, ledger, token_metadata)
            main.PauseOwnerOrOperatorTransfer.__init__(self)
            main.Admin.__init__(self, administrator)

    class SingleAssetTestFull(
        main.Admin,
        main.SingleAsset,
        main.ChangeMetadata,
        main.WithdrawMutez,
        main.MintSingleAsset,
        main.BurnSingleAsset,
        main.OffchainviewTokenMetadata,
        main.OnchainviewBalanceOf,
    ):
        def __init__(self, administrator, metadata, ledger, token_metadata):
            main.OnchainviewBalanceOf.__init__(self)
            main.OffchainviewTokenMetadata.__init__(self)
            main.BurnSingleAsset.__init__(self)
            main.MintSingleAsset.__init__(self)
            main.WithdrawMutez.__init__(self)
            main.ChangeMetadata.__init__(self)
            main.SingleAsset.__init__(self, metadata, ledger, token_metadata)
            main.Admin.__init__(self, administrator)


@sp.add_test(name="SingleAsset", is_default=True)
def test():
    ledger = {alice.address: 42}
    token_metadata = tok0_md

    # Default SingleAsset
    c1 = m.SingleAssetTestFull(
        administrator=administrator.address,
        metadata=METADATA,
        ledger=ledger,
        token_metadata=token_metadata,
    )

    # No transfer
    c2 = m.SingleAssetTestNoTransfer(
        metadata=METADATA,
        ledger=ledger,
        token_metadata=token_metadata,
    )

    # Owner transfer
    c3 = m.SingleAssetTestOwnerTransfer(
        metadata=METADATA,
        ledger=ledger,
        token_metadata=token_metadata,
    )

    # Empty SingleAsset
    c4 = m.SingleAssetTestFull(
        administrator=administrator.address,
        metadata=METADATA,
        ledger={},
        token_metadata=token_metadata,
    )

    # Pause owner or operator transfer
    c5 = m.SingleAssetTestPauseOwnerOrOperatorTransfer(
        administrator=administrator.address,
        metadata=METADATA,
        ledger=ledger,
        token_metadata=token_metadata,
    )

    kwargs = {
        "modules": [fa2.t, fa2.main, m],
        "ledger_type": "SingleAsset",
    }

    # Standard features
    testing.test_core_interfaces(c1, **kwargs)
    testing.test_transfer(c1, **kwargs)
    testing.test_balance_of(c1, **kwargs)
    # Policies
    testing.test_owner_or_operator_transfer(c1, **kwargs)
    testing.test_no_transfer(c2, **kwargs)
    testing.test_owner_transfer(c3, **kwargs)

    # Non standard features
    testing.NS.test_admin(c1, **kwargs)
    testing.NS.test_mint(c4, **kwargs)
    testing.NS.test_burn(c1, supports_transfer=True, supports_operator=True, **kwargs)
    testing.NS.test_withdraw_mutez(c1, **kwargs)
    testing.NS.test_change_metadata(c1, **kwargs)
    testing.NS.test_get_balance_of(c1, **kwargs)
    # Non standard policies
    testing.NS.test_pause(c5, **kwargs)