def test_keys(self):
engine = PlumberyEngine()
self.assertEqual(engine._sharedKeyFiles, [])
with self.assertRaises(ValueError):
engine.set_shared_key_files('this_does_not_exist')
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertEqual(engine._sharedKeyFiles,
engine.get_shared_key_files())
with mock.patch.object(engine, 'get_shared_key_files') as patched:
patched.return_value = []
files = ['*unknown-file*', '**me-too*']
with self.assertRaises(ValueError):
engine.set_shared_key_files(files)
file = os.path.abspath(
os.path.dirname(__file__))+'/fixtures/dummy_rsa.pub'
engine.set_shared_key_files(file)
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertEqual(engine.get_shared_key_files()[0], file)
self.assertEqual(engine._sharedKeyFiles,
engine.get_shared_key_files())
with mock.patch.object(engine, 'get_shared_key_files') as patched:
patched.return_value = []
files = [file, file, file]
engine.set_shared_key_files(files)
self.assertEqual(engine._sharedKeyFiles, [file])
if 'SHARED_KEY' in os.environ:
memory = os.environ["SHARED_KEY"]
else:
memory = None
engine._sharedKeyFiles = []
os.environ["SHARED_KEY"] = 'this_does_not_exist'
with self.assertRaises(ValueError):
engine.set_shared_key_files()
with self.assertRaises(ValueError):
engine.get_shared_key_files()
self.assertTrue(isinstance(engine._sharedKeyFiles, list))
engine._sharedKeyFiles = []
os.environ["SHARED_KEY"] = file
engine.set_shared_key_files()
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertEqual(engine.get_shared_key_files()[0], file)
self.assertEqual(engine._sharedKeyFiles,
engine.get_shared_key_files())
if memory is None:
os.environ.pop("SHARED_KEY")
else:
os.environ["SHARED_KEY"] = memory
with mock.patch.object(engine, 'get_shared_key_files') as patched:
patched.return_value = []
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
engine.set_shared_key_files()
self.assertTrue(plogging.foundErrors())
with self.assertRaises(ValueError):
engine.set_shared_key_files('this_does_not_exist')
file = os.path.abspath(
os.path.dirname(__file__))+'/fixtures/dummy_rsa.pub'
engine.set_shared_key_files(file)
self.assertEqual(engine._sharedKeyFiles, [file])
def test_keys(self):
engine = PlumberyEngine()
self.assertEqual(engine._sharedKeyFiles, [])
with self.assertRaises(ValueError):
engine.set_shared_key_files('this_does_not_exist')
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertEqual(engine._sharedKeyFiles, engine.get_shared_key_files())
with mock.patch.object(engine, 'get_shared_key_files') as patched:
patched.return_value = []
files = ['*unknown-file*', '**me-too*']
with self.assertRaises(ValueError):
engine.set_shared_key_files(files)
file = os.path.abspath(
os.path.dirname(__file__)) + '/fixtures/dummy_rsa.pub'
engine.set_shared_key_files(file)
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertEqual(engine.get_shared_key_files()[0], file)
self.assertEqual(engine._sharedKeyFiles, engine.get_shared_key_files())
with mock.patch.object(engine, 'get_shared_key_files') as patched:
patched.return_value = []
files = [file, file, file]
engine.set_shared_key_files(files)
self.assertEqual(engine._sharedKeyFiles, [file])
if 'SHARED_KEY' in os.environ:
memory = os.environ["SHARED_KEY"]
else:
memory = None
engine._sharedKeyFiles = []
os.environ["SHARED_KEY"] = 'this_does_not_exist'
with self.assertRaises(ValueError):
engine.set_shared_key_files()
with self.assertRaises(ValueError):
engine.get_shared_key_files()
self.assertTrue(isinstance(engine._sharedKeyFiles, list))
engine._sharedKeyFiles = []
os.environ["SHARED_KEY"] = file
engine.set_shared_key_files()
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertEqual(engine.get_shared_key_files()[0], file)
self.assertEqual(engine._sharedKeyFiles, engine.get_shared_key_files())
if memory is None:
os.environ.pop("SHARED_KEY")
else:
os.environ["SHARED_KEY"] = memory
with mock.patch.object(engine, 'get_shared_key_files') as patched:
patched.return_value = []
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
engine.set_shared_key_files()
self.assertTrue(plogging.foundErrors())
with self.assertRaises(ValueError):
engine.set_shared_key_files('this_does_not_exist')
file = os.path.abspath(
os.path.dirname(__file__)) + '/fixtures/dummy_rsa.pub'
engine.set_shared_key_files(file)
self.assertEqual(engine._sharedKeyFiles, [file])
