def test_drop_all_privileges(self, *args): pwd.getpwuid.return_value.pw_dir = 'fakeusr' with EnvironmentVarGuardRestore(): os.environ['PKEXEC_UID'] = '1000' misc.drop_all_privileges() os.setreuid.assert_called_once_with(1000, 1000) os.setregid.assert_called_once_with(1000, 1000) os.setgroups.assert_called_once_with([1234]) self.assertEqual(os.environ['HOME'], 'fakeusr')
def subprocess_setup(): misc.drop_all_privileges() signal.signal(signal.SIGINT, signal.SIG_DFL) signal.signal(signal.SIGPIPE, signal.SIG_DFL) os.setsid()