tests/krb5: Keep track of the type of each created account

This allows us to determine which parts of an authentication policy
apply to a particular account, which will be necessary to test audit
logging.

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

diff --git a/python/samba/tests/krb5/kdc_base_test.py b/python/samba/tests/krb5/kdc_base_test.py
index b8412cd57ff1a2d9f2998955c6ea28a100c72f9d..26b2dfbe655260e6cc07b5be0f07ff4b2ae8260c 100644 (file)
--- a/python/samba/tests/krb5/kdc_base_test.py
+++ b/python/samba/tests/krb5/kdc_base_test.py
@@ -879,6 +879,7 @@ class KDCBaseTest(TestCaseInTempDir, RawKerberosTest):
         creds.set_dn(ldb.Dn(samdb, dn))
         creds.set_upn(upn)
         creds.set_spn(spn)
+        creds.set_type(account_type)
 
         self.creds_set_enctypes(creds)
 

diff --git a/python/samba/tests/krb5/raw_testcase.py b/python/samba/tests/krb5/raw_testcase.py
index 889a29a38c8e6d33d713c05dbf563457db5e2b70..193fea80adc4e592906dbdbf2c58801bded818c5 100644 (file)
--- a/python/samba/tests/krb5/raw_testcase.py
+++ b/python/samba/tests/krb5/raw_testcase.py
@@ -396,6 +396,7 @@ class KerberosCredentials(Credentials):
         self.dn = None
         self.upn = None
         self.spn = None
+        self.account_type = None
 
     def set_as_supported_enctypes(self, value):
         self.as_supported_enctypes = int(value)
@@ -528,6 +529,12 @@ class KerberosCredentials(Credentials):
     def get_upn(self):
         return self.upn
 
+    def set_type(self, account_type):
+        self.account_type = account_type
+
+    def get_type(self):
+        return self.account_type
+
     def update_password(self, password):
         self.set_password(password)
         self.set_kvno(self.get_kvno() + 1)