aswi2021jmsd-gitlab/tests/integration_tests/services/conftest.py @ 43439c9f

import os
import sqlite3
from sqlite3 import Connection

import pytest

from src.dao.certificate_repository import CertificateRepository
from src.dao.private_key_repository import PrivateKeyRepository
from src.db.init_queries import SCHEMA_SQL, DEFAULT_VALUES_SQL
from src.services.certificate_service import CertificateService
from src.services.cryptography import CryptographyService
from src.services.key_service import KeyService


# scope="module" means that this fixture is run once per module
@pytest.fixture(scope="module")
def connection():
    print("Creating a new SQLITE connection to the test DB")
    test_db_file = "test.sqlite"
    connection: Connection = sqlite3.connect(test_db_file)

    # yield the created connection
    yield connection

    # after tests have finished delete the created db file
    try:
        print("Deleting the test DB")
        os.unlink(test_db_file)
    except FileNotFoundError:
        print(f"Could not delete {test_db_file} file containing the test DB")
        pass


@pytest.fixture(scope="module")
def cursor(connection):
    cursor = connection.cursor()

    # execute db initialisation script
    cursor.executescript(SCHEMA_SQL)

    # insert default values
    cursor.executescript(DEFAULT_VALUES_SQL)

    return cursor


# scope defaults to "function" which means that the fixture is run once per test (function)
@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)


# TODO improve this (lots of duplicated code) - some test cases need a DB connection that is not shared with other tests
@pytest.fixture
def connection_unique():
    print("Creating a new unique SQLITE connection to the test DB")
    test_db_file = "test_unique.sqlite"
    connection: Connection = sqlite3.connect(test_db_file)

    # yield the created connection
    yield connection

    # after tests have finished delete the created db file
    try:
        print("Deleting the unique test DB")
        os.unlink(test_db_file)
    except FileNotFoundError:
        print(f"Could not delete {test_db_file} file containing the unique test DB")
        pass


@pytest.fixture
def cursor_unique(connection_unique):
    cursor = connection_unique.cursor()

    # execute db initialisation script
    cursor.executescript(SCHEMA_SQL)

    # insert default values
    cursor.executescript(DEFAULT_VALUES_SQL)

    return cursor


# scope defaults to "function" which means that the fixture is run once per test (function)
@pytest.fixture
def certificate_repository_unique(connection_unique, cursor_unique):
    return CertificateRepository(connection_unique, cursor_unique)


@pytest.fixture
def private_key_repository_unique(connection_unique, cursor_unique):
    return PrivateKeyRepository(connection_unique, cursor_unique)


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


@pytest.fixture
def private_key_service_unique(private_key_repository_unique, cryptography_service_unique):
    return KeyService(cryptography_service_unique, private_key_repository_unique)


@pytest.fixture
def certificate_service_unique(certificate_repository_unique, cryptography_service_unique):
    return CertificateService(cryptography_service_unique, certificate_repository_unique)