chore: import upstream snapshot with attribution

2026-07-13 12:47:58 +08:00
commit b16403ea71
789 changed files with 115226 additions and 0 deletions
@@ -0,0 +1,89 @@
+import datetime
+
+from e2b.envd.filesystem import filesystem_pb2
+from e2b.sandbox.filesystem.filesystem import (
+    FileType,
+    WriteInfo,
+    map_entry_info,
+    map_file_type_str,
+)
+from e2b.volume.client.models import VolumeEntryStat as VolumeEntryStatApi
+from e2b.volume.client.models import VolumeEntryStatType
+from e2b.volume.utils import convert_volume_entry_stat
+
+
+def test_write_info_from_dict_converts_type_to_enum():
+    info = WriteInfo.from_dict(
+        {"name": "a.txt", "type": "file", "path": "/home/user/a.txt"}
+    )
+    assert info.type is FileType.FILE
+
+    info = WriteInfo.from_dict({"name": "dir", "type": "dir", "path": "/home/user/dir"})
+    assert info.type is FileType.DIR
+
+
+def test_write_info_from_dict_handles_missing_or_unknown_type():
+    info = WriteInfo.from_dict({"name": "a.txt", "path": "/home/user/a.txt"})
+    assert info.type is None
+
+    info = WriteInfo.from_dict(
+        {"name": "a.txt", "type": "symlink", "path": "/home/user/a.txt"}
+    )
+    assert info.type is None
+
+
+def test_map_file_type_str():
+    assert map_file_type_str("file") is FileType.FILE
+    assert map_file_type_str("dir") is FileType.DIR
+    assert map_file_type_str("unknown") is None
+    assert map_file_type_str(None) is None
+
+
+def test_map_entry_info_modified_time_is_timezone_aware():
+    entry = filesystem_pb2.EntryInfo(
+        name="a.txt",
+        type=filesystem_pb2.FileType.FILE_TYPE_FILE,
+        path="/home/user/a.txt",
+        size=4,
+        mode=0o644,
+        permissions="-rw-r--r--",
+        owner="user",
+        group="user",
+    )
+    entry.modified_time.FromDatetime(
+        datetime.datetime(2026, 1, 2, 3, 4, 5, tzinfo=datetime.timezone.utc)
+    )
+
+    info = map_entry_info(entry)
+
+    assert info.modified_time.tzinfo == datetime.timezone.utc
+    assert info.modified_time == datetime.datetime(
+        2026, 1, 2, 3, 4, 5, tzinfo=datetime.timezone.utc
+    )
+
+
+def test_convert_volume_entry_stat_normalizes_naive_times_to_utc():
+    naive = datetime.datetime(2026, 1, 2, 3, 4, 5)
+    aware = datetime.datetime(2026, 1, 2, 3, 4, 5, tzinfo=datetime.timezone.utc)
+
+    api_stat = VolumeEntryStatApi(
+        name="a.txt",
+        type_=VolumeEntryStatType.FILE,
+        path="/a.txt",
+        size=4,
+        mode=0o644,
+        uid=1000,
+        gid=1000,
+        atime=naive,
+        mtime=naive,
+        ctime=aware,
+    )
+
+    stat = convert_volume_entry_stat(api_stat)
+
+    assert stat.atime == aware
+    assert stat.mtime == aware
+    assert stat.ctime == aware
+    assert stat.atime.tzinfo == datetime.timezone.utc
+    assert stat.mtime.tzinfo == datetime.timezone.utc
+    assert stat.ctime.tzinfo == datetime.timezone.utc