Hua/nanobot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: save DingTalk downloaded files to media dir instead of /tmp

Browse Source
This commit is contained in:
Meng Yuhang
2026-03-14 23:24:36 +08:00
committed by Xubin Ren
parent 34358eabc9
commit f9ba6197de
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_dingtalk_channel.py
View File

@@ -194,14 +194,17 @@ async def test_download_dingtalk_file(tmp_path, monkeypatch) -> None:
])
channel._http._responses[1].content = file_content
# Redirect temp dir to tmp_path
monkeypatch.setattr("tempfile.gettempdir", lambda: str(tmp_path))
# Redirect media dir to tmp_path
monkeypatch.setattr(
"nanobot.config.paths.get_media_dir",
lambda channel_name=None: tmp_path / channel_name if channel_name else tmp_path,
)
result = await channel._download_dingtalk_file("code123", "test.xlsx", "user1")
assert result is not None
assert result.endswith("test.xlsx")
assert (tmp_path / "nanobot_dingtalk" / "user1" / "test.xlsx").read_bytes() == file_content
assert (tmp_path / "dingtalk" / "user1" / "test.xlsx").read_bytes() == file_content
# Verify API calls
assert channel._http.calls[0]["method"] == "POST"