ticket-decoder/source/test/python/test_decode_uic918.py at master · user4223/ticket-decoder · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# SPDX-FileCopyrightText: (C) 2022 user4223 and (other) contributors to ticket-decoder <https://github.com/user4223/ticket-decoder>
# SPDX-License-Identifier: GPL-3.0-or-later

from unittest import TestCase, skipIf
from json import loads
from pathlib import Path
from base64 import b64encode
from ticket_decoder import DecoderFacade

class TestDecodeUIC918(TestCase):

    @staticmethod
    def loadRaw(file_name) -> str:
        path = Path(__file__).parent.parent / 'interpreter' / 'etc' / file_name
        with open(path, mode='rb') as file:
            return b64encode(file.read())

    def test_decode_uic918(self):
        decoder_facade = DecoderFacade()
        result = loads(decoder_facade.decode_uic918(TestDecodeUIC918.loadRaw('Muster 918-9 CityTicket.raw')))
        assert result['records']['U_FLEX']['transportDocuments'][0]['openTicket']['fromStationName'] == 'Kassel+City'
        assert result['validated'] == 'false'

    def test_decode_uic918_fail(self):
        with self.assertRaisesRegex(RuntimeError, '^Decoding failed with: Interpretation failed, this could be due to version mismatch or missing implementation for the given type or just due to an error:.*'):
            decoder_facade = DecoderFacade()
            decoder_facade.decode_uic918('no base64 encoded UIC918 data')

    pdf_input_file = 'images/Muster-UIC918-9/Muster 918-9 CityTicket.pdf'

    @skipIf(not Path(pdf_input_file).exists(), "Missing input file: " + pdf_input_file)
    def test_decode_pdf_files(self):
        decoder_facade = DecoderFacade()
        result = decoder_facade.decode_files(self.pdf_input_file)
        if not len(result):
            self.skipTest("Poppler input support not compiled in: " + self.pdf_input_file)

        records = loads(result[0][1])
        assert records['records']['U_FLEX']['transportDocuments'][0]['openTicket']['fromStationName'] == 'Kassel+City'
        assert records['validated'] == 'false'

    image_input_file = 'images/Muster-UIC918-9/Muster 918-9 Deutschland-Ticket.png'

    @skipIf(not Path(image_input_file).exists(), "Missing input file: " + image_input_file)
    def test_decode_image_files(self):
        decoder_facade = DecoderFacade()
        result = decoder_facade.decode_files(self.image_input_file)

        records = loads(result[0][1])
        assert records['records']['U_FLEX']['transportDocuments'][0]['openTicket']['tariffs'][0]['tariffDesc'] == 'Deutschland-Ticket'
        assert records['validated'] == 'false'


    def test_decode_files_not_existing(self):
        with self.assertRaisesRegex(RuntimeError, '^Decoding failed with: Path to load input elements from does not exist: Not existing file$'):
            decoder_facade = DecoderFacade()
            decoder_facade.decode_files('Not existing file')

    def test_decode_files_without_aztec_code(self):
        decoder_facade = DecoderFacade()
        result = decoder_facade.decode_files('source/test/io/etc/minimal.pdf')
        assert len(result) == 0

    def test_two_instances(self):
        decoder_facadeA = DecoderFacade()
        decoder_facadeB = DecoderFacade()