add create_token action test

StackStorm-Exchange · Jan 27, 2021 · 9fb8ab5 · 9fb8ab5
1 parent f4e05cf
commit 9fb8ab5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/test_action_create_token.py b/tests/test_action_create_token.py
@@ -5,6 +5,10 @@
 class CreateTokenActionTestCase(VaultActionTestCase):
     action_cls = VaultCreateTokenAction
 
+    # Use None here to avoid this error:
+    #   "expiring root tokens cannot create non-expiring root tokens"
+    default_token_lease = None
+
     def test_method(self):
         action = self.get_action_instance(config=self.dummy_pack_config)
         result = action.run(
@@ -19,3 +23,4 @@ def test_method(self):
             # orphan=False,
             # wrap_ttl=None,
         )
+        assert result["auth"]["client_token"]
diff --git a/tests/vault_action_tests_base.py b/tests/vault_action_tests_base.py
@@ -8,6 +8,7 @@ class VaultActionTestCase(HvacIntegrationTestCase, BaseActionTestCase):
     dummy_pack_config = None
     secret_v1 = False
     secret_v2 = False
+    default_token_lease = "1h"
 
     # in setUp() and tearDown(), explicitly handle super()-like calls.
     #  - HvacIntegrationTestCase does not call super().
@@ -56,7 +57,7 @@ def build_dummy_pack_config(self, url="https://localhost:8200"):
         # based on create_client() in hvac/tests/utils/__init__.py
         server_cert_path = get_config_file_path("server-cert.pem")
 
-        token_result = self.client.create_token(lease="1h")
+        token_result = self.client.create_token(lease=self.default_token_lease)
         token = token_result["auth"]["client_token"]
 
         dummy_pack_config = {