OnroerendErfgoed / oe_utils

tests/test_audit.py

# -*- coding: utf-8 -*-
import unittest
from oe_utils.audit import audit, audit_with_request
from oe_utils.data.data_managers import AuditManager
from oe_utils.data.models import Wijziging
from oe_utils.data.db_audit_manager import includeme
from oe_utils.data.db_audit_manager import audit_manager

try:
    from unittest.mock import Mock, MagicMock
except ImportError:
    from mock import Mock, MagicMock, call  # pragma: no cover
try:
    import configparser
except ImportError:
    import ConfigParser as configparser
import os
from sqlalchemy import engine_from_config
from sqlalchemy.orm import sessionmaker
from zope.sqlalchemy import ZopeTransactionExtension
from oe_utils.data.models import Base
from jsonpublish import register_adapter


def init_test_db(engine):
    Base.metadata.drop_all(engine)
    Base.metadata.create_all(engine)
    session_maker = sessionmaker(
        bind=engine
    )
    session = session_maker()

    session.commit()
    session.close()


class Resource:
    def __init__(self):
        self.id = None,
        self.naam = None
        self.voornaam = None


@register_adapter(Resource)
def resource_adapter(obj, request=None):
    return {
        'id': obj.id,
        'naam': obj.naam,
        'voornaam': obj.voornaam
}


class DummyParent(object):

    def __init__(self):
        self.request = MagicMock()

    @audit()
    def dummy(self):
        resource = Resource()
        resource.id = 123
        resource.naam = 'crash test'
        resource.voornaam = 'dummy'
        return resource

    @audit()
    def dummy_none(self):
        return 'test'

    @audit(result_id_key='naam')
    def dummy_other_id(self):
        resource = Resource()
        resource.id = 123
        resource.naam = '456'
        resource.voornaam = 'dummy'
        return resource

    @audit(actie='bijlage bewerken')
    def dummy_bijlage(self):
        return None


@audit_with_request(actie='bijlage blijven bewerken')
def dummy_bijlage_with_request(request):
    return None


class AuditTests(unittest.TestCase):

    @classmethod

This code seems to be duplicated in your project.

    def setUpClass(cls):
        config = configparser.ConfigParser()
        config.read(os.path.join(os.path.dirname(__file__), 'test.ini'))
        settings = config.items('app:oe_utils')
        settings = dict((s[0], s[1]) for s in settings)
        cls.engine = engine_from_config(settings, prefix='sqlalchemy.')
        cls.session_maker = sessionmaker(
            bind=cls.engine,
            extension=ZopeTransactionExtension()
        )

    def setUp(self):
        init_test_db(self.engine)
        self.session = self.session_maker()
        self.dummy_parent = DummyParent()
        self.audit_manager = AuditManager(self.session)
        self.dummy_parent.request.audit_manager = self.audit_manager

    def tearDown(self):
        self.session.close()

    def test_audit_get(self):

This code seems to be duplicated in your project.

        self.dummy_parent.request.user = {'actor': {'uri': 'uri', 'omschrijving': 'omschrijving'}}
        self.dummy_parent.request.method = 'GET'
        self.dummy_parent.request.matchdict = {'id': 123}
        self.dummy_parent.dummy()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('opvragen', wijziging.actie)
        self.assertEqual('omschrijving', wijziging.updated_by_omschrijving)
        self.assertEqual('crash test', wijziging.resource_object_json['naam'])

    def test_audit_put(self):
        self.dummy_parent.request.method = 'PUT'
        self.dummy_parent.request.matchdict = {'id': 123}
        self.dummy_parent.dummy()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('bewerken', wijziging.actie)
        self.assertEqual('crash test', wijziging.resource_object_json['naam'])
        self.assertEqual('publiek', wijziging.updated_by_omschrijving)

    def test_audit_post(self):
        self.dummy_parent.request.method = 'POST'
        self.dummy_parent.dummy()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('aanmaken', wijziging.actie)
        self.assertEqual('crash test', wijziging.resource_object_json['naam'])
        self.assertEqual('crash test', wijziging.resource_object_json['naam'])

    def test_audit_delete(self):

This code seems to be duplicated in your project.

        self.dummy_parent.request.method = 'DELETE'
        self.dummy_parent.request.matchdict = {'id': 123}
        self.dummy_parent.dummy_none()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('verwijderen', wijziging.actie)
        self.assertEqual(123, wijziging.resource_object_id)
        self.assertIsNone(wijziging.resource_object_json)

    def test_audit_other(self):

This code seems to be duplicated in your project.

        self.dummy_parent.request.method = 'HEAD'
        self.dummy_parent.request.matchdict = {'id': 123}
        self.dummy_parent.dummy_none()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('onbekend', wijziging.actie)
        self.assertEqual(123, wijziging.resource_object_id)
        self.assertIsNone(wijziging.resource_object_json)

    def test_audit_bijlage(self):

This code seems to be duplicated in your project.

        self.dummy_parent.request.method = 'PUT'
        self.dummy_parent.request.matchdict = {'id': 123}
        self.dummy_parent.dummy_bijlage()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('bijlage bewerken', wijziging.actie)
        self.assertEqual(123, wijziging.resource_object_id)
        self.assertIsNone(wijziging.resource_object_json)

    def test_db_audit_manager(self):
        config = Mock()
        includeme(config)
        request = MagicMock()
        request.db = self.session
        manager = audit_manager(request)
        self.assertIsInstance(manager, AuditManager)

    def test_audit_bijlage_with_request(self):
        request = MagicMock()
        request.audit_manager = AuditManager(self.session)
        request.method = 'PUT'
        request.matchdict = {'id': 1234}
        dummy_bijlage_with_request(request)
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('bijlage blijven bewerken', wijziging.actie)
        self.assertEqual(1234, wijziging.resource_object_id)
        self.assertIsNone(wijziging.resource_object_json)

    def test_audit_other_id_key(self):

This code seems to be duplicated in your project.

        self.dummy_parent.request.user = {'actor': {'uri': 'uri', 'omschrijving': 'omschrijving'}}
        self.dummy_parent.request.method = 'GET'
        self.dummy_parent.request.matchdict = {'id': None}
        self.dummy_parent.dummy_other_id()
        wijziging = self.session.query(Wijziging).order_by(Wijziging.updated_at.desc()).first()
        self.assertEqual('opvragen', wijziging.actie)
        self.assertEqual('omschrijving', wijziging.updated_by_omschrijving)
        self.assertEqual('456', wijziging.resource_object_json['naam'])
        self.assertEqual(456, wijziging.resource_object_id)