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 4bbff6389a588450f4664e8520799aae065ef194..79b38377ddc67e958b7e86e5e72f8883f0e8b88a 100644 (file)
--- 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