mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-07-08 17:46:54 +00:00
60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
|
from datetime import datetime
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from unstructured.ingest.connector.sharepoint import SharepointIngestDoc
|
||
|
|
from unstructured.ingest.interfaces import ProcessorConfig, ReadConfig
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
("time_created", "time_last_modified", "expected_created", "expected_modified"),
|
||
|
|
[
|
||
|
|
(
|
||
|
|
"2023-06-16T05:05:05+00:00",
|
||
|
|
datetime(2023, 6, 16, 5, 5, 5),
|
||
|
|
"2023-06-16T05:05:05+00:00",
|
||
|
|
"2023-06-16T05:05:05",
|
||
|
|
),
|
||
|
|
("2023-06-16 05:05:05", "2023-06-16", "2023-06-16T05:05:05", "2023-06-16T00:00:00"),
|
||
|
|
# Add more pairs of input strings and their expected ISO format results here
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_datetime_handling_in_update_source_metadata(
|
||
|
|
mocker, time_created, time_last_modified, expected_created, expected_modified
|
||
|
|
):
|
||
|
|
"""Test the handling of various datetime formats in update_source_metadata."""
|
||
|
|
# Create a mock SharePoint response directly in the test
|
||
|
|
mock_sharepoint_response = mocker.MagicMock()
|
||
|
|
mock_sharepoint_response.time_created = time_created
|
||
|
|
mock_sharepoint_response.time_last_modified = time_last_modified
|
||
|
|
|
||
|
|
# Patch the SharePoint interaction methods to use the mock response
|
||
|
|
mocker.patch(
|
||
|
|
"unstructured.ingest.connector.sharepoint.SharepointIngestDoc._fetch_file",
|
||
|
|
return_value=mock_sharepoint_response,
|
||
|
|
)
|
||
|
|
mocker.patch(
|
||
|
|
"unstructured.ingest.connector.sharepoint.SharepointIngestDoc._fetch_page",
|
||
|
|
return_value=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Instantiate your document with dummy data
|
||
|
|
ingest_doc = SharepointIngestDoc(
|
||
|
|
connector_config=MagicMock(),
|
||
|
|
site_url="dummy_url",
|
||
|
|
server_path="dummy_path",
|
||
|
|
is_page=False,
|
||
|
|
file_path="dummy_path.html",
|
||
|
|
processor_config=ProcessorConfig(),
|
||
|
|
read_config=ReadConfig(),
|
||
|
|
)
|
||
|
|
|
||
|
|
# Execute the method under test
|
||
|
|
ingest_doc.update_source_metadata()
|
||
|
|
|
||
|
|
# Assertions to verify the datetime handling against expected results
|
||
|
|
assert ingest_doc.source_metadata is not None
|
||
|
|
assert ingest_doc.source_metadata.date_created.startswith(expected_created)
|
||
|
|
assert ingest_doc.source_metadata.date_modified.startswith(expected_modified)
|