class EncryptedDirectoryTest(unittest.TestCase):
def setUp(self):
if not os.path.isdir(TEST_DIR):
os.makedirs(TEST_DIR)
self.encdir = EncryptedDirectory(TEST_DIR, DummyKeypair())
def tearDown(self):
shutil.rmtree(TEST_DIR)
def test_encrypted_file_path(self):
generated_path = self.encdir.encrypted_file_path('dev',
'secrets1',
ext='yaml')
self.assertEqual(
os.path.join(TEST_DIR, 'dev', 'dummy12345-secrets1.yaml.smime'),
generated_path)
def test_read_file(self):
path = self.encdir.encrypted_file_path('dev', 'secrets2', ext='yaml')
os.makedirs(os.path.dirname(path))
with open(path, 'wb') as fp:
fp.write(b"E:some secret data")
plaintext = self.encdir.read_file('dev', 'secrets2', ext='yaml')
self.assertEqual(plaintext, b"some secret data")
def test_read_yaml_file(self):
path = self.encdir.encrypted_file_path('dev', 'secrets3', ext='yaml')
os.makedirs(os.path.dirname(path))
with open(path, 'wb') as fp:
fp.write(b'E:rootKey:\n')
fp.write(b' subKey1: value\n')
fp.write(b' subKey2: 123\n')
plainobj = self.encdir.read_yaml_file('dev', 'secrets3', ext='yaml')
self.assertEqual(plainobj,
{'rootKey': {
'subKey1': 'value',
'subKey2': 123
}})
def test_write_file(self):
path = self.encdir.encrypted_file_path('dev', 'secrets4', 'yaml')
contents = b"another secret datum"
self.encdir.write_file(BytesIO(contents),
'dev',
'secrets4',
ext='yaml')
with open(path, 'rb') as fp:
self.assertEqual(fp.read(), b"E:" + contents)
class EncryptedDirectoryTest(unittest.TestCase):
def setUp(self):
if not os.path.isdir(TEST_DIR):
os.makedirs(TEST_DIR)
self.encdir = EncryptedDirectory(TEST_DIR, DummyKeypair())
def tearDown(self):
shutil.rmtree(TEST_DIR)
def test_encrypted_file_path(self):
generated_path = self.encdir.encrypted_file_path('dev', 'secrets1', ext='yaml')
self.assertEqual(
os.path.join(TEST_DIR, 'dev', 'dummy12345-secrets1.yaml.smime'),
generated_path)
def test_read_file(self):
path = self.encdir.encrypted_file_path('dev', 'secrets2', ext='yaml')
os.makedirs(os.path.dirname(path))
with open(path, 'w') as fp:
fp.write("E:some secret data")
plaintext = self.encdir.read_file('dev', 'secrets2', ext='yaml')
self.assertEqual(plaintext, "some secret data")
def test_read_yaml_file(self):
path = self.encdir.encrypted_file_path('dev', 'secrets3', ext='yaml')
os.makedirs(os.path.dirname(path))
with open(path, 'w') as fp:
fp.write('E:rootKey:\n')
fp.write(' subKey1: value\n')
fp.write(' subKey2: 123\n')
plainobj = self.encdir.read_yaml_file('dev', 'secrets3', ext='yaml')
self.assertEqual(
plainobj,
{'rootKey': {'subKey1': 'value', 'subKey2': 123}})
def test_write_file(self):
path = self.encdir.encrypted_file_path('dev', 'secrets4', 'yaml')
contents = "another secret datum"
self.encdir.write_file(StringIO(contents), 'dev', 'secrets4', ext='yaml')
with open(path) as fp:
self.assertEqual(fp.read(), "E:" + contents)
def setUp(self):
if not os.path.isdir(TEST_DIR):
os.makedirs(TEST_DIR)
self.encdir = EncryptedDirectory(TEST_DIR, DummyKeypair())
def setUp(self):
if not os.path.isdir(TEST_DIR):
os.makedirs(TEST_DIR)
self.encdir = EncryptedDirectory(TEST_DIR, DummyKeypair())