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
