From: Joseph Sutton <josephsutton@catalyst.net.nz>
Date: Tue, 15 Jun 2021 05:10:44 +0000 (+1200)
Subject: tests/krb5/raw_testcase.py: Cache obtained credentials
X-Git-Tag: tevent-0.11.0~65
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=22a90aea82ba6ef86bde835f2369daa6e23ed2fd;p=thirdparty%2Fsamba.git

tests/krb5/raw_testcase.py: Cache obtained credentials

If credentials are used more than once, we can now use the credentials
that we already obtained and so avoid fetching them again.

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Stefan Metzmacher <metze@samba.org>
---

diff --git a/python/samba/tests/krb5/kdc_base_test.py b/python/samba/tests/krb5/kdc_base_test.py
index 7ae22bc5929..120084616e9 100644
--- a/python/samba/tests/krb5/kdc_base_test.py
+++ b/python/samba/tests/krb5/kdc_base_test.py
@@ -75,6 +75,7 @@ class KDCBaseTest(RawKerberosTest):
 
     @classmethod
     def setUpClass(cls):
+        super().setUpClass()
         cls._lp = None
         cls.host = os.environ["SERVER"]
 
diff --git a/python/samba/tests/krb5/raw_testcase.py b/python/samba/tests/krb5/raw_testcase.py
index 9c0f5800b42..5b59eede806 100644
--- a/python/samba/tests/krb5/raw_testcase.py
+++ b/python/samba/tests/krb5/raw_testcase.py
@@ -371,6 +371,14 @@ class RawKerberosTest(TestCaseInTempDir):
         e = self.etype_test_permutations[idx]
         return (e['name'], e['etypes'])
 
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+
+        # A dictionary containing credentials that have already been
+        # obtained.
+        cls.creds_dict = {}
+
     def setUp(self):
         super().setUp()
         self.do_asn1_print = False
@@ -441,11 +449,11 @@ class RawKerberosTest(TestCaseInTempDir):
                                                 allow_missing=allow_missing)
         return val
 
-    def _get_krb5_creds(self, prefix,
-                        default_username=None,
-                        allow_missing_password=False,
-                        allow_missing_keys=True,
-                        require_strongest_key=False):
+    def _get_krb5_creds_from_env(self, prefix,
+                                 default_username=None,
+                                 allow_missing_password=False,
+                                 allow_missing_keys=True,
+                                 require_strongest_key=False):
         c = KerberosCredentials()
         c.guess()
 
@@ -515,6 +523,26 @@ class RawKerberosTest(TestCaseInTempDir):
 
         return c
 
+    def _get_krb5_creds(self,
+                        prefix,
+                        default_username=None,
+                        allow_missing_password=False,
+                        allow_missing_keys=True,
+                        require_strongest_key=False):
+        if prefix not in self.creds_dict:
+            # We don't have the credentials already
+            creds = self._get_krb5_creds_from_env(prefix,
+                                                  default_username=default_username,
+                                                  allow_missing_password=allow_missing_password,
+                                                  allow_missing_keys=allow_missing_keys,
+                                                  require_strongest_key=require_strongest_key)
+            self.assertIsNotNone(creds)
+
+            # Save the obtained credentials
+            self.creds_dict[prefix] = creds
+
+        return self.creds_dict[prefix]
+
     def get_user_creds(self,
                        allow_missing_password=False,
                        allow_missing_keys=True):