def execute(self, args): manager = SSLManager(self.client) certificates = manager.list_certs(args['--status']) table = Table(['id', 'common_name', 'days_until_expire', 'notes']) for certificate in certificates: table.add_row([ certificate['id'], certificate['commonName'], certificate['validityDays'], certificate.get('notes', blank()) ]) table.sortby = args['--sortby'] return table
def execute(self, args): template = {'id': args['<id>']} if args['--crt']: template['certificate'] = open(args['--crt']).read() if args['--key']: template['privateKey'] = open(args['--key']).read() if args['--csr']: template['certificateSigningRequest'] = open(args['--csr']).read() if args['--icc']: template['intermediateCertificate'] = open(args['--icc']).read() if args['--notes']: template['notes'] = args['--notes'] manager = SSLManager(self.client) manager.edit_certificate(template)
def execute(self, args): def write_cert(filename, content): """ Writes certificate body to the given file path """ with open(filename, 'w') as cert_file: cert_file.write(content) manager = SSLManager(self.client) certificate = manager.get_certificate(args['<id>']) write_cert( certificate['commonName'] + '.crt', certificate['certificate']) write_cert( certificate['commonName'] + '.key', certificate['privateKey']) if 'intermediateCertificate' in certificate: write_cert( certificate['commonName'] + '.icc', certificate['intermediateCertificate']) if 'certificateSigningRequest' in certificate: write_cert( certificate['commonName'] + '.csr', certificate['certificateSigningRequest'])
def execute(self, args): template = { 'intermediateCertificate': '', 'certificateSigningRequest': '', 'notes': args['--notes'], } try: template['certificate'] = open(args['--crt']).read() template['privateKey'] = open(args['--key']).read() if args['--csr']: template['certificateSigningRequest'] = \ open(args['--csr']).read() if args['--icc']: template['intermediateCertificate'] = \ open(args['--icc']).read() except IOError: raise CLIAbort("File does not exist") manager = SSLManager(self.client) manager.add_certificate(template)
def execute(self, args): def write_cert(filename, content): try: fo = open(filename, 'w') fo.write(content) finally: fo.close() manager = SSLManager(self.client) certificate = manager.get_certificate(args['<id>']) write_cert( certificate['commonName'] + '.crt', certificate['certificate']) write_cert( certificate['commonName'] + '.key', certificate['privateKey']) if 'intermediateCertificate' in certificate: write_cert( certificate['commonName'] + '.icc', certificate['intermediateCertificate']) if 'certificateSigningRequest' in certificate: write_cert( certificate['commonName'] + '.csr', certificate['certificateSigningRequest'])
def setUp(self): self.client = FixtureClient() self.ssl = SSLManager(self.client) self.test_id = 10
class SSLTests(unittest.TestCase): def setUp(self): self.client = FixtureClient() self.ssl = SSLManager(self.client) self.test_id = 10 def test_list_certs(self): self.ssl.list_certs('valid') f = self.client['Account'].getValidSecurityCertificates f.assert_called_once_with(mask=ANY) self.ssl.list_certs('expired') f = self.client['Account'].getExpiredSecurityCertificates f.assert_called_once_with(mask=ANY) self.ssl.list_certs('all') f = self.client['Account'].getSecurityCertificates f.assert_called_once_with(mask=ANY) def test_add_certificate(self): test_cert = { 'certificate': 'cert', 'privateKey': 'key', } self.ssl.add_certificate(test_cert) f = self.client['Security_Certificate'].createObject f.assert_called_once_with(test_cert) def test_remove_certificate(self): self.ssl.remove_certificate(self.test_id) f = self.client['Security_Certificate'].deleteObject f.assert_called_once_with(id=self.test_id) def test_edit_certificate(self): test_cert = { 'id': self.test_id, 'certificate': 'cert', 'privateKey': 'key' } self.ssl.edit_certificate(test_cert) f = self.client['Security_Certificate'].editObject f.assert_called_once_with( { 'id': self.test_id, 'certificate': 'cert', 'privateKey': 'key' }, id=self.test_id) def test_get_certificate(self): self.ssl.get_certificate(self.test_id) f = self.client['Security_Certificate'].getObject f.assert_called_once_with(id=self.test_id)
def get_ssl_manager(): return SSLManager(get_client())
def execute(self, args): manager = SSLManager(self.client) if args['--really'] or no_going_back('yes'): manager.remove_certificate(args['<id>']) raise CLIAbort("Aborted.")
def setUp(self): self.client = MagicMock() self.ssl = SSLManager(self.client) self.test_id = 10