fix tests

GreyNoise-Intelligence · Mar 30, 2021 · e1fddb0 · e1fddb0
1 parent b7dc8dd
commit e1fddb0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/greynoise/cli/decorator.py b/src/greynoise/cli/decorator.py
@@ -116,7 +116,10 @@ def wrapper(*args, **kwargs):
                 offering = config["offering"]
 
         api_client = GreyNoise(
-            api_key=api_key, offering=offering, timeout=config["timeout"], integration_name="cli"
+            api_key=api_key,
+            offering=offering,
+            timeout=config["timeout"],
+            integration_name="cli",
         )
         return function(api_client, *args, **kwargs)
 

diff --git a/tests/cli/test_subcommand.py b/tests/cli/test_subcommand.py
@@ -27,6 +27,7 @@ def api_client():
             "api_key": "<api_key>",
             "api_server": "<api_server>",
             "timeout": DEFAULT_CONFIG["timeout"],
+            "offering": "enterprise"
         }
         with api_client_cls_patcher as api_client_cls:
             api_client = api_client_cls()