refactor: add missing foreing_metadata attr to tests

This commit is contained in:
Igor Ilic 2025-01-20 17:38:28 +01:00
parent ab8d95cc30
commit 5c17501bb8
4 changed files with 7 additions and 0 deletions

View file

@ -29,6 +29,7 @@ def test_AudioDocument():
id=uuid.uuid4(), id=uuid.uuid4(),
name="audio-dummy-test", name="audio-dummy-test",
raw_data_location="", raw_data_location="",
foreign_metadata="",
mime_type="", mime_type="",
) )
with patch.object(AudioDocument, "create_transcript", return_value=TEST_TEXT): with patch.object(AudioDocument, "create_transcript", return_value=TEST_TEXT):

View file

@ -18,6 +18,7 @@ def test_ImageDocument():
id=uuid.uuid4(), id=uuid.uuid4(),
name="image-dummy-test", name="image-dummy-test",
raw_data_location="", raw_data_location="",
foreign_metadata="",
mime_type="", mime_type="",
) )
with patch.object(ImageDocument, "transcribe_image", return_value=TEST_TEXT): with patch.object(ImageDocument, "transcribe_image", return_value=TEST_TEXT):

View file

@ -32,6 +32,7 @@ def test_TextDocument(input_file, chunk_size):
id=uuid.uuid4(), id=uuid.uuid4(),
name=input_file, name=input_file,
raw_data_location=test_file_path, raw_data_location=test_file_path,
foreign_metadata="",
mime_type="", mime_type="",
) )

View file

@ -39,6 +39,7 @@ def test_UnstructuredDocument():
id=uuid.uuid4(), id=uuid.uuid4(),
name="example.pptx", name="example.pptx",
raw_data_location=pptx_file_path, raw_data_location=pptx_file_path,
foreign_metadata="",
mime_type="application/vnd.openxmlformats-officedocument.presentationml.presentation", mime_type="application/vnd.openxmlformats-officedocument.presentationml.presentation",
) )
@ -46,6 +47,7 @@ def test_UnstructuredDocument():
id=uuid.uuid4(), id=uuid.uuid4(),
name="example.docx", name="example.docx",
raw_data_location=docx_file_path, raw_data_location=docx_file_path,
foreign_metadata="",
mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document", mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
) )
@ -53,6 +55,7 @@ def test_UnstructuredDocument():
id=uuid.uuid4(), id=uuid.uuid4(),
name="example.csv", name="example.csv",
raw_data_location=csv_file_path, raw_data_location=csv_file_path,
foreign_metadata="",
mime_type="text/csv", mime_type="text/csv",
) )
@ -60,6 +63,7 @@ def test_UnstructuredDocument():
id=uuid.uuid4(), id=uuid.uuid4(),
name="example.xlsx", name="example.xlsx",
raw_data_location=xlsx_file_path, raw_data_location=xlsx_file_path,
foreign_metadata="",
mime_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", mime_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
) )