def test_11_get_user(self, mock_file, mock_host):
"""Test11 NixAuthentication().get_user()."""
auth = NixAuthentication()
auth.passwd_file = ""
auth.get_user("user")
self.assertTrue(mock_host.called)
self.assertFalse(mock_file.called)
mock_host.reset_mock()
mock_file.reset_mock()
auth = NixAuthentication()
auth.passwd_file = "passwd"
auth.get_user("user")
self.assertFalse(mock_host.called)
self.assertTrue(mock_file.called)
def test_04_user_in_subgid(self, mock_file, mock_host):
"""Test04 NixAuthentication().user_in_subgid."""
mock_host.return_value = ("user", "1000", "1000", "usr", "/home/user",
"/bin/bash")
auth = NixAuthentication()
auth.subuid_file = "/etc/subgid"
subgid_line = StringIO('user:100000:65536')
with patch(BUILTINS + '.open') as mopen:
mopen.return_value.__iter__ = (
lambda self: iter(subgid_line.readline, ''))
listuser = auth.user_in_subgid("user")
self.assertEqual(listuser, [("100000", "65536")])
self.assertTrue(mock_host.called)
mock_file.return_value = ("user", "1000", "1000", "usr", "/home/user",
"/bin/bash")
auth = NixAuthentication()
auth.passwd_file = "/etc/passwd"
auth.subgid_file = "/etc/subgid"
subgid_line = StringIO('user:100000:65536')
with patch(BUILTINS + '.open') as mopen:
mopen.return_value.__iter__ = (
lambda self: iter(subgid_line.readline, ''))
listuser = auth.user_in_subgid("user")
self.assertEqual(listuser, [("100000", "65536")])
self.assertTrue(mock_file.called)
def test_07__get_user_from_file(self):
"""Test07 NixAuthentication()._get_user_from_file()."""
auth = NixAuthentication()
auth.passwd_file = "passwd"
passwd_line = StringIO('root:x:0:0:root:/root:/bin/bash')
with patch(BUILTINS + '.open') as mopen:
mopen.return_value.__iter__ = (
lambda self: iter(passwd_line.readline, ''))
(name, uid, gid, gecos, _dir,
shell) = auth._get_user_from_file("root")
self.assertEqual(name, "root")
self.assertEqual(uid, "0")
self.assertEqual(gid, "0")
self.assertEqual(gecos, "root")
self.assertEqual(_dir, "/root")
self.assertEqual(shell, "/bin/bash")
passwd_line = StringIO('root:x:0:0:root:/root:/bin/bash')
auth = NixAuthentication()
with patch(BUILTINS + '.open') as mopen:
mopen.return_value.__iter__ = (
lambda self: iter(passwd_line.readline, ''))
(name, uid, gid, gecos, _dir, shell) = auth._get_user_from_file(0)
self.assertEqual(name, "root")
self.assertEqual(uid, "0")
self.assertEqual(gid, "0")
self.assertEqual(gecos, "root")
self.assertEqual(_dir, "/root")
self.assertEqual(shell, "/bin/bash")
def test_08__get_group_from_file(self):
"""Test08 NixAuthentication()._get_group_from_file()."""
auth = NixAuthentication()
auth.passwd_file = "passwd"
group_line = StringIO('root:x:0:a,b,c')
with patch(BUILTINS + '.open') as mopen:
mopen.return_value.__iter__ = (
lambda self: iter(group_line.readline, ''))
(name, gid, mem) = auth._get_group_from_file("root")
self.assertEqual(name, "root")
self.assertEqual(gid, "0")
self.assertEqual(mem, "a,b,c")
group_line = StringIO('root:x:0:a,b,c')
auth = NixAuthentication()
with patch(BUILTINS + '.open') as mopen:
mopen.return_value.__iter__ = (
lambda self: iter(group_line.readline, ''))
(name, gid, mem) = auth._get_group_from_file(0)
self.assertEqual(name, "root")
self.assertEqual(gid, "0")
self.assertEqual(mem, "a,b,c")