def test_display_defi_vaults(mocker):
# MOCK EXPORT_DATA
mocker.patch(
target="gamestonk_terminal.cryptocurrency.defi.coindix_view.export_data"
)
coindix_view.display_defi_vaults(
chain=None,
protocol=None,
kind=None,
top=None,
sortby="apy",
descend=False,
link=True,
export="",
)
def test_display_defi_vaults_empty_df(mocker):
# MOCK GET_DEFI_VAULTS
mocker.patch(
target="gamestonk_terminal.cryptocurrency.defi.coindix_view.coindix_model.get_defi_vaults",
return_value=pd.DataFrame(),
)
coindix_view.display_defi_vaults(
chain=None,
protocol=None,
kind=None,
top=None,
sortby="apy",
descend=False,
link=True,
export="",
)
def call_vaults(self, other_args: List[str]):
"""Process swaps command"""
parser = argparse.ArgumentParser(
add_help=False,
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
prog="vaults",
description="""
Display Top DeFi Vaults.
[Source: https://coindix.com/]
""",
)
parser.add_argument(
"-c",
"--chain",
dest="chain",
type=str,
help="Blockchain name e.g. ethereum, terra",
default=None,
choices=coindix_model.CHAINS,
required=False,
)
parser.add_argument(
"-p",
"--protocol",
dest="protocol",
type=str,
help="DeFi protocol name e.g. aave, uniswap",
default=None,
choices=coindix_model.PROTOCOLS,
required=False,
)
parser.add_argument(
"-k",
"--kind",
dest="kind",
type=str,
help="Kind/type of vault e.g. lp, single, noimploss, stable",
default=None,
choices=coindix_model.VAULT_KINDS,
required=False,
)
parser.add_argument(
"-t",
"--top",
dest="limit",
type=check_positive,
help="Number of records to display",
default=10,
)
parser.add_argument(
"-s",
"--sort",
dest="sortby",
type=str,
help="Sort by given column. Default: timestamp",
default="apy",
choices=coindix_model.VAULTS_FILTERS,
)
parser.add_argument(
"--descend",
action="store_false",
help="Flag to sort in descending order (lowest first)",
dest="descend",
default=False,
)
parser.add_argument(
"-l",
"--links",
action="store_false",
help="Flag to show vault link",
dest="link",
default=True,
)
ns_parser = parse_known_args_and_warn(parser, other_args,
EXPORT_ONLY_RAW_DATA_ALLOWED)
if ns_parser:
coindix_view.display_defi_vaults(
chain=ns_parser.chain,
kind=ns_parser.kind,
protocol=ns_parser.protocol,
top=ns_parser.limit,
sortby=ns_parser.sortby,
descend=ns_parser.descend,
link=ns_parser.link,
export=ns_parser.export,
)