Source code for pki.tests.test_models.test_certificate_model

"""Tests for the CertificateModel class."""

# ruff: noqa: F811  # ruff does not like pytest fixtures as arguments

import pytest
from cryptography import x509
from cryptography.hazmat.primitives import serialization
from trustpoint_core.serializer import CertificateSerializer

from pki.models.certificate import CertificateModel
from pki.tests.fixtures import self_signed_cert_with_ext  # noqa: F401


@pytest.mark.django_db

[docs]
def test_save_certificate_method(self_signed_cert_with_ext: x509.Certificate) -> None:
    """Test that save_certificate method creates and stores a certificate model instance."""
    cert_model = CertificateModel.save_certificate(self_signed_cert_with_ext)
    assert isinstance(cert_model, CertificateModel)
    assert CertificateModel.objects.get(pk=cert_model.pk) == cert_model



@pytest.mark.django_db

[docs]
def test_save_certificate_with_serializer(self_signed_cert_with_ext: x509.Certificate) -> None:
    """Test saving a certificate using a CertificateSerializer instead of a raw x509.Certificate."""
    cert_pem = self_signed_cert_with_ext.public_bytes(serialization.Encoding.PEM)
    serializer = CertificateSerializer.from_pem(cert_pem)
    cert_model = CertificateModel.save_certificate(serializer)
    assert cert_model