def load_services(CONF_DIRS, config_data): """If we can't find a service file, load the default services. This is handy the first time we boot the system. """ for config_dir in reversed(CONF_DIRS): mykey_config = os.path.exists(os.path.expanduser( config_dir + mykey + ".dat")) if mykey_config: break if mykey_config: hosting = consuming = None else: service = "freedombuddy" listener_url = "https://localhost:{0}".format( utilities.safe_load( config_data,"https-listener", "socket_port", 8080)) monitor_url = "https://localhost:{0}".format( utilities.safe_load( config_data,"https-listener", "socket_port", 8081)) hosting = {mykey: {service: [listener_url], service + "-monitor" : [monitor_url + "/" + service]}} consuming = {mykey: {service: [listener_url], service + "-monitor" : [monitor_url + "/" + service]}} return hosting, consuming
def setUp(self): self.gpg = gnupg.GPG(homedir='src/tests/data/test_gpg_home') config = utilities.load_config("src/tests/data/test_gpg.cfg") self.key_id = utilities.safe_load(config, "pgpprocessor", "keyid", 0) self.recipient = "*****@*****.**" self.message = {'lol': 'cats'}
def setUp(self, *args, **kwargs): super(RoundTrip, self).setUp(*args, **kwargs) self.config = self.load_default_configs() self.mykey = utilities.safe_load(self.config, "general", "keyid") if not self.mykey: raise RuntimeError("No key configured for testing.")
def setUp(self): self.iterations = 3 self.gpg = gnupg.GPG(homedir='src/tests/data/test_gpg_home') config = utilities.load_config("src/tests/data/test_gpg.cfg") self.key_id = utilities.safe_load(config, "pgpprocessor", "keyid", 0) self.messages = utilities.multi_sign(message="hi", gpg=self.gpg, iterations=self.iterations, keyid=self.key_id)
def setUp(self): self.iterations = 3 self.gpg = gnupg.GPG(homedir='src/tests/data/test_gpg_home') config = utilities.load_config("src/tests/data/test_gpg.cfg") self.key_id = utilities.safe_load(config, "pgpprocessor", "keyid", 0) self.messages = utilities.multi_sign( message = "hi", gpg = self.gpg, iterations = self.iterations, keyid = self.key_id)
def load_connectors(protocols, config_data): """Loop through the protocols, loading settings for each used connector. We're combining disparate connector settings from the config file into a single dictionary of connectors. """ connectors = {} for protocol in protocols: protocol_connectors = listify_string( utilities.safe_load(config_data, protocol, "connectors")) if not protocol_connectors: continue for connector in protocol_connectors: connectors[connector] = dict( utilities.safe_load(config_data, connector, None, {})) return connectors
def setUp(self): super(ValidSubKeyButRevokedSuperKeyTest, self).setUp() self.key_id = utilities.safe_load(self.config, "pgpprocessor", "sub_keyid", 0) self.messages = utilities.multi_sign(message="hi", gpg=self.gpg_expired_then_valid, iterations=self.iterations, keyid=self.key_id) self.unwrapper = pgpprocessor.Unwrapper(self.messages[-1], self.gpg_expired)
def load_config(config): """Load data from the specified configuration file.""" listify_string = lambda x: [item.strip() for item in x.split(",")] config = utilities.load_config(config) mykey = utilities.safe_load(config, "pgpprocessor", "keyid", 0) protocols = listify_string( utilities.safe_load(config, "connectors", "protocols")) connectors = {} force_sender = utilities.safe_load(config, "connectors", "force_sender") # services to host and consume url = utilities.safe_load(config, "general", "url") if protocols == ['']: raise RuntimeError("No protocols detected. Have you run 'make'?") # loop through the protocols, finding connectors each protocol uses # load the settings for each connector. for protocol in protocols: protocol_connectors = listify_string( utilities.safe_load(config, protocol, "connectors")) if not protocol_connectors: continue for connector in protocol_connectors: connectors[connector] = dict( utilities.safe_load(config, connector, None, {})) return mykey, protocols, connectors, force_sender, url
def setUp(self): super(ValidSubKeyButRevokedSuperKeyTest, self).setUp() self.key_id = utilities.safe_load(self.config, "pgpprocessor", "sub_keyid", 0) self.messages = utilities.multi_sign( message = "hi", gpg = self.gpg_expired_then_valid, iterations = self.iterations, keyid = self.key_id) self.unwrapper = pgpprocessor.Unwrapper(self.messages[-1], self.gpg_expired)
def test_default_value_returned_if_config_file_does_not_exist(self): self.config = utilities.load_config('d') self.assertEqual("test", utilities.safe_load(self.config,"pgpprocessor","keyid","test"))
def test_default_value_returned_from_config_file_with_incorrect_section(self): self.config = utilities.load_config(self.config_to_use) self.assertEqual(None, utilities.safe_load(self.config,"incorrectly_named_section"))
def test_correct_section_returned_from_config_file(self): self.config = utilities.load_config(self.config_to_use) self.assertEqual([('keyid', '95801F1ABE01C28B05ADBE5FA7C860604DAE2628')], utilities.safe_load(self.config,"pgpprocessor"))
def test_correct_value_returned_from_config_file(self): self.config = utilities.load_config(self.config_to_use) self.assertEqual("95801F1ABE01C28B05ADBE5FA7C860604DAE2628", utilities.safe_load(self.config,"pgpprocessor","keyid"))
def test_correct_section_returned_from_config_file(self): self.config = utilities.load_config(self.config_to_use) self.assertEqual( [('keyid', '95801F1ABE01C28B05ADBE5FA7C860604DAE2628')], utilities.safe_load(self.config, "pgpprocessor"))
def test_correct_value_returned_from_config_file(self): self.config = utilities.load_config(self.config_to_use) self.assertEqual( "95801F1ABE01C28B05ADBE5FA7C860604DAE2628", utilities.safe_load(self.config, "pgpprocessor", "keyid"))
def test_None_returned_if_config_file_does_not_exist_and_default_value_not_set(self): self.config = utilities.load_configs('d') self.assertEqual(None, utilities.safe_load(self.config,"pgpprocessor","keyid"))
def test_default_value_returned_if_config_file_does_not_exist(self): self.config = utilities.load_config('d') self.assertEqual( "test", utilities.safe_load(self.config, "pgpprocessor", "keyid", "test"))
def test_default_value_returned_from_config_file_with_incorrect_section( self): self.config = utilities.load_config(self.config_to_use) self.assertEqual( None, utilities.safe_load(self.config, "incorrectly_named_section"))