|
18 | 18 |
|
19 | 19 | class TestFileKeyValidation(TestCase): |
20 | 20 | """Tests for file_key_validation functions""" |
| 21 | + |
21 | 22 | def test_is_file_in_directory_root(self): |
22 | 23 | test_cases = [ |
23 | 24 | ("test_file.csv", True), |
@@ -79,37 +80,48 @@ def test_validate_file_key(self, mock_hkeys, mock_hget): |
79 | 80 | # File key with no '.' |
80 | 81 | (VALID_FLU_EMIS_FILE_KEY.replace(".", ""), "Initial file validation failed: missing file extension"), |
81 | 82 | # File key with additional '.' in vaccine type |
82 | | - (VALID_FLU_EMIS_FILE_KEY[:2] + "." + VALID_FLU_EMIS_FILE_KEY[2:],"Initial file validation failed: unsupported vaccine type"), |
| 83 | + (VALID_FLU_EMIS_FILE_KEY[:2] + "." + VALID_FLU_EMIS_FILE_KEY[2:], |
| 84 | + "Initial file validation failed: unsupported vaccine type"), |
83 | 85 | # File key with additional '_' in vaccine type |
84 | | - (VALID_FLU_EMIS_FILE_KEY[:2] + "_" + VALID_FLU_EMIS_FILE_KEY[2:],"Initial file validation failed: unsupported vaccine type"), |
| 86 | + (VALID_FLU_EMIS_FILE_KEY[:2] + "_" + VALID_FLU_EMIS_FILE_KEY[2:], |
| 87 | + "Initial file validation failed: unsupported vaccine type"), |
85 | 88 | # File key with missing '_' |
86 | | - (VALID_FLU_EMIS_FILE_KEY.replace("_", "", 1), "Initial file validation failed: not enough parts in file key"), |
| 89 | + (VALID_FLU_EMIS_FILE_KEY.replace("_", "", 1), |
| 90 | + "Initial file validation failed: not enough parts in file key"), |
87 | 91 | # File key with missing '_' |
88 | 92 | (VALID_FLU_EMIS_FILE_KEY.replace("_", ""), "Initial file validation failed: not enough parts in file key"), |
89 | 93 | # File key with missing extension |
90 | 94 | (VALID_FLU_EMIS_FILE_KEY.replace(".csv", ""), "Initial file validation failed: missing file extension"), |
91 | 95 | # File key with invalid vaccine type |
92 | | - (VALID_FLU_EMIS_FILE_KEY.replace("FLU", "Flue"), "Initial file validation failed: unsupported vaccine type"), |
| 96 | + (VALID_FLU_EMIS_FILE_KEY.replace("FLU", "Flue"), |
| 97 | + "Initial file validation failed: unsupported vaccine type"), |
93 | 98 | # File key with missing vaccine type |
94 | 99 | (VALID_FLU_EMIS_FILE_KEY.replace("FLU", ""), "Initial file validation failed: unsupported vaccine type"), |
95 | 100 | # File key with invalid vaccinations element |
96 | | - (VALID_FLU_EMIS_FILE_KEY.replace("Vaccinations", "Vaccination"),"Initial file validation failed: file key must contain VACCINATIONS"), |
| 101 | + (VALID_FLU_EMIS_FILE_KEY.replace("Vaccinations", "Vaccination"), |
| 102 | + "Initial file validation failed: file key must contain VACCINATIONS"), |
97 | 103 | # File key with missing vaccinations element |
98 | | - (VALID_FLU_EMIS_FILE_KEY.replace("Vaccinations", ""), "Initial file validation failed: file key must contain VACCINATIONS"), |
| 104 | + (VALID_FLU_EMIS_FILE_KEY.replace("Vaccinations", ""), |
| 105 | + "Initial file validation failed: file key must contain VACCINATIONS"), |
99 | 106 | # File key with invalid version |
100 | 107 | (VALID_FLU_EMIS_FILE_KEY.replace("v5", "v4"), "Initial file validation failed: unsupported file version"), |
101 | 108 | # File key with missing version |
102 | 109 | (VALID_FLU_EMIS_FILE_KEY.replace("v5", ""), "Initial file validation failed: unsupported file version"), |
103 | 110 | # File key with invalid ODS code |
104 | | - (VALID_FLU_EMIS_FILE_KEY.replace("YGM41", "YGAM"), "Initial file validation failed: unregistered supplier ODS code"), |
| 111 | + (VALID_FLU_EMIS_FILE_KEY.replace("YGM41", "YGAM"), |
| 112 | + "Initial file validation failed: unregistered supplier ODS code"), |
105 | 113 | # File key with missing ODS code |
106 | | - (VALID_FLU_EMIS_FILE_KEY.replace("YGM41", ""), "Initial file validation failed: unregistered supplier ODS code"), |
| 114 | + (VALID_FLU_EMIS_FILE_KEY.replace("YGM41", ""), |
| 115 | + "Initial file validation failed: unregistered supplier ODS code"), |
107 | 116 | # File key with invalid timestamp |
108 | | - (VALID_FLU_EMIS_FILE_KEY.replace("20000101T00000001", "20200132T12345600"),"Initial file validation failed: invalid timestamp"), |
| 117 | + (VALID_FLU_EMIS_FILE_KEY.replace("20000101T00000001", "20200132T12345600"), |
| 118 | + "Initial file validation failed: invalid timestamp"), |
109 | 119 | # File key with missing timestamp |
110 | | - (VALID_FLU_EMIS_FILE_KEY.replace("20000101T00000001", ""),"Initial file validation failed: invalid timestamp"), |
| 120 | + (VALID_FLU_EMIS_FILE_KEY.replace("20000101T00000001", ""), |
| 121 | + "Initial file validation failed: invalid timestamp"), |
111 | 122 | # File key with incorrect extension |
112 | | - (VALID_FLU_EMIS_FILE_KEY.replace(".csv", ".xlsx"), "Initial file validation failed: unsupported file extension"), |
| 123 | + (VALID_FLU_EMIS_FILE_KEY.replace(".csv", ".xlsx"), |
| 124 | + "Initial file validation failed: unsupported file extension"), |
113 | 125 | ] |
114 | 126 |
|
115 | 127 | for file_key, expected_result in test_cases_for_failure_scenarios: |
|
0 commit comments