fix: modified DirectoryDataComponent to support user defined custom file types (#4017)

test_data_components.py: Updated 'test_directory_component_build_with_multithreading' to expect an extra argument
This commit is contained in:
EDllT 2024-10-07 14:28:12 +03:00 committed by GitHub
commit 79a1257b7d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 6 deletions

View file

@ -1,7 +1,7 @@
import os
import tempfile
from pathlib import Path
from unittest.mock import Mock, patch
from unittest.mock import Mock, patch, ANY
import httpx
import pytest
@ -158,7 +158,7 @@ def test_directory_component_build_with_multithreading(
# Assert
mock_resolve_path.assert_called_once_with(path)
mock_retrieve_file_paths.assert_called_once_with(path, load_hidden, recursive, depth)
mock_retrieve_file_paths.assert_called_once_with(path, load_hidden, recursive, depth, ANY)
mock_parallel_load_data.assert_called_once_with(
mock_retrieve_file_paths.return_value, silent_errors, max_concurrency
)