mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
863e0a4e24
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from basic_memory.schemas.response import RelationResponse
|
|
|
|
|
|
def test_relation_response_resolves_from_to_from_dict_fallbacks():
|
|
data = {
|
|
"permalink": "rel/1",
|
|
"relation_type": "relates_to",
|
|
"context": "ctx",
|
|
"to_name": None,
|
|
"from_entity": {"permalink": None, "file_path": "From.md"},
|
|
"to_entity": {"permalink": None, "file_path": "To.md", "title": "To Title"},
|
|
}
|
|
|
|
rel = RelationResponse.model_validate(data)
|
|
assert rel.from_id == "From.md"
|
|
assert rel.to_id == "To.md"
|
|
assert rel.to_name == "To Title"
|
|
|
|
|
|
def test_relation_response_resolves_from_to_from_orm_like_object_fallbacks():
|
|
class EntityLike:
|
|
def __init__(self, permalink, file_path, title=None):
|
|
self.permalink = permalink
|
|
self.file_path = file_path
|
|
self.title = title
|
|
|
|
class RelationLike:
|
|
def __init__(self):
|
|
self.permalink = "rel/2"
|
|
self.relation_type = "relates_to"
|
|
self.context = "ctx"
|
|
self.to_name = None
|
|
self.from_entity = EntityLike(permalink=None, file_path="From2.md")
|
|
self.to_entity = EntityLike(permalink=None, file_path="To2.md", title="To2 Title")
|
|
|
|
rel = RelationResponse.model_validate(RelationLike())
|
|
assert rel.from_id == "From2.md"
|
|
assert rel.to_id == "To2.md"
|
|
assert rel.to_name == "To2 Title"
|