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