From: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Date: Sun, 17 Aug 2025 09:59:07 +0000 (+0000)
Subject: pytest: run user_keytrust tests as computer keytrust tests
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=624a8c2261cfcb84e0080b19c2a6bb48f8c40750;p=thirdparty%2Fsamba.git

pytest: run user_keytrust tests as computer keytrust tests

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Gary Lockyer <gary@catalyst.net.nz>
---

diff --git a/python/samba/tests/samba_tool/user_keytrust.py b/python/samba/tests/samba_tool/user_keytrust.py
index 53262234d2e..1ba879c50b5 100644
--- a/python/samba/tests/samba_tool/user_keytrust.py
+++ b/python/samba/tests/samba_tool/user_keytrust.py
@@ -364,3 +364,19 @@ class SambaToolUserKeyTrustTest(SambaToolCmdTest):
                                        self.user2, GOOD_CERTS[0])
         self.assertCmdSuccess(result, out, err)
         self.assertEqual(len(self.get_links(self.user2)), 2)
+
+
+class SambaToolComputesrKeyTrustTest(SambaToolUserKeyTrustTest):
+    cmd = "computer"
+    user1 = 'ADDC'
+    user2 = 'ADDC'
+    ou_name = 'OU=Domain Controllers'
+
+    def get_links(self, username):
+        if username[-1] != '$':
+            username += '$'
+        result = self.samdb.search(expression=f'sAMAccountName={username}',
+                                   attrs=['msDS-KeyCredentialLink'])
+        self.assertEqual(len(result), 1)
+        links = result[0].get('msDS-KeyCredentialLink', [])
+        return [kcl.KeyCredentialLinkDn(self.samdb, v) for v in links]