def test_create_ssh_wrapper(self):
self.assertEqual(check_data_writable(), [])
filename = get_wrapper_filename()
create_ssh_wrapper()
with open(filename, 'r') as handle:
data = handle.read()
self.assertTrue(ssh_file('known_hosts') in data)
self.assertTrue(ssh_file('id_rsa') in data)
self.assertTrue(settings.DATA_DIR in data)
self.assertTrue(os.access(filename, os.X_OK))
# Second run should not touch the file
timestamp = os.stat(filename).st_mtime
create_ssh_wrapper()
self.assertEqual(timestamp, os.stat(filename).st_mtime)
def test_create_ssh_wrapper(self):
self.assertEqual(check_data_writable(), [])
wrapper = SSHWrapper()
filename = wrapper.filename
wrapper.create()
with open(filename) as handle:
data = handle.read()
self.assertTrue(ssh_file("known_hosts") in data)
self.assertTrue(ssh_file("id_rsa") in data)
self.assertTrue(settings.DATA_DIR in data)
self.assertTrue(os.access(filename, os.X_OK))
# Second run should not touch the file
timestamp = os.stat(filename).st_mtime
wrapper.create()
self.assertEqual(timestamp, os.stat(filename).st_mtime)
def ssh_key(request):
with open(ssh_file(RSA_KEY)) as handle:
data = handle.read()
response = HttpResponse(data, content_type="text/plain")
response["Content-Disposition"] = f"attachment; filename={RSA_KEY}"
response["Content-Length"] = len(data)
return response
def test_create_ssh_wrapper(self):
check_data_writable()
filename = get_wrapper_filename()
create_ssh_wrapper()
with open(filename, 'r') as handle:
data = handle.read()
self.assertTrue(ssh_file('known_hosts') in data)
self.assertTrue(ssh_file('id_rsa') in data)
self.assertTrue(settings.DATA_DIR in data)
self.assertTrue(
os.access(filename, os.X_OK)
)
# Second run should not touch the file
timestamp = os.stat(filename).st_mtime
create_ssh_wrapper()
self.assertEqual(timestamp, os.stat(filename).st_mtime)
def ssh_key(request):
with open(ssh_file(RSA_KEY), 'r') as handle:
data = handle.read()
response = HttpResponse(data, content_type='text/plain')
response['Content-Disposition'] = 'attachment; filename={0}'.format(
RSA_KEY)
response['Content-Length'] = len(data)
return response
