DocumentEngineTest

Complexity

Total Complexity

4

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	test_get_document()	0	6	1
A	setUp()	0	4	1
A	tearDown()	0	2	1
A	test_create_documents()	0	15	1

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

import json
import responses
from oe_utils.document.documentengine import DocumentEngine
from tests import init_document_responses

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

test_base_url = 'https://dgen.onroerenderfgoed.be'
test_template_id = 'test_template_id'
test_system_token = 'test_system_token'

here = os.path.dirname(os.path.abspath(__file__))
with open(os.path.join(here, 'fixtures', 'Test_document.pdf'), 'rb') as test_document:
    data = test_document.read()


class DocumentEngineTest(unittest.TestCase):
    def setUp(self):
        init_document_responses()
        self.documentengine = DocumentEngine(test_base_url, test_template_id)
        self.system_token = 'test_token'

    def tearDown(self):
        pass

    @responses.activate
    def test_create_documents(self):
        generation_id = self.documentengine.create_documents({'id': 'test_id'}, test_system_token,
                                                             'https://dgen.onroerenderfgoed.be/generations/1/document',
                                                             'POST',
                                                             '1')
        self.assertEqual(1, generation_id)
        self.assertEqual(len(responses.calls), 1)
        self.assertEqual(responses.calls[0].request.url, 'https://dgen.onroerenderfgoed.be/generations')
        self.assertDictEqual(json.loads(responses.calls[0].request.body),
                             {'template_id': test_template_id,
                              'template_version': '1',
                              'data': {'id': 'test_id'},
                              'callback_method': 'POST',
                              'callback_url': 'https://dgen.onroerenderfgoed.be/generations/1/document'}
                             )

    @responses.activate
    def test_get_document(self):
        res = self.documentengine.get_document(1, test_system_token)
        self.assertEqual(data, res)
        self.assertEqual(len(responses.calls), 1)
        self.assertEqual(responses.calls[0].request.url, 'https://dgen.onroerenderfgoed.be/generations/1/document')