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_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_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)