83 lines
3.7 KiB
Python
83 lines
3.7 KiB
Python
# (C) Copyright 2016 Hewlett Packard Enterprise Development Company LP
|
|
|
|
import logging
|
|
import unittest
|
|
|
|
from mock import patch
|
|
|
|
from monasca_setup.detection.plugins.cert_check import CertificateCheck
|
|
|
|
LOG = logging.getLogger('monasca_setup.detection.plugins.cert_check')
|
|
|
|
|
|
class TestCertCheck(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
unittest.TestCase.setUp(self)
|
|
with patch.object(CertificateCheck, '_detect') as mock_detect:
|
|
self.cert_obj = CertificateCheck('temp_dir')
|
|
self.assertTrue(mock_detect.called)
|
|
self.cert_obj.args = {'urls': 'http://fake-cert.com'}
|
|
|
|
def test_detect(self):
|
|
self.cert_obj.available = False
|
|
with patch.object(self.cert_obj, '_check_required_args',
|
|
return_value=True) as mock_check_required_args:
|
|
self.cert_obj._detect()
|
|
self.assertTrue(self.cert_obj.available)
|
|
self.assertTrue(mock_check_required_args.called)
|
|
|
|
def _build_config(self):
|
|
with patch.object(self.cert_obj, '_build_instance',
|
|
return_value={}) as mock_build_instance:
|
|
result = self.cert_obj.build_config()
|
|
self.assertTrue(mock_build_instance.called)
|
|
self.assertEqual(result['cert_check']['instances'][0]['url'],
|
|
'http://fake-cert.com')
|
|
self.assertEqual(result['cert_check']['instances'][0]['name'],
|
|
'http://fake-cert.com')
|
|
return result
|
|
|
|
def test_build_config_without_args(self):
|
|
result = self._build_config()
|
|
self.assertEqual(result['cert_check']['init_config']['ca_certs'],
|
|
'/etc/ssl/certs/ca-certificates.crt')
|
|
self.assertEqual(result['cert_check']['init_config']['ciphers'],
|
|
'HIGH:-aNULL:-eNULL:-PSK:RC4-SHA:RC4-MD5')
|
|
self.assertEqual(result['cert_check']['init_config']['collect_period'],
|
|
3600)
|
|
self.assertEqual(result['cert_check']['init_config']['timeout'],
|
|
1.0)
|
|
|
|
def test_build_config_with_args(self):
|
|
self.cert_obj.args.update({'ca_certs': '/tmp/ssl/certs/ca-certificates.crt',
|
|
'ciphers': 'fake-cipher',
|
|
'timeout': 3.0,
|
|
'collect_period': 1200})
|
|
result = self._build_config()
|
|
self.assertEqual(result['cert_check']['init_config']['ca_certs'],
|
|
'/tmp/ssl/certs/ca-certificates.crt')
|
|
self.assertEqual(result['cert_check']['init_config']['ciphers'],
|
|
'fake-cipher')
|
|
self.assertEqual(result['cert_check']['init_config']['collect_period'],
|
|
1200)
|
|
self.assertEqual(result['cert_check']['init_config']['timeout'],
|
|
3.0)
|
|
|
|
def test_timeout_error_log(self):
|
|
self.cert_obj.args.update({'ca_certs': '/tmp/ssl/certs/ca-certificates.crt',
|
|
'ciphers': 'fake-cipher',
|
|
'timeout': 0.0,
|
|
'collect_period': 1200})
|
|
with patch.object(LOG, 'error') as mock_log:
|
|
result = self._build_config()
|
|
self.assertEqual(result['cert_check']['init_config']['ca_certs'],
|
|
'/tmp/ssl/certs/ca-certificates.crt')
|
|
self.assertEqual(result['cert_check']['init_config']['ciphers'],
|
|
'fake-cipher')
|
|
self.assertEqual(result['cert_check']['init_config']['collect_period'],
|
|
1200)
|
|
self.assertEqual(result['cert_check']['init_config']['timeout'],
|
|
1.0)
|
|
self.assertTrue(mock_log.called)
|