base.py

import unittest
from pathlib import Path
from typing import Any, Optional

import psycopg2
from flask import url_for

from openatlas import app
from openatlas.models.entity import Entity
from openatlas.models.type import Type


class TestBaseCase(unittest.TestCase):

    def setUp(self) -> None:
        app.testing = True
        app.config.from_pyfile('testing.py')
        self.setup_database()
        self.client = app.test_client()
        with app.app_context():
            self.client.post(
                url_for('login'),
                data={'username': 'Alice', 'password': 'test'})
            with app.test_request_context():
                app.preprocess_request()
                self.alice_id = 2
                self.precision_type = \
                    Type.get_hierarchy('External reference match')
                self.test_path = Path(app.root_path).parent / 'tests'
                self.static_path = Path(app.root_path) / 'static'
        app.app_context().push()

    def setup_database(self) -> None:
        connection = psycopg2.connect(
            database=app.config['DATABASE_NAME'],
            host=app.config['DATABASE_HOST'],
            user=app.config['DATABASE_USER'],
            password=app.config['DATABASE_PASS'],
            port=app.config['DATABASE_PORT'])
        connection.autocommit = True
        self.cursor = connection.cursor()
        for file_name in [
                '0_extensions',
                '1_structure',
                '2_data_model',
                '3_data_web',
                '4_data_type',
                'data_test']:
            with open(
                    Path(app.root_path).parent / 'install' /
                    f'{file_name}.sql', encoding='utf8') as sql_file:
                self.cursor.execute(sql_file.read())
        if app.config['LOAD_WINDOWS_TEST_SQL']:  # pragma: no cover
            with open(
                    Path(app.root_path).parent / 'install' /
                    'data_test_windows.sql', encoding='utf8') as sql_file:
                self.cursor.execute(sql_file.read())


class ApiTestCase(TestBaseCase):

    def setUp(self) -> None:
        super().setUp()
        with open(
                Path(app.root_path).parent / 'install' / 'data_test_api.sql',
                encoding='utf8') as sql_file:
            self.cursor.execute(sql_file.read())

    @staticmethod
    def get_classes(data: list[dict[str, Any]]) -> bool:
        return (data[0]['systemClass']
                and data[0]['crmClass']
                and data[0]['view']
                and data[0]['icon']
                and data[0]['en'])

    @staticmethod
    def get_class_mapping(data: dict[str, Any], locale: str) -> bool:
        return (data['locale'] == locale
                and data['results'][0]['systemClass']
                and data['results'][0]['crmClass']
                and data['results'][0]['view']
                and data['results'][0]['icon']
                and data['results'][0]['label'])


class ExportImportTestCase(TestBaseCase):
    def setUp(self) -> None:
        super().setUp()
        with open(
                Path(app.root_path).parent / 'install' / 'data_test_api.sql',
                encoding='utf8') as sql_file:
            self.cursor.execute(sql_file.read())


def insert(
        class_: str,
        name: str,
        description: Optional[str] = None) -> Entity:
    entity = Entity.insert(class_, name, description)
    if class_ in ['artifact', 'feature', 'place', 'stratigraphic_unit']:
        entity.link(
            'P53',
            Entity.insert('object_location', f'Location of {name}'))
    return entity


def get_hierarchy(name: str) -> Type:
    return Type.get_hierarchy(name)