data-validation-engine/tests/test_core_engine/test_models.py at 751766cf7f6dfd9ebc0bbd858e2ef3c0bb83b11d · NHSDigital/data-validation-engine · 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
"""Unit tests for core engine models."""
from typing import Any, Dict, Tuple
from uuid import uuid4

import pytest

from dve.core_engine.models import SubmissionInfo

CONSTANT_SUBMISSION_ID = uuid4().hex


@pytest.mark.parametrize(
    "testcases",
    [
        # perfect submission info
        (
            {
                "submitted": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test0",
                    "file_name": "my_file",
                    "file_extension": "csv",
                },
                "expected": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test0",
                    "file_name": "my_file",
                    "file_extension": "csv",
                },
            },
        ),
        # submission with multiple file extensions within the file name
        (
            {
                "submitted": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test1",
                    "file_name": "my_file",
                    "file_extension": ".csv.csv",
                },
                "expected": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test1",
                    "file_name": "my_file",
                    "file_extension": "csv",
                },
            },
        ),
        # submission with multiple file extensions
        (
            {
                "submitted": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test2",
                    "file_name": "my_file",
                    "file_extension": "csv.csv.xml",
                },
                "expected": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test2",
                    "file_name": "my_file",
                    "file_extension": "xml",
                },
            },
        ),
        # submission with multiple file extensions
        (
            {
                "submitted": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test2",
                    "file_name": "my.file.perfect",
                    "file_extension": "csv",
                },
                "expected": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test2",
                    "file_name": "my.file.perfect",
                    "file_extension": "csv",
                },
            },
        ),
        # submission with multiple file extensions
        (
            {
                "submitted": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test2",
                    "file_name": "m_y%fil\ne.perfect",
                    "file_extension": "csv",
                },
                "expected": {
                    "submission_id": CONSTANT_SUBMISSION_ID,
                    "dataset_id": "test2",
                    "file_name": "m_y%fil\ne.perfect",
                    "file_extension": "csv",
                },
            },
        ),
    ],
)
def test_submission_info(  # pylint: disable=missing-function-docstring
    testcases: Tuple[Dict[str, Dict[str, Any]]],
):
    ignore = [
        "date_updated",
        "time_updated",
        "submission_method",
        "submitting_org",
        "reporting_period_start",
        "reporting_period_end",
        "file_size",
        "datetime_received",
    ]
    for testcase in testcases:
        actual = SubmissionInfo(**testcase["submitted"])
        expected = testcase["expected"]

        assert {k: v for k, v in actual.dict().items() if k not in ignore} == expected
        assert actual.file_name_with_ext == f"{expected['file_name']}.{expected['file_extension']}"


def test_submission_info_eq():  # pylint: disable=missing-function-docstring
    data = {
        "submission_id": uuid4().hex,
        "dataset_id": "test0",
        "file_name": "my_file.csv",
        "file_extension": "csv",
    }
    assert SubmissionInfo(**data) == SubmissionInfo(**data)  # type: ignore