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_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 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)