From: Rob van der Linde <rob@catalyst.net.nz>
Date: Tue, 30 Apr 2024 11:54:13 +0000 (+1200)
Subject: selftest: add test for User.get_primary_group method
X-Git-Tag: tdb-1.4.11~804
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=5aafd2fb4042a2f787dd6a7bc9584494d29c405f;p=thirdparty%2Fsamba.git

selftest: add test for User.get_primary_group method

Signed-off-by: Rob van der Linde <rob@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
---

diff --git a/python/samba/tests/samba_tool/domain_models.py b/python/samba/tests/samba_tool/domain_models.py
index 4bbff6389a5..79b38377ddc 100644
--- a/python/samba/tests/samba_tool/domain_models.py
+++ b/python/samba/tests/samba_tool/domain_models.py
@@ -116,6 +116,21 @@ class ModelTests(SambaToolCmdTest):
         self.assertIn("whenCreated", silo_dict)
 
 
+class UserModelTests(SambaToolCmdTest):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.samdb = cls.getSamDB("-H", HOST, CREDS)
+        super().setUpClass()
+
+    def test_get_primary_group(self):
+        jane = User.get(self.samdb, account_name="jane")
+        domain_sid = self.samdb.domain_sid
+        expected_group = Group.get(self.samdb,
+                                   object_sid=f"{domain_sid}-{jane.primary_group_id}")
+        self.assertEqual(jane.get_primary_group(self.samdb), expected_group)
+
+
 class ComputerModelTests(SambaToolCmdTest):
 
     @classmethod