Update test_metadata_enricher.py

This commit is contained in:
erinhmclark
2025-02-18 19:46:57 +00:00
parent ce5a200d1f
commit 7b88df72cb
4 changed files with 5 additions and 5 deletions

View File

@@ -76,14 +76,14 @@ def test_get_metadata_error_handling(mock_run, mock_logger_error, enricher):
mock_logger_error.assert_called_once()
@pytest.mark.skip(reason="Requires ExifTool to be installed. TODO mock")
def test_metadata_pickle(enricher, unpickle):
# Uses a pickle of a YouTube short
@patch("subprocess.run")
def test_metadata_pickle(mock_run, enricher, unpickle):
# Uses pickled values
mock_run.return_value = unpickle("metadata_enricher_exif.pickle")
metadata = unpickle("metadata_enricher_ytshort_input.pickle")
expected = unpickle("metadata_enricher_ytshort_expected.pickle")
enricher.enrich(metadata)
expected_media = expected.media
actual_media = metadata.media
assert len(expected_media) == len(actual_media)
assert actual_media[0].properties.get("metadata") == expected_media[0].properties.get("metadata")
assert metadata == expected
assert actual_media[0].properties.get("metadata") == expected_media[0].properties.get("metadata")