wehub-resource-sync
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
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
|