Skip to content

Add conversion test from entity to file (ECC-7368) #27

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
seebi merged 1 commit into from
May 21, 2025
Merged

Add conversion test from entity to file (ECC-7368) #27

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 16 additions & 1 deletion tests/typed_entities/test_typed_entities.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,10 @@
from collections.abc import Sequence
from pathlib import Path

import pytest

from cmem_plugin_base.dataintegration.context import ExecutionContext
from cmem_plugin_base.dataintegration.entity import Entities
from cmem_plugin_base.dataintegration.entity import Entities, Entity
from cmem_plugin_base.dataintegration.plugins import WorkflowPlugin
from cmem_plugin_base.dataintegration.ports import FixedNumberOfInputs, FixedSchemaPort
from cmem_plugin_base.dataintegration.typed_entities.file import FileEntitySchema, LocalFile
Expand Down Expand Up @@ -59,6 +61,19 @@ def test_files(self) -> None:
output_str = output_file.read()
assert output_str == "ABC123"

def test_file_entity_conversion(self) -> None:
"""Test conversion from entity to file"""
file_entity = Entity(uri="test.uri", values=[["test.txt"], ["Project"], []])
assert FileEntitySchema().from_entity(file_entity)

file_entity = Entity(uri="test.uri", values=[["test.txt"], ["Project"], [""]])
assert FileEntitySchema().from_entity(file_entity)

with pytest.raises(ValueError, match="File 'test.txt' has unexpected type 'Wrong Type'"):
FileEntitySchema().from_entity(
Entity(uri="test.uri", values=[["test.txt"], ["Wrong Type"], []])
)


if __name__ == "__main__":
unittest.main()