def test_check_membership(self):
uc = self.config['tsi.user_cache']
user = self.getlogin()
gid = uc.get_gid_4user(user)
(group, _, _, _) = grp.getgrgid(gid)
result = BecomeUser.check_membership(group, gid, user, self.config)
self.assertTrue(result)
self.config['tsi.enforce_os_gids'] = False
other_gid = 1
result = BecomeUser.check_membership(group, other_gid, user,
self.config)
self.assertTrue(result)
def setUp(self):
# setup logger
self.LOG = logging.getLogger("tsi.testing")
self.LOG.setLevel(logging.INFO)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
ch.setFormatter(formatter)
self.LOG.handlers = [ch]
self.config = {'tsi.testing': True}
TSI.setup_defaults(self.config)
BecomeUser.initialize(self.config, self.LOG)
def test_get_groups(self):
uc = self.config['tsi.user_cache']
user = self.getlogin()
gid = uc.get_gid_4user(user)
(group, _, _, _) = grp.getgrgid(gid)
new_gid = BecomeUser.get_primary_group(group, user, uc, True,
self.config, self.LOG)
self.assertEqual(gid, new_gid)
groups = [group]
sup_gids = BecomeUser.get_supplementary_groups(groups, new_gid, user,
self.config, self.LOG)
print("Got groups: %s" % sup_gids)
groups = ["DEFAULT_GID"]
sup_gids = BecomeUser.get_supplementary_groups(groups, new_gid, user,
self.config, self.LOG)
print(sup_gids)
print("Got groups: %s" % sup_gids)
def test_PING2(self):
cwd = os.getcwd()
version = TSI.MY_VERSION
config = {'tsi.testing': True, 'tsi.enforce_os_gids': False}
config['tsi.use_id_to_resolve_gids'] = False
msg = """#TSI_PING_UID
#TSI_IDENTITY nobody NONE
ENDOFMESSAGE
"""
control_source = io.BufferedReader(io.BytesIO(msg.encode("UTF-8")))
control_in = io.TextIOWrapper(control_source)
control_out = io.StringIO()
BecomeUser.initialize(config, self.LOG)
connector = MockConnector.MockConnector(control_in, control_out, None,
None, self.LOG)
TSI.process(connector, config, self.LOG)
result = control_out.getvalue()
print(result)
self.assertTrue(version in result)
control_source.close()
os.chdir(cwd)
def test_become_user(self):
user = self.getlogin()
result = BecomeUser.become_user(user, ["DEFAULT_GID"], self.config,
self.LOG)
self.assertTrue(result)
BecomeUser.restore_id(self.config, self.LOG)
def setUp(self):
self.LOG = Log.Logger("tsi.testing", use_syslog=False)
self.config = {'tsi.testing': True, "tsi.switch_uid": False}
TSI.setup_defaults(self.config)
BecomeUser.initialize(self.config, self.LOG)