Query all usernames and group names

Author: CarstenGrohmann

test/test_modules.py

@@ -272,6 +272,13 @@ def test_user(host):
     assert user.password == "*"
 
 
+def test_user_get_all_users(host):
+    user_list = host.user("root").get_all_users
+    assert "root" in user_list
+    assert "man" in user_list
+    assert "nobody" in user_list
+
+
 def test_user_password_days(host):
     assert host.user("root").password_max_days == 99999
     assert host.user("root").password_min_days == 0
@@ -303,6 +310,12 @@ def test_current_user(host):
 def test_group(host):
     assert host.group("root").exists
     assert host.group("root").gid == 0
+    group_list = host.group("root").get_all_groups
+    assert "root" in group_list
+    assert "bin" in group_list
+    group_list = host.group("root").get_local_groups
+    assert "root" in group_list
+    assert "bin" in group_list
 
 
 def test_empty_command_output(host):

testinfra/modules/group.py

@@ -33,6 +33,32 @@ def exists(self):
 
         return self.run_expect([0, 2], "getent group %s", self.name).rc == 0
 
+    @property
+    def get_all_groups(self):
+        """Returns a list of local and remote group names
+
+        >>> host.group("anyname").get_all_groups
+        ["root", "wheel", "man", "tty", <...>]
+        """
+        all_groups = [
+            line.split(":")[0]
+            for line in self.check_output("getent group").splitlines()
+        ]
+        return all_groups
+
+    @property
+    def get_local_groups(self):
+        """Returns a list of local group names
+
+        >>> host.group("anyname").get_local_groups
+        ["root", "wheel", "man", "tty", <...>]
+        """
+        local_groups = [
+            line.split(":")[0]
+            for line in self.check_output("cat /etc/group").splitlines()
+        ]
+        return local_groups
+
     @property
     def gid(self):
         return int(self.check_output("getent group %s | cut -d':' -f3", self.name))

testinfra/modules/user.py

@@ -133,6 +133,34 @@ def expiration_date(self):
             epoch = datetime.datetime.utcfromtimestamp(0)
             return epoch + datetime.timedelta(days=int(days))
 
+    @property
+    def get_all_users(self):
+        """Returns a list of local and remote user names
+
+        >>> host.user().get_all_users
+        ["root", "bin", "daemon", "lp", <...>]
+        """
+        all_users = [
+            line.split(":")[0]
+            for line in self.check_output("getent passwd").splitlines()
+        ]
+        return all_users
+
+    @property
+    def get_local_users(self):
+        """Returns a list of local user names
+
+        >>> host.user().get_local_users
+        ["root", "bin", "daemon", "lp", <...>]
+        """
+        local_users = [
+            line.split(":")[0]
+            for line in self.check_output("cat /etc/passwd").splitlines()
+        ]
+        # strip NIS compat mode entries
+        local_users = [i for i in local_users if not i.startswith("+")]
+        return local_users
+
     @classmethod
     def get_module_class(cls, host):
         if host.system_info.type.endswith("bsd"):