Fix path

pyfireconsole/models/pyfire_model.py

@@ -376,6 +376,7 @@ def find(cls, path: str, allow_empty: bool = False) -> 'PyfireDoc':
         if path.find('/') == -1:
             collection_name = cls.collection_name()
             id = path
+            path = f"{collection_name}/{id}"
         else:
             collection_name, id = path.rsplit('/', 1)
 

tests/test_usecase.py

@@ -106,6 +106,35 @@ def test_find_not_found(mock_db):
         Book.find("99999")
 
 
+def test_find_obj_ref_key(mock_db):
+    book = Book.new(
+        title="Math",
+        user_id="12345",
+        published_at=datetime.now(),
+        authors=["John", "Mary"],
+        publisher_ref="publisher/12345",
+    ).save()
+
+    assert book.obj_ref_key() == f"books/{book.id}"
+    assert book.obj_collection_name() == "books"
+    assert Book.find(book.id).obj_ref_key() == f"books/{book.id}"
+
+
+def test_deep_find_obj_ref_key(mock_db):
+    book = Book.new(
+        title="Math",
+        user_id="12345",
+        published_at=datetime.now(),
+        authors=["John", "Mary"],
+        publisher_ref=Publisher.new(name="publisher1").save().obj_ref_key(),
+    ).save()
+
+    tag = book.tags.add(Tag.new(name="mathmatics"))
+
+    assert tag.obj_ref_key() == f"books/{book.id}/tags/{tag.id}"
+    assert Tag.find(f"books/{book.id}/tags/{tag.id}").obj_ref_key() == f"books/{book.id}/tags/{tag.id}"
+
+
 def test_subcollection(mock_db):
     book = Book.new(
         title="Math",