ouyangbo/EvoScientist
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 3 Wiki Activity

feat(tests): mock import for telegram and discord in TestStepChannels

Browse Source
This commit is contained in:
X-iZhang
2026-02-16 13:52:01 +00:00
parent 283ae49a91
commit 1446267360
1 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_onboard.py
+18 -2
View File
@@ -617,8 +617,16 @@ class TestStepChannels:
config = EvoScientistConfig()
_real_import = __builtins__.__import__ if hasattr(__builtins__, '__import__') else __import__
def _fake_import(name, *args, **kwargs):
if name == "telegram":
return # pretend installed
return _real_import(name, *args, **kwargs)
with patch("EvoScientist.config.onboard.questionary") as mock_q, \
patch("EvoScientist.config.onboard._probe_channel"):
patch("EvoScientist.config.onboard._probe_channel"), \
patch("builtins.__import__", side_effect=_fake_import):
mock_q.checkbox.return_value.ask.return_value = ["telegram"]
mock_q.text.return_value.ask.return_value = "test-token"
result = _step_channels(config)
@@ -632,8 +640,16 @@ class TestStepChannels:
config = EvoScientistConfig()
_real_import = __builtins__.__import__ if hasattr(__builtins__, '__import__') else __import__
def _fake_import(name, *args, **kwargs):
if name == "discord":
return # pretend installed
return _real_import(name, *args, **kwargs)
with patch("EvoScientist.config.onboard.questionary") as mock_q, \
patch("EvoScientist.config.onboard._probe_channel"):
patch("EvoScientist.config.onboard._probe_channel"), \
patch("builtins.__import__", side_effect=_fake_import):
mock_q.checkbox.return_value.ask.return_value = ["discord"]
mock_q.text.return_value.ask.return_value = "discord-token"
result = _step_channels(config)