Fixing tests.

lemur/certificates/models.py

@@ -79,6 +79,7 @@ class Certificate(db.Model):
                             secondaryjoin=id == certificate_replacement_associations.c.replaced_certificate_id,  # noqa
                             backref='replaced')
 
+    views = relationship("View", backref="certificate")
     endpoints = relationship("Endpoint", backref='certificate')
 
     def __init__(self, **kwargs):

lemur/certificates/service.py

@@ -138,8 +138,8 @@ def log_private_key_view(certificate, user):
     :param user:
     :return:
     """
-    view = View(user=user)
-    certificate.views.append(view)
+    view = View(user_id=user.id, certificate_id=certificate.id)
+    database.add(view)
     database.commit()
 
 

lemur/tests/test_certificates.py

@@ -430,7 +430,7 @@ def test_private_key_audit(client, certificate):
 
 
 @pytest.mark.parametrize("token,status", [
-    (VALID_USER_HEADER_TOKEN, 403),
+    (VALID_USER_HEADER_TOKEN, 200),
     (VALID_ADMIN_HEADER_TOKEN, 200),
     ('', 401)
 ])
@@ -528,15 +528,6 @@ def test_certificates_patch(client, token, status):
     assert client.patch(api.url_for(CertificatesList), data={}, headers=token).status_code == status
 
 
-@pytest.mark.parametrize("token,status", [
-    (VALID_USER_HEADER_TOKEN, 403),
-    (VALID_ADMIN_HEADER_TOKEN, 200),
-    ('', 401)
-])
-def test_certificate_credentials_get(client, token, status):
-    assert client.get(api.url_for(CertificatePrivateKey, certificate_id=1), headers=token).status_code == status
-
-
 @pytest.mark.parametrize("token,status", [
     (VALID_USER_HEADER_TOKEN, 405),
     (VALID_ADMIN_HEADER_TOKEN, 405),

lemur/users/models.py

@@ -85,10 +85,8 @@ class View(db.Model):
     __tablename__ = 'views'
     id = Column(Integer, primary_key=True)
     certificate_id = Column(Integer, ForeignKey('certificates.id'))
-    certificates = relationship("Certificate", backref='views')
     viewed_at = Column(ArrowType(), PassiveDefault(func.now()), nullable=False)
     user_id = Column(Integer, ForeignKey('users.id'))
-    users = relationship("User", backref="views")
 
 
 listen(User, 'before_insert', hash_password)

tox.ini

@@ -1,2 +1,2 @@
 [tox]
-envlist = py27,py35
+envlist = py35