def test_check_certificate_error_loading(self, mock_load): mock_load.side_effect = Exception('some error') self.user.auth.cert = 'some cert' before = datetime.datetime.now() after = datetime.datetime.now() mock_load.return_value.get_notBefore.return_value = before.strftime( tasks.CERT_DATE_FMT) mock_load.return_value.get_notAfter.return_value = after.strftime( tasks.CERT_DATE_FMT) with self.assertRaises(tasks.CertificateError): tasks.check_certificate(self.user)
def test_check_certificate(self, mock_load): self.user.auth.cert = 'some cert' before = datetime.datetime.now() - datetime.timedelta(days=10) after = datetime.datetime.now() + datetime.timedelta(days=10) mock_load.return_value.get_notBefore.return_value = before.strftime( tasks.CERT_DATE_FMT) mock_load.return_value.get_notAfter.return_value = after.strftime( tasks.CERT_DATE_FMT) self.assertTrue(tasks.check_certificate(self.user))
def test_check_certificate_missing_certificate(self): with self.assertRaises(tasks.CertificateError): tasks.check_certificate(self.user)