def setUp(self):
""" setup unittest """
models_mock = MagicMock()
models_mock.acme_srv.db_handler.DBstore.return_value = FakeDBStore
modules = {'acme_srv.db_handler': models_mock}
patch.dict('sys.modules', modules).start()
import logging
logging.basicConfig(level=logging.CRITICAL)
self.logger = logging.getLogger('test_a2c')
from acme_srv.trigger import Trigger
from acme_srv.order import Order
self.order = Order(False, 'http://tester.local', self.logger)
self.trigger = Trigger(False, 'http://tester.local', self.logger)
def trigger(request):
""" ca trigger"""
if request.method == 'POST':
with Trigger(DEBUG, get_url(request.META), LOGGER) as trigger_:
response_dic = trigger_.parse(request.body)
# create the response
if 'data' in response_dic:
response = JsonResponse(status=response_dic['code'],
data=response_dic['data'])
else:
response = HttpResponse(status=response_dic['code'])
# generate additional header elements
for element in response_dic['header']:
response[element] = response_dic['header'][element]
# logging
logger_info(LOGGER, request.META['REMOTE_ADDR'],
request.META['PATH_INFO'], response_dic)
# send response
return response
else:
return JsonResponse(status=405,
data={
'status': 405,
'message': 'Method Not Allowed',
'detail': 'Wrong request type. Expected POST.'
})
def trigger(environ, start_response):
""" ca trigger handler """
if environ['REQUEST_METHOD'] == 'POST':
with Trigger(DEBUG, get_url(environ), LOGGER) as trigger_:
request_body = get_request_body(environ)
response_dic = trigger_.parse(request_body)
# create header
headers = create_header(response_dic)
start_response(
'{0} {1}'.format(response_dic['code'],
HTTP_CODE_DIC[response_dic['code']]), headers)
# logging
logger_info(LOGGER, environ['REMOTE_ADDR'], environ['PATH_INFO'],
response_dic)
if 'data' in response_dic:
return [json.dumps(response_dic['data']).encode('utf-8')]
else:
return []
# start_response('200 {0}'.format(HTTP_CODE_DIC[200]), [('Content-Type', 'application/json')])
# return [json.dumps({'status':200, 'message':HTTP_CODE_DIC[200], 'detail': 'OK'}).encode('utf-8')]
else:
start_response('405 {0}'.format(HTTP_CODE_DIC[405]),
[('Content-Type', 'application/json')])
return [
json.dumps({
'status': 405,
'message': HTTP_CODE_DIC[405],
'detail': 'Wrong request type. Expected POST.'
}).encode('utf-8')
]
class TestACMEHandler(unittest.TestCase):
""" test class for ACMEHandler """
acme = None
def setUp(self):
""" setup unittest """
models_mock = MagicMock()
models_mock.acme_srv.db_handler.DBstore.return_value = FakeDBStore
modules = {'acme_srv.db_handler': models_mock}
patch.dict('sys.modules', modules).start()
import logging
logging.basicConfig(level=logging.CRITICAL)
self.logger = logging.getLogger('test_a2c')
from acme_srv.trigger import Trigger
from acme_srv.order import Order
self.order = Order(False, 'http://tester.local', self.logger)
self.trigger = Trigger(False, 'http://tester.local', self.logger)
@patch('importlib.import_module')
@patch('acme_srv.certificate.Certificate.certlist_search')
@patch('acme_srv.trigger.cert_pubkey_get')
def test_001_trigger__certname_lookup(self, mock_cert_pub,
mock_search_list, mock_import):
""" trigger._certname_lookup() failed bcs. of empty certificate list """
mock_cert_pub.return_value = 'foo'
mock_search_list.return_value = []
mock_import.return_value = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.assertEqual([], self.trigger._certname_lookup('cert_pem'))
@patch('importlib.import_module')
@patch('acme_srv.certificate.Certificate.certlist_search')
@patch('acme_srv.trigger.cert_pubkey_get')
def test_002_trigger__certname_lookup(self, mock_cert_pub,
mock_search_list, mock_import):
""" trigger._certname_lookup() failed bcs. of wrong certificate list """
mock_cert_pub.return_value = 'foo'
mock_search_list.return_value = [{'foo': 'bar'}]
mock_import.return_value = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.assertEqual([], self.trigger._certname_lookup('cert_pem'))
@patch('importlib.import_module')
@patch('acme_srv.certificate.Certificate.certlist_search')
@patch('acme_srv.trigger.cert_pubkey_get')
def test_003_trigger__certname_lookup(self, mock_cert_pub,
mock_search_list, mock_import):
""" trigger._certname_lookup() failed bcs. of emty csr field """
mock_cert_pub.return_value = 'foo'
mock_search_list.return_value = [{'csr': None}]
mock_import.return_value = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.assertEqual([], self.trigger._certname_lookup('cert_pem'))
@patch('importlib.import_module')
@patch('acme_srv.trigger.csr_pubkey_get')
@patch('acme_srv.certificate.Certificate.certlist_search')
@patch('acme_srv.trigger.cert_pubkey_get')
def test_004_trigger__certname_lookup(self, mock_cert_pub,
mock_search_list, mock_csr_pub,
mock_import):
""" trigger._certname_lookup() failed bcs. of emty csr field """
mock_cert_pub.return_value = 'foo'
mock_csr_pub.return_value = 'foo1'
mock_search_list.return_value = [{'csr': None}]
mock_import.return_value = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.assertEqual([], self.trigger._certname_lookup('cert_pem'))
@patch('importlib.import_module')
@patch('acme_srv.trigger.csr_pubkey_get')
@patch('acme_srv.certificate.Certificate.certlist_search')
@patch('acme_srv.trigger.cert_pubkey_get')
def test_005_trigger__certname_lookup(self, mock_cert_pub,
mock_search_list, mock_csr_pub,
mock_import):
""" trigger._certname_lookup() failed bcs. of emty csr field """
mock_cert_pub.return_value = 'foo'
mock_csr_pub.return_value = 'foo'
mock_search_list.return_value = [{
'csr': 'csr',
'name': 'cert_name',
'order__name': 'order_name'
}]
mock_import.return_value = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.assertEqual([{
'cert_name': 'cert_name',
'order_name': 'order_name'
}], self.trigger._certname_lookup('cert_pem'))
def test_006_trigger_parse(self):
""" Trigger.parse() with empty payload """
payload = ""
result = {
'header': {},
'code': 400,
'data': {
'detail': 'payload missing',
'message': 'malformed',
'status': 400
}
}
self.assertEqual(result, self.trigger.parse(payload))
def test_007_trigger_parse(self):
""" Trigger.parse() with wrong payload """
payload = '{"foo": "bar"}'
result = {
'header': {},
'code': 400,
'data': {
'detail': 'payload missing',
'message': 'malformed',
'status': 400
}
}
self.assertEqual(result, self.trigger.parse(payload))
def test_008_trigger_parse(self):
""" Trigger.parse() with empty payload key"""
payload = '{"payload": ""}'
result = {
'header': {},
'code': 400,
'data': {
'detail': 'payload empty',
'message': 'malformed',
'status': 400
}
}
self.assertEqual(result, self.trigger.parse(payload))
@patch('acme_srv.trigger.Trigger._payload_process')
def test_009_trigger_parse(self, mock_process):
""" Trigger.parse() with payload mock result 400"""
payload = '{"payload": "foo"}'
mock_process.return_value = (400, 'message', 'detail')
result = {
'header': {},
'code': 400,
'data': {
'detail': 'detail',
'message': 'message',
'status': 400
}
}
self.assertEqual(result, self.trigger.parse(payload))
@patch('acme_srv.trigger.Trigger._payload_process')
def test_010_trigger_parse(self, mock_process):
""" Trigger.parse() with payload mock result 200"""
payload = '{"payload": "foo"}'
mock_process.return_value = (200, 'message', 'detail')
result = {
'header': {},
'code': 200,
'data': {
'detail': 'detail',
'message': 'message',
'status': 200
}
}
self.assertEqual(result, self.trigger.parse(payload))
def test_011_trigger__payload_process(self):
""" Trigger._payload_process() without payload"""
payload = {}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', None,
None))
self.assertEqual((400, 'payload malformed', None),
self.trigger._payload_process(payload))
def test_012_trigger__payload_process(self):
""" Trigger._payload_process() without certbunde and cert_raw"""
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', None,
None))
self.assertEqual((400, 'error', None),
self.trigger._payload_process(payload))
def test_013_trigger__payload_process(self):
""" Trigger._payload_process() with bundle and without cart_raw"""
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', 'bundle',
None))
self.assertEqual((400, 'error', None),
self.trigger._payload_process(payload))
def test_014_trigger__payload_process(self):
""" Trigger._payload_process() with bundle and without cart_raw"""
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', None,
'raw'))
self.assertEqual((400, 'error', None),
self.trigger._payload_process(payload))
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_015_trigger__payload_process(self, mock_cobystr, mock_der2pem,
mock_b64dec, mock_lookup):
""" Trigger._payload_process() with certificae_name"""
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', 'bundle',
'raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = [{
'cert_name': 'certificate_name',
'order_name': None
}]
self.assertEqual((200, 'OK', None),
self.trigger._payload_process(payload))
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_016_trigger__payload_process(self, mock_cobystr, mock_der2pem,
mock_b64dec, mock_lookup):
""" Trigger._payload_process() without certificate_name """
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', 'bundle',
'raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = [{
'cert_name': None,
'order_name': 'order_name'
}]
self.assertEqual((200, 'OK', None),
self.trigger._payload_process(payload))
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_017_trigger__payload_process(self, mock_cobystr, mock_der2pem,
mock_b64dec, mock_lookup):
""" Trigger._payload_process() _certname.lookup() returned empty list """
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', 'bundle',
'raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = []
self.assertEqual((400, 'certificate_name lookup failed', None),
self.trigger._payload_process(payload))
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_018_trigger__payload_process(self, mock_cobystr, mock_der2pem,
mock_b64dec, mock_lookup):
""" Trigger._payload_process() without certificate_name """
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', 'bundle',
'raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = [{
'cert_name': 'certificate_name',
'order_name': 'order_name'
}]
self.order.dbstore.order_update.return_value = None
self.assertEqual((200, 'OK', None),
self.trigger._payload_process(payload))
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_019_trigger__payload_process(self, mock_cobystr, mock_der2pem,
mock_b64dec, mock_lookup):
""" Trigger._payload_process() without certificate_name """
payload = {'payload': 'foo'}
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=('error', 'bundle',
'raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = [{
'cert_name': 'certificate_name1',
'order_name': 'order_name1'
}, {
'cert_name': 'certificate_name2',
'order_name': 'order_name2'
}]
self.assertEqual((200, 'OK', None),
self.trigger._payload_process(payload))
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_020_trigger__payload_process(self, mock_cobystr, mock_lookup,
mock_der2pem, mock_b64dec):
""" test Trigger._payload_process - dbstore.order_update() raises an exception """
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=(None,
'certificate',
'certificate_raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = [{
'cert_name': 'certificate_name1',
'order_name': 'order_name1'
}, {
'cert_name': 'certificate_name2',
'order_name': 'order_name2'
}]
self.trigger.dbstore.certificate_add.return_value = True
self.trigger.dbstore.order_update.side_effect = Exception(
'exc_trigger_order_upd')
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._payload_process('payload')
self.assertIn(
'CRITICAL:test_a2c:acme2certifier database error in trigger._payload_process() upd: exc_trigger_order_upd',
lcm.output)
@patch('acme_srv.trigger.b64_decode')
@patch('acme_srv.trigger.cert_der2pem')
@patch('acme_srv.trigger.Trigger._certname_lookup')
@patch('acme_srv.trigger.convert_byte_to_string')
def test_021_trigger__payload_process(self, mock_cobystr, mock_lookup,
mock_der2pem, mock_b64dec):
""" test Trigger._payload_process - dbstore.order_update() raises an exception """
ca_handler_module = importlib.import_module(
'examples.ca_handler.skeleton_ca_handler')
self.trigger.cahandler = ca_handler_module.CAhandler
self.trigger.cahandler.trigger = Mock(return_value=(None,
'certificate',
'certificate_raw'))
mock_der2pem.return_value = 'der2pem'
mock_cobystr.return_value = 'cert_pem'
mock_b64dec.return_value = 'b64dec'
mock_lookup.return_value = [{
'cert_name': 'certificate_name1',
'order_name': 'order_name1'
}, {
'cert_name': 'certificate_name2',
'order_name': 'order_name2'
}]
self.trigger.dbstore.certificate_add.side_effect = Exception(
'exc_trigger_order_add')
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._payload_process('payload')
self.assertIn(
'CRITICAL:test_a2c:acme2certifier database error in trigger._payload_process() add: exc_trigger_order_add',
lcm.output)
@patch('acme_srv.trigger.load_config')
def test_022_config_load(self, mock_load_cfg):
""" test _config_load missing ca_handler """
mock_load_cfg.return_value = {}
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._config_load()
self.assertIn(
'ERROR:test_a2c:Helper.ca_handler_load(): CAhandler configuration missing in config file',
lcm.output)
@patch('acme_srv.trigger.Trigger._config_load')
def test_023__enter__(self, mock_cfg):
""" test enter """
mock_cfg.return_value = True
self.trigger.__enter__()
self.assertTrue(mock_cfg.called)
@patch('acme_srv.trigger.load_config')
def test_024_config_load(self, mock_load_cfg):
""" test _config_load empty config """
parser = configparser.ConfigParser()
# parser['Account'] = {'foo': 'bar'}
mock_load_cfg.return_value = parser
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._config_load()
self.assertFalse(self.trigger.tnauthlist_support)
self.assertIn(
'ERROR:test_a2c:Helper.ca_handler_load(): CAhandler configuration missing in config file',
lcm.output)
@patch('acme_srv.trigger.load_config')
def test_025_config_load(self, mock_load_cfg):
""" test _config_load bogus ca_handler """
parser = configparser.ConfigParser()
parser['CAhandler'] = {'handler_file': 'foo'}
mock_load_cfg.return_value = parser
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._config_load()
self.assertIn(
"CRITICAL:test_a2c:Helper.ca_handler_load(): loading CAhandler configured in cfg failed with err: 'NoneType' object has no attribute 'loader'",
lcm.output)
@patch('importlib.import_module')
@patch('acme_srv.trigger.load_config')
def test_026_config_load(self, mock_load_cfg, mock_imp):
""" test _config_load missing ca_handler """
parser = configparser.ConfigParser()
parser['CAhandler'] = {'handler_file': 'foo'}
mock_load_cfg.return_value = parser
mock_imp.return_value = Mock()
self.trigger._config_load()
self.assertTrue(self.trigger.cahandler)
@patch('importlib.import_module')
@patch('acme_srv.trigger.load_config')
def test_027_config_load(self, mock_load_cfg, mock_imp):
""" test _config_load missing ca_handler """
parser = configparser.ConfigParser()
parser['CAhandler'] = {'foo': 'bar'}
mock_load_cfg.return_value = parser
mock_imp.return_value = Mock()
self.trigger._config_load()
self.assertTrue(self.trigger.cahandler)
@patch('acme_srv.trigger.load_config')
def test_028_config_load(self, mock_load_cfg):
""" test _config_load empty config """
parser = configparser.ConfigParser()
parser['Order'] = {'tnauthlist_support': False}
mock_load_cfg.return_value = parser
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._config_load()
self.assertFalse(self.trigger.tnauthlist_support)
@patch('acme_srv.trigger.load_config')
def test_029_config_load(self, mock_load_cfg):
""" test _config_load empty config """
parser = configparser.ConfigParser()
parser['Order'] = {'tnauthlist_support': True}
mock_load_cfg.return_value = parser
with self.assertLogs('test_a2c', level='INFO') as lcm:
self.trigger._config_load()
self.assertTrue(self.trigger.tnauthlist_support)