def test_read_failure(self, mock_read):
key_name = "my_booth.key"
mock_read.side_effect = RawFileError("mock file type",
RawFileError.ACTION_READ,
"some error")
path = os.path.join(settings.booth_config_dir, key_name)
conf = ConfigFacade.create([], [])
conf.set_authfile(path)
with self.assertRaises(RawFileError):
config_files.get_authfile_name_and_data(conf)
def test_success(self, mock_read):
key_data = "some key data".encode("utf-8")
key_name = "my_booth.key"
mock_read.return_value = key_data
path = os.path.join(settings.booth_config_dir, key_name)
conf = ConfigFacade.create([], [])
conf.set_authfile(path)
self.assertEqual(
(key_name, key_data, []),
config_files.get_authfile_name_and_data(conf),
)
def assert_path_check(self, path):
conf = ConfigFacade.create([], [])
conf.set_authfile(path)
result = config_files.get_authfile_name_and_data(conf)
self.assertEqual(result[0], None)
self.assertEqual(result[1], None)
assert_report_item_list_equal(
result[2],
[
fixture.warn(
report_codes.BOOTH_UNSUPPORTED_FILE_LOCATION,
file_type_code=file_type_codes.BOOTH_KEY,
file_path=path,
expected_dir=settings.booth_config_dir,
),
],
)
def test_no_auth_file(self):
conf = ConfigFacade.create([], [])
self.assertEqual((None, None, []),
config_files.get_authfile_name_and_data(conf))
def test_no_addrs(self):
sites = []
arbitrators = []
conf = ConfigFacade.create(sites, arbitrators)
self.assertEqual(sites, conf.get_sites())
self.assertEqual(arbitrators, conf.get_arbitrators())
def test_both(self):
sites = ["1.1.1.1", "2.2.2.2", "3.3.3.3"]
arbitrators = ["4.4.4.4", "5.5.5.5"]
conf = ConfigFacade.create(sites, arbitrators)
self.assertEqual(sites, conf.get_sites())
self.assertEqual(arbitrators, conf.get_arbitrators())
def test_arbitrators(self):
sites = []
arbitrators = ["4.4.4.4", "5.5.5.5"]
conf = ConfigFacade.create(sites, arbitrators)
self.assertEqual(sites, conf.get_sites())
self.assertEqual(arbitrators, conf.get_arbitrators())
