python/samba/tests/krb5/raw_testcase.py KerberosCredentials.self.forced_salt

diff --git a/python/samba/tests/krb5/raw_testcase.py b/python/samba/tests/krb5/raw_testcase.py
index 966d110ae27461406b78590103a8888721b845fb..fc96659674545401f997ed720679efdaeced1cf8 100644 (file)
--- a/python/samba/tests/krb5/raw_testcase.py
+++ b/python/samba/tests/krb5/raw_testcase.py
@@ -169,6 +169,8 @@ class KerberosCredentials(Credentials):
         self.as_supported_enctypes = all_enc_types
         self.tgs_supported_enctypes = all_enc_types
         self.ap_supported_enctypes = all_enc_types
+
+        self.forced_salt = None
         return
 
     def set_as_supported_enctypes(self, value):
@@ -204,6 +206,13 @@ class KerberosCredentials(Credentials):
     def get_ap_krb5_etypes(self):
         return self._get_krb5_etypes(self.ap_supported_enctypes)
 
+    def set_forced_salt(self, salt):
+        self.forced_salt = bytes(salt)
+        return
+
+    def get_forced_salt(self):
+        return self.forced_salt
+
 class RawKerberosTest(TestCase):
     """A raw Kerberos Test case."""