|
8 | 8 | import pytest # type: ignore |
9 | 9 | from nisystemlink.clients.core import ApiException |
10 | 10 | from nisystemlink.clients.file import FileClient |
11 | | -from nisystemlink.clients.file.models import LinqFileQueryRequest, UpdateMetadataRequest |
| 11 | +from nisystemlink.clients.file.models import FileLinqQueryRequest, UpdateMetadataRequest |
12 | 12 | from nisystemlink.clients.file.utilities import rename_file |
13 | 13 |
|
14 | 14 | FILE_NOT_FOUND_ERR = "Not Found" |
@@ -209,47 +209,38 @@ def test__back_off_retry__works(self, client: FileClient, test_file): |
209 | 209 | def test__query_files_linq__filter_by_name_succeeds( |
210 | 210 | self, client: FileClient, test_file, random_filename_extension: str |
211 | 211 | ): |
212 | | - """Test LINQ query filtering by file name.""" |
213 | | - # Upload the test file with random name |
214 | 212 | file_id = test_file(file_name=random_filename_extension, cleanup=False) |
215 | 213 |
|
216 | | - try: |
217 | | - # Query by exact name match |
218 | | - query_request = LinqFileQueryRequest( |
219 | | - filter=f'name == "{random_filename_extension}"' |
220 | | - ) |
221 | | - response = client.query_files_linq(query=query_request) |
222 | | - |
223 | | - assert response.available_files is not None |
224 | | - assert len(response.available_files) == 1 |
225 | | - assert response.available_files[0].id == file_id |
226 | | - assert response.available_files[0].properties is not None |
227 | | - assert ( |
228 | | - response.available_files[0].properties["Name"] |
229 | | - == random_filename_extension |
230 | | - ) |
231 | | - assert response.total_count.value == 1 |
232 | | - assert response.total_count.relation == "eq" |
233 | | - |
234 | | - finally: |
235 | | - client.delete_file(id=file_id) |
| 214 | + query_request = FileLinqQueryRequest( |
| 215 | + filter=f'name == "{random_filename_extension}"' |
| 216 | + ) |
| 217 | + response = client.query_files_linq(query=query_request) |
| 218 | + |
| 219 | + assert response.available_files is not None |
| 220 | + assert len(response.available_files) == 1 |
| 221 | + assert response.available_files[0].id == file_id |
| 222 | + assert response.available_files[0].properties is not None |
| 223 | + assert ( |
| 224 | + response.available_files[0].properties["Name"] |
| 225 | + == random_filename_extension |
| 226 | + ) |
| 227 | + assert response.total_count.value == 1 |
| 228 | + assert response.total_count.relation == "eq" |
| 229 | + |
| 230 | + client.delete_file(id=file_id) |
236 | 231 |
|
237 | 232 | def test__query_files_linq__invalid_filter_raises(self, client: FileClient): |
238 | | - """Test LINQ query with invalid filter syntax raises exception.""" |
239 | | - # Query with malformed filter |
240 | | - query_request = LinqFileQueryRequest(filter="invalid filter syntax:") |
| 233 | + query_request = FileLinqQueryRequest(filter="invalid filter syntax:") |
241 | 234 |
|
242 | 235 | with pytest.raises(ApiException): |
243 | 236 | client.query_files_linq(query=query_request) |
244 | 237 |
|
245 | 238 | def test__query_files_linq__filter_returns_no_results(self, client: FileClient): |
246 | | - """Test LINQ query with valid filter that matches no files.""" |
247 | | - # Query with a filter that should match no existing files |
248 | 239 | unique_nonexistent_name = ( |
249 | 240 | f"{PREFIX}nonexistent_file_{randint(100000, 999999)}.random_extension" |
250 | 241 | ) |
251 | 242 |
|
252 | | - query_request = LinqFileQueryRequest( |
| 243 | + query_request = FileLinqQueryRequest( |
253 | 244 | filter=f'name == "{unique_nonexistent_name}"' |
254 | 245 | ) |
255 | 246 | response = client.query_files_linq(query=query_request) |
|
0 commit comments