From: Walter Dörwald <walter@livinglogic.de>
Date: Tue, 15 Apr 2003 15:39:08 +0000 (+0000)
Subject: Fix the test so that it works even when /etc/passwd has two entries
X-Git-Tag: v2.3c1~1184
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=66e1e508b98f5538066a3e4196b83eb5b6904269;p=python

Fix the test so that it works even when /etc/passwd has two entries
for the same uid.
---

diff --git a/Lib/test/test_pwd.py b/Lib/test/test_pwd.py
index 78f9a695f5..6cc52d7d0a 100644
--- a/Lib/test/test_pwd.py
+++ b/Lib/test/test_pwd.py
@@ -7,6 +7,7 @@ class PwdTest(unittest.TestCase):
 
     def test_values(self):
         entries = pwd.getpwall()
+        entriesbyuid = {}
 
         for e in entries:
             self.assertEqual(len(e), 7)
@@ -26,7 +27,17 @@ class PwdTest(unittest.TestCase):
             self.assert_(isinstance(e.pw_shell, basestring))
 
             self.assertEqual(pwd.getpwnam(e.pw_name), e)
-            self.assertEqual(pwd.getpwuid(e.pw_uid), e)
+            # The following won't work, because of duplicate entries
+            # for one uid
+            #    self.assertEqual(pwd.getpwuid(e.pw_uid), e)
+            # instead of this collect all entries for one uid
+            # and check afterwards
+            entriesbyuid.setdefault(e.pw_uid, []).append(e)
+
+        # check whether the entry returned by getpwuid()
+        # for each uid is among those from getpwall() for this uid
+        for e in entries:
+            self.assert_(pwd.getpwuid(e.pw_uid) in entriesbyuid[e.pw_uid])
 
     def test_errors(self):
         self.assertRaises(TypeError, pwd.getpwuid)