aswi2021jmsd-gitlab4/tests/services/certificate/conftest.py @ d3fa9147

import pytest
import sqlite3
import os
from sqlite3 import Connection, Cursor

from src.dao.certificate_repository import CertificateRepository
from src.dao.private_key_repository import PrivateKeyRepository
from src.services.certificate_service import CertificateService
from src.services.cryptography import CryptographyService
from src.services.key_service import KeyService


@pytest.fixture
def connection():
    try:
        os.remove("../../../test.sqlite")
    except FileNotFoundError:
        pass
    connection: Connection = sqlite3.connect("../../../" + "test.sqlite")
    return connection


@pytest.fixture
def cursor(connection):
    cursor = connection.cursor()

    # execute db initialisation script
    with open('../../../SQLite_database.sql', 'r') as sql_file:
        sql_script = sql_file.read()
        cursor.executescript(sql_script)

    # insert default values
    with open('../../../SQLite_default_values.sql', 'r') as sql_file:
        sql_script = sql_file.read()
        cursor.executescript(sql_script)

    return cursor


@pytest.fixture
def certificate_repository(connection, cursor):
    return CertificateRepository(connection, cursor)


@pytest.fixture
def private_key_repository(connection, cursor):
    return PrivateKeyRepository(connection, cursor)


@pytest.fixture
def cryptography_service():
    return CryptographyService()


@pytest.fixture
def private_key_service(private_key_repository, cryptography_service):
    return KeyService(cryptography_service, private_key_repository)


@pytest.fixture
def certificate_service(certificate_repository, cryptography_service):
    return CertificateService(cryptography_service, certificate_repository)