misc: Add test for supplementary groups

Signed-off-by: Sachin Prabhu <[email protected]>

Author: spuiuk

testcases/misc/test_supplemantary_group.py

@@ -0,0 +1,91 @@
+# This test is to check writes as part of a supplementary group.
+#
+# Requirements:
+# 1. user1 exists
+# 2. user1 is part of supplementary group sg
+#
+# Steps:
+# 1. Create folder test_subdir/ with group "sg" and mode 0770
+# 2. Upload file to test_subdir/test-cp
+#
+# Expected:
+# Copy passes
+
+import testhelper
+import pytest
+import shutil
+import pwd
+import grp
+import os
+from pathlib import Path
+
+test_info_file = os.getenv("TEST_INFO_FILE")
+test_info = testhelper.read_yaml(test_info_file)
+test_string = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
+test_subdir = Path("supplementary_group")
+test_sgroup = "sg"
+
+
+def check_reqs():
+    try:
+        pwd.getpwnam("test1")
+        if "test1" not in grp.getgrnam(test_sgroup).gr_mem:
+            return False
+    except KeyError:
+        return False
+    return True
+
+
+def setup_local_testdir(root: Path, group: str) -> Path:
+    testdir = root / test_subdir
+    testdir.mkdir(exist_ok=True)
+    shutil.chown(testdir, group=group)
+    testdir.chmod(0o770)
+    return testdir
+
+
+def write_file_remote(
+    mount_point: Path,
+    ipaddr: str,
+    share_name: str,
+) -> None:
+    mount_params = testhelper.get_mount_parameters(test_info, share_name)
+    mount_params["host"] = ipaddr
+    try:
+        test_file = testhelper.get_tmp_file(mount_point)
+        test_file_remote = test_subdir / Path("test-cp")
+        with open(test_file, "w") as f:
+            f.write(test_string)
+        put_cmds = "put  %s %s" % (test_file, test_file_remote)
+        (ret, output) = testhelper.smbclient(mount_params, put_cmds)
+        assert ret == 0, "Failed to copy file to server: " + output
+    finally:
+        if test_file.exists():
+            test_file.unlink()
+
+
+def gen_supplementary_group_param(test_info: dict) -> list:
+    if not check_reqs():
+        return []
+    if not test_info:
+        return []
+    arr = []
+    for ipaddr in test_info["public_interfaces"]:
+        for share in testhelper.get_shares_with_directmnt(test_info):
+            arr.append((ipaddr, share["name"]))
+    return arr
+
+
+@pytest.mark.parametrize(
+    "ipaddr,share_name", gen_supplementary_group_param(test_info)
+)
+def test_supplementary_group(ipaddr: str, share_name: str) -> None:
+    share = testhelper.get_share(test_info, share_name)
+    fs_path = Path(share["directmnt"])
+    testdir = setup_local_testdir(fs_path, test_sgroup)
+    try:
+        tmp_root = testhelper.get_tmp_root()
+        mount_point = testhelper.get_tmp_mount_point(tmp_root)
+        write_file_remote(mount_point, ipaddr, share_name)
+    finally:
+        shutil.rmtree(testdir)