@@ -23,29 +23,45 @@ def cli_runner() -> CliRunner:
2323
2424
2525@pytest .fixture
26- def with_cli_env_vars () -> Generator :
26+ def with_spot_secrets () -> Generator :
2727 """Setup some environment variables for th CLI tests"""
2828
2929 if not all (
3030 (
3131 spot_api_key := os .getenv ("SPOT_API_KEY" ),
3232 spot_secret_key := os .getenv ("SPOT_SECRET_KEY" ),
33- futures_api_key := os .getenv ("FUTURES_API_KEY" ),
34- futures_secret_key := os .getenv ("FUTURES_SECRET_KEY" ),
3533 ),
3634 ):
3735 pytest .fail ("No API keys provided for CLI tests!" )
3836
3937 os .environ ["KRAKEN_SPOT_API_KEY" ] = spot_api_key
4038 os .environ ["KRAKEN_SPOT_SECRET_KEY" ] = spot_secret_key
39+
40+ yield
41+
42+ for var in ("KRAKEN_SPOT_API_KEY" , "KRAKEN_SPOT_SECRET_KEY" ):
43+ if os .getenv (var ):
44+ del os .environ [var ]
45+
46+
47+ @pytest .fixture
48+ def with_futures_secrets () -> Generator :
49+ """Setup some environment variables for the CLI tests"""
50+
51+ if not all (
52+ (
53+ futures_api_key := os .getenv ("FUTURES_API_KEY" ),
54+ futures_secret_key := os .getenv ("FUTURES_SECRET_KEY" ),
55+ ),
56+ ):
57+ pytest .fail ("No API keys provided for CLI tests!" )
58+
4159 os .environ ["KRAKEN_FUTURES_API_KEY" ] = futures_api_key
4260 os .environ ["KRAKEN_FUTURES_SECRET_KEY" ] = futures_secret_key
4361
4462 yield
4563
4664 for var in (
47- "KRAKEN_SPOT_API_KEY" ,
48- "KRAKEN_SPOT_SECRET_KEY" ,
4965 "KRAKEN_FUTURES_API_KEY" ,
5066 "KRAKEN_FUTURES_SECRET_KEY" ,
5167 ):
0 commit comments