def __init__(self, agent_conf_file, access_conf_file): self.agent_entries = snmp_config.AgentConfig(agent_conf_file).entries self.access_entries = snmp_config.UserConfig(access_conf_file).entries utils.setup_log( log_file_path=self.agent_entries.default_section.log_file, level=self.agent_entries.default_section.log_level, max_bytes=self.agent_entries.default_section.log_file_maxbytes, max_file_count=self.agent_entries.default_section.log_file_count) self.transport_dispatcher = None
def test_agent_config_save_reload(self, conf_file): def _read_passwd(conf_file): pattern = ('password='******'password = '******'r') as f: return [ line.split('=')[1].strip() for line in f.readlines() if line.startswith(pattern) ] conf = config.AgentConfig(conf_file) conf.save() encrypt_passwds = _read_passwd(conf_file) self.assertEqual(len(encrypt_passwds), 2) for passwd in encrypt_passwds: self.assertTrue(cipher.is_encrypted(passwd)) self.test_agent_config(conf_file)
def test_agent_config(self, conf_file): conf = config.AgentConfig(conf_file) self.assertEqual(2, len(conf.entries)) for name, entry in conf.entries.items(): self.assertEqual('0.0.0.0', entry.agent_ip) if name == 'unity-1': self.assertEqual('11161', entry.agent_port) self.assertEqual('unity', entry.model) self.assertEqual('10.10.10.21', entry.mgmt_ip) self.assertEqual('admin', entry.user) self.assertEqual('Password21', entry.password.raw) elif name == 'unity-2': self.assertEqual('11162', entry.agent_port) self.assertEqual('unity', entry.model) self.assertEqual('10.10.10.22', entry.mgmt_ip) self.assertEqual('admin', entry.user) self.assertEqual('Password22!', entry.password.raw) else: self.assertFalse(True)
def validate_conf(cls, agent_conf_file): access_conf_file = access.get_access_data_path() snmp_config.AgentConfig(agent_conf_file).raise_if_error() snmp_config.UserConfig(access_conf_file).raise_if_error()
def do(self): agent_config = config.AgentConfig(self.args['--conf_file']) agent_config.save(encrypt=False) LOG.info("Decrypted config file '{}' successfully.".format( self.args['--conf_file']))