From 8e5e8fdd030162e261a72f1d446a8ee56314dc08 Mon Sep 17 00:00:00 2001
From: Hossein Shafagh <hshafagh@netflix.com>
Date: Thu, 4 Mar 2021 19:12:57 -0800
Subject: [PATCH] tests

---
 lemur/tests/test_certificates.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/lemur/tests/test_certificates.py b/lemur/tests/test_certificates.py
index 962c40b4..fa90280c 100644
--- a/lemur/tests/test_certificates.py
+++ b/lemur/tests/test_certificates.py
@@ -84,6 +84,27 @@ def test_get_by_serial(session, certificate):
     assert found
 
 
+def test_get_all_certs_attached_to_endpoint_without_autorotate(session):
+    from lemur.certificates.service import get_all_certs_attached_to_endpoint_without_autorotate, \
+        cleanup_after_revoke
+    from lemur.tests.factories import EndpointFactory, CertificateFactory
+
+    # add a certificate with endpoint
+    s = EndpointFactory()
+    CertificateFactory(endpoint=s)
+    session.commit()
+
+    list_before = get_all_certs_attached_to_endpoint_without_autorotate()
+    len_list_before = len(list_before)
+    assert len_list_before > 0
+    # revoked the first certificate
+    first_cert_with_endpoitn = list_before[0]
+    cleanup_after_revoke(first_cert_with_endpoitn)
+
+    list_after = get_all_certs_attached_to_endpoint_without_autorotate()
+    assert len(list_after) + 1 == len_list_before
+
+
 def test_delete_cert(session):
     from lemur.certificates.service import delete, get
     from lemur.tests.factories import CertificateFactory