From fabf0d1565ab7c9ab89f70ad98616c0161c4c4ff Mon Sep 17 00:00:00 2001 From: Stefan Metzmacher Date: Thu, 6 Feb 2025 16:47:30 +0100 Subject: [PATCH] python:tests/krb5: add remove_pac_buffers() Signed-off-by: Stefan Metzmacher Reviewed-by: Ralph Boehme --- python/samba/tests/krb5/kdc_base_test.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/python/samba/tests/krb5/kdc_base_test.py b/python/samba/tests/krb5/kdc_base_test.py index 23d2b4b799a..23577dccf12 100644 --- a/python/samba/tests/krb5/kdc_base_test.py +++ b/python/samba/tests/krb5/kdc_base_test.py @@ -2362,6 +2362,19 @@ class KDCBaseTest(TestCaseInTempDir, RawKerberosTest): return pac + def remove_pac_buffers(self, pac, *, pac_buffer_type=None): + + old_buffers = pac.buffers + new_buffers = [] + for pac_buffer in old_buffers: + if pac_buffer.type != pac_buffer_type: + new_buffers.append(pac_buffer) + + pac.buffers = new_buffers + pac.num_buffers = len(new_buffers) + + return pac + def get_cached_creds(self, *, account_type: AccountType, opts: Optional[dict]=None, -- 2.47.2