def test_public(self):
self.assertEqual(check_data_writable(), [])
# This will generate new key
key = get_gpg_public_key()
self.assertIsNotNone(key)
# Check cache access
self.assertEqual(key, get_gpg_public_key())
def test_generate(self):
self.assertEqual(check_data_writable(), [])
self.assertIsNone(get_gpg_key(silent=True))
key = generate_gpg_key()
self.check_errors()
self.assertIsNotNone(key)
self.assertEqual(key, get_gpg_key())
def test_public(self):
self.assertEqual(check_data_writable(), [])
# This will generate new key
key = get_gpg_public_key()
self.assertIsNotNone(key)
# Check cache access
self.assertEqual(key, get_gpg_public_key())
def test_ssh_generate(self):
self.assertEqual(check_data_writable(), [])
response = self.client.get(reverse('admin:ssh'))
self.assertContains(response, 'Generate SSH key')
response = self.client.post(reverse('admin:ssh'),
{'action': 'generate'})
self.assertContains(response, 'Created new SSH key')
def test_ssh_generate(self):
self.assertEqual(check_data_writable(), [])
response = self.client.get(reverse("manage-ssh"))
self.assertContains(response, "Generate SSH key")
response = self.client.post(reverse("manage-ssh"), {"action": "generate"})
self.assertContains(response, "Created new SSH key")
response = self.client.get(reverse("manage-ssh-key"))
self.assertContains(response, "PRIVATE KEY")
def test_get(self):
self.assertEqual(check_data_writable(), [])
# This will generate new key
key = get_gpg_sign_key()
self.assertIsNotNone(key)
# Check cache access
self.assertEqual(key, get_gpg_sign_key())
# Check empty cache
cache.delete('gpg-key-id')
self.assertEqual(key, get_gpg_sign_key())
def test_get(self):
self.assertEqual(check_data_writable(), [])
# This will generate new key
key = get_gpg_sign_key()
self.assertIsNotNone(key)
# Check cache access
self.assertEqual(key, get_gpg_sign_key())
# Check empty cache
cache.delete('gpg-key-id')
self.assertEqual(key, get_gpg_sign_key())
def test_ssh_generate(self):
self.assertEqual(check_data_writable(), [])
response = self.client.get(reverse('admin:ssh'))
self.assertContains(response, 'Generate SSH key')
response = self.client.post(
reverse('admin:ssh'),
{'action': 'generate'}
)
self.assertContains(response, 'Created new SSH key')
def test_create_ssh_wrapper(self):
self.assertEqual(check_data_writable(), [])
wrapper = SSHWrapper()
filename = wrapper.filename
wrapper.create()
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
wrapper.create()
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, '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
wrapper.create()
self.assertEqual(timestamp, os.stat(filename).st_mtime)
def test_ssh_add(self):
self.assertEqual(check_data_writable(), [])
try:
oldpath = os.environ["PATH"]
os.environ["PATH"] = ":".join((get_test_file(""), os.environ["PATH"]))
# Verify there is button for adding
response = self.client.get(reverse("manage-ssh"))
self.assertContains(response, "Add host key")
# Add the key
response = self.client.post(
reverse("manage-ssh"), {"action": "add-host", "host": "github.com"}
)
self.assertContains(response, "Added host key for github.com")
finally:
os.environ["PATH"] = oldpath
# Check the file contains it
hostsfile = os.path.join(settings.DATA_DIR, "ssh", "known_hosts")
with open(hostsfile) as handle:
self.assertIn("github.com", handle.read())
def test_ssh_add(self):
self.assertEqual(check_data_writable(), [])
try:
oldpath = os.environ['PATH']
os.environ['PATH'] = ':'.join((get_test_file(''), os.environ['PATH']))
# Verify there is button for adding
response = self.client.get(reverse('manage-ssh'))
self.assertContains(response, 'Add host key')
# Add the key
response = self.client.post(
reverse('manage-ssh'), {'action': 'add-host', 'host': 'github.com'}
)
self.assertContains(response, 'Added host key for github.com')
finally:
os.environ['PATH'] = oldpath
# Check the file contains it
hostsfile = os.path.join(settings.DATA_DIR, 'ssh', 'known_hosts')
with open(hostsfile) as handle:
self.assertIn('github.com', handle.read())
def test_ssh_add(self):
self.assertEqual(check_data_writable(), [])
try:
oldpath = os.environ['PATH']
os.environ['PATH'] = ':'.join(
(get_test_file(''), os.environ['PATH'])
)
# Verify there is button for adding
response = self.client.get(reverse('admin:ssh'))
self.assertContains(response, 'Add host key')
# Add the key
response = self.client.post(
reverse('admin:ssh'),
{'action': 'add-host', 'host': 'github.com'}
)
self.assertContains(response, 'Added host key for github.com')
finally:
os.environ['PATH'] = oldpath
# Check the file contains it
hostsfile = os.path.join(settings.DATA_DIR, 'ssh', 'known_hosts')
with open(hostsfile) as handle:
self.assertIn('github.com', handle.read())
def test_generate(self):
self.assertEqual(check_data_writable(), [])
self.assertIsNone(get_gpg_key())
key = generate_gpg_key()
self.assertIsNotNone(key)
self.assertEqual(key, get_gpg_key())
def test_parse(self):
self.assertEqual(check_data_writable(), [])
shutil.copy(TEST_HOSTS, os.path.join(settings.DATA_DIR, 'ssh'))
hosts = get_host_keys()
self.assertEqual(len(hosts), 50)
def test_parse(self):
self.assertEqual(check_data_writable(), [])
shutil.copy(TEST_HOSTS, os.path.join(settings.DATA_DIR, 'ssh'))
hosts = get_host_keys()
self.assertEqual(len(hosts), 50)
