f631fdde2c
License headers were missing from test files. Also remove empty tests/util.py file. Change-Id: Iecd13930c4c4b4c5235f13274f92e56a3f9a75f5
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
# (c) Copyright 2015 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import datetime
|
|
import tempfile
|
|
import unittest
|
|
|
|
import freezegun
|
|
import mock
|
|
|
|
from cathead import certwatch
|
|
import cathead.x509
|
|
|
|
|
|
class CertWatcherTestCase(unittest.TestCase):
|
|
|
|
@mock.patch.object(cathead.x509, 'get_expire_date')
|
|
def test_expires_in_window(self, mock_get_expire_date):
|
|
key = tempfile.NamedTemporaryFile()
|
|
cert = tempfile.NamedTemporaryFile()
|
|
watcher = certwatch.CertWatcher(key.name, cert.name,
|
|
"common name", None,
|
|
refresh_window=40)
|
|
mock_get_expire_date.return_value = datetime.datetime(2014, 12, 19,
|
|
15, 18, 53)
|
|
with freezegun.freeze_time("2014-12-19 15:18:10"):
|
|
self.assertFalse(watcher._expires_in_window())
|
|
|
|
with freezegun.freeze_time("2014-12-19 15:18:14"):
|
|
self.assertTrue(watcher._expires_in_window())
|
|
|
|
@mock.patch.object(cathead.x509, 'get_expire_date')
|
|
@mock.patch.object(cathead.x509, 'generate_csr')
|
|
def test_check_and_update(self, mock_generate_csr, mock_get_expire_date):
|
|
mock_get_expire_date.return_value = datetime.datetime(2014, 12, 19,
|
|
15, 18, 53)
|
|
mock_generate_csr.return_value = "hello csr"
|
|
|
|
callback = mock.Mock()
|
|
key = tempfile.NamedTemporaryFile()
|
|
cert = tempfile.NamedTemporaryFile()
|
|
watcher = certwatch.CertWatcher(key.name, cert.name, 'common_name',
|
|
None, on_refresh_success=callback,
|
|
refresh_window=40)
|
|
|
|
watcher.ca_driver = mock.Mock()
|
|
watcher.ca_driver.sign.return_value = "hello cert"
|
|
|
|
watcher.check_and_update()
|
|
|
|
self.assertEqual("hello cert", cert.file.read())
|
|
watcher.ca_driver.called_once_with("hello csr")
|
|
callback.assert_called_once_with()
|