def test_load_roles_lookup_incomplete_row(self):
ra = RadiusAuth()
role_map = ra.loadRolesMap(file_path=os.path.join("test_role_map", "roles_map_incomplete_row.csv"))
self.assertEqual(len(role_map), 1)
self.assertTrue('bobama' in role_map)
def test_load_roles_lookup_filtered(self):
ra = RadiusAuth()
role_map = ra.loadRolesMap(file_path=os.path.join("test_role_map", "roles_map.csv"), username="******")
self.assertEqual(len(role_map), 1)
self.assertEqual(len(role_map['jdoe']), 3)
self.assertEqual(role_map['jdoe'], ['admin', 'power', 'user'])
def test_load_roles_lookup_caps_insensitive(self):
ra = RadiusAuth()
role_map = ra.loadRolesMap(file_path=os.path.join("test_role_map", "roles_map_caps.csv"))
self.assertEqual(len(role_map), 2)
self.assertEqual(len(role_map['jdoe']), 3)
self.assertEqual(role_map['jdoe'], ['admin', 'power', 'user'])
def test_load_roles_invalid_lookup(self):
bad_file = None
try:
# Create the junk file
bad_file = tempfile.NamedTemporaryFile(delete=False, suffix="_roles_map.csv")
# Write out some junk in the directory and see if it handled well (we will write a CSV since it isn't valid JSON)
bad_file.write(b"This is just some stuff that isn't a valid CSV file")
bad_file.close()
# Try to load the file
ra = RadiusAuth()
role_map = ra.loadRolesMap(file_path=bad_file.name)
self.assertEqual(role_map, {})
finally:
if bad_file is not None:
os.remove(bad_file.name)
def test_load_roles_lookup_no_file(self):
ra = RadiusAuth()
role_map = ra.loadRolesMap(file_path=os.path.join("non_existent_path", "roles_map.csv"))
self.assertEqual(role_map, None)
