tests: add filesystem tests

Author: NTFSvolume

src/mega/data_structures.py

@@ -212,7 +212,7 @@ class Node(_DictDumper):
     type: NodeType
     attributes: Attributes
     created_at: TimeStamp
-    keys: MappingProxyType[UserID, str]
+    keys: MappingProxyType[UserID, str] = dataclasses.field(compare=False)
     share_owner: UserID | None
     share_key: str | None
 
@@ -250,7 +250,7 @@ def from_dump(cls, dump: dict[str, Any], /) -> Self:
         dump = dump | dict(  # noqa: C408
             type=NodeType[str(dump["type"]).upper()],
             attributes=Attributes(**dump["attributes"]) if dump["attributes"] else None,
-            _crypto=Crypto.from_dump(dump["_crypto"]),
+            _crypto=Crypto.from_dump(dump["_crypto"]) if dump["_crypto"] else None,
         )
 
         return cls(**dump)

src/mega/filesystem.py

@@ -1,14 +1,18 @@
+from __future__ import annotations
+
 import dataclasses
 import errno
-from collections.abc import Generator, Iterable, Iterator, Sequence
-from os import PathLike
 from pathlib import PurePosixPath
 from types import MappingProxyType
-from typing import Any, Self
+from typing import TYPE_CHECKING, Any, Self
 
 from mega.data_structures import Node, NodeID, NodeType, _DictDumper
 from mega.errors import MultipleNodesFoundError
 
+if TYPE_CHECKING:
+    from collections.abc import Generator, Iterable, Iterator, Sequence
+    from os import PathLike
+
 _POSIX_ROOT = PurePosixPath("/")
 
 

tests/fake_fs.json

@@ -421,7 +421,7 @@
       "owner": "me",
       "type": "file",
       "attributes": {
-        "name": "main.py",
+        "name": "logo.png",
         "label": "blue",
         "favorited": false
       },
@@ -868,7 +868,7 @@
     "MBlNlb2P": "/styles.css",
     "7N9QiWZ9": "/tests",
     "RDJJI2lv": "/tests/logo.png",
-    "pHWVIQLd": "/tests/main.py",
+    "pHWVIQLd": "/tests/logo.png",
     "oImwb6nN": "/tests/script.js",
     "FIXitv4F": "/tests/scripts",
     "0fPFklV3": "/tests/scripts/notes.txt",

tests/test_filesystem.py

@@ -0,0 +1,110 @@
+from __future__ import annotations
+
+import json
+from pathlib import Path, PurePosixPath
+
+import pytest
+
+from mega.data_structures import Node
+from mega.errors import MultipleNodesFoundError
+from mega.filesystem import UserFileSystem
+
+NODE_ID = "0fPFklV3"
+FIND_NODE_ID = "0fPFklV3"
+DELETED_NODE_IDS = "t8HkzBH2", "8EwHVJna"
+
+
+@pytest.fixture(name="fs")
+async def filesystem() -> UserFileSystem:
+    return UserFileSystem.from_dump(json.loads(Path("fake_fs.json").read_text()))
+
+
+def test_filesystem_has_root(fs: UserFileSystem) -> None:
+    assert isinstance(fs, UserFileSystem)
+    assert fs.root
+    assert fs.inbox
+    assert fs.trash_bin
+
+
+def test_magic_methods(fs: UserFileSystem) -> None:
+    assert len(fs) == 20
+    assert len(fs) == fs.file_count + fs.folder_count + 3
+    assert isinstance(next(iter(fs)), Node)
+    node = fs[NODE_ID]
+    assert isinstance(node, Node)
+    assert node.id == NODE_ID
+    assert node.id in fs
+    assert node not in fs
+
+
+def test_deleted(fs: UserFileSystem) -> None:
+    deleted = set(fs.deleted)
+    assert deleted == {fs[DELETED_NODE_IDS[0]]}
+    all_deleted = set(fs.iterdir(fs.trash_bin.id, recursive=True))
+    assert all_deleted == {
+        fs[DELETED_NODE_IDS[0]],
+        fs[DELETED_NODE_IDS[1]],
+    }
+
+
+def test_path_resolve(fs: UserFileSystem) -> None:
+    path = "tests/scripts/script.js"
+    rel_path = PurePosixPath(path)
+    node = fs.find(path)
+    assert node is fs.find(rel_path)
+    assert fs.relative_path(node.id) == rel_path
+    assert fs.resolve(node.id) == PurePosixPath("/") / rel_path
+
+
+def test_search(fs: UserFileSystem) -> None:
+    query = "tests/script"
+    results = dict(fs.search(query))
+    expected = (
+        "/tests/script.js",
+        "/tests/scripts",
+        "/tests/scripts/notes.txt",
+        "/tests/scripts/script.js",
+        "/tests/scripts/styles.css",
+    )
+
+    assert sorted(results.values()) == sorted(PurePosixPath(v) for v in expected)
+
+
+def test_find(fs: UserFileSystem) -> None:
+    node = fs.find("/tests/scripts/notes.txt")
+    assert node is fs[FIND_NODE_ID]
+
+    with pytest.raises(FileNotFoundError):
+        fs.find("/this/path/does/not/exists")
+
+    with pytest.raises(FileNotFoundError):
+        fs.find("/tests/script")
+
+    with pytest.raises(MultipleNodesFoundError):
+        fs.find("/tests/logo.png")
+
+
+def test_iter_dir(fs: UserFileSystem) -> None:
+    children = (
+        "/tests/logo.png",
+        "/tests/logo.png",
+        "/tests/script.js",
+        "/tests/scripts",
+        "/tests/setup.sh",
+        "/tests/utils.py",
+    )
+
+    recursive_children = (
+        *children,
+        "/tests/scripts/notes.txt",
+        "/tests/scripts/script.js",
+        "/tests/scripts/styles.css",
+    )
+
+    node = fs.find("/tests")
+
+    def get_path(recursive: bool) -> list[str]:
+        return sorted(str(fs.resolve(n.id)) for n in fs.iterdir(node.id, recursive=recursive))
+
+    assert get_path(recursive=False) == sorted(children)
+    assert get_path(recursive=True) == sorted(recursive_children)