OnroerendErfgoed / oe_utils

tests/test_views.py

# -*- coding: utf-8 -*-
import os
import unittest

import pretend
from cornice import Service
from pyramid.httpexceptions import HTTPBadRequest, HTTPNotFound
from sqlalchemy import engine_from_config
from sqlalchemy.orm import sessionmaker

from oe_utils.data.data_managers import DataManager
from oe_utils.views import get_resource, get_json_from_request, RouteConfigArgs
from tests import init_test_db, DummyDossier, DummySchema

from zope.sqlalchemy import ZopeTransactionExtension

try:
    from unittest.mock import Mock, patch, MagicMock
except ImportError:
    from mock import Mock, patch, MagicMock
try:
    import configparser
except ImportError:
    import ConfigParser as configparser

service = Service('dummy_service', path='/test')


class ViewsTests(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()
        )
        init_test_db(cls.engine)

    def setUp(self):
        self.session = self.session_maker()
        self.data_manager = DataManager(self.session, DummyDossier)

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

    def test_get_resource_not_found(self):
        self.assertRaises(HTTPNotFound, get_resource, self.data_manager, 987654321)

    def test_get_resource(self):
        get_resource(self.data_manager, 1)

    def test_get_json_from_request_no_json_body(self):
        request = pretend.stub(method="POST", path='/test')
        with self.assertRaises(HTTPBadRequest) as e:
            get_json_from_request(request)
        self.assertIn('Request bevat geen json body. \n', e.exception.__str__())

    def test_get_json_from_request(self):
        request = pretend.stub(method='POST', path='/test', json_body={'test': 'test'})
        res = get_json_from_request(request)
        self.assertEqual({'test': 'test'}, res)


class RouteConfigArgsTests(unittest.TestCase):
    def setUp(self):
        self.route_config_args = RouteConfigArgs([], DummySchema, tags=['Test'])

    def tearDown(self):
        del self.route_config_args

    def test_init(self):
        self.assertEquals(['Test'], self.route_config_args.tags)
        self.assertEquals([], self.route_config_args.response_schemas)
        self.assertEquals(DummySchema, self.route_config_args.schema)