Merge branch 'test/qi-utils' into 'dev'
test: added qi_utils test See merge request ics/sp/2025/n25b/pepperplus-ri!12
This commit was merged in pull request #12.
This commit is contained in:
90
test/unit/test_qi_utils.py
Normal file
90
test/unit/test_qi_utils.py
Normal file
@@ -0,0 +1,90 @@
|
|||||||
|
import sys
|
||||||
|
|
||||||
|
# Import module under test
|
||||||
|
import robot_interface.utils.qi_utils as qi_utils
|
||||||
|
|
||||||
|
|
||||||
|
def reload_qi_utils_with(qi_module):
|
||||||
|
"""
|
||||||
|
Helper: reload qi_utils after injecting a fake qi module.
|
||||||
|
Python 2 uses built-in reload().
|
||||||
|
Just changing sys.modules[qi] won't affect the already imported module.
|
||||||
|
"""
|
||||||
|
if qi_module is None:
|
||||||
|
if "qi" in sys.modules:
|
||||||
|
del sys.modules["qi"]
|
||||||
|
else:
|
||||||
|
sys.modules["qi"] = qi_module
|
||||||
|
|
||||||
|
# Python 2 reload
|
||||||
|
global qi_utils
|
||||||
|
qi_utils = reload(qi_utils)
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_qi_session_no_qi_module():
|
||||||
|
"""
|
||||||
|
Tests the 'qi is None' path.
|
||||||
|
"""
|
||||||
|
reload_qi_utils_with(None)
|
||||||
|
|
||||||
|
session = qi_utils.get_qi_session()
|
||||||
|
assert session is None
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_qi_session_no_qi_url_argument(monkeypatch):
|
||||||
|
"""
|
||||||
|
Tests the '--qi-url not in sys.argv' path.
|
||||||
|
"""
|
||||||
|
class FakeQi:
|
||||||
|
pass
|
||||||
|
|
||||||
|
reload_qi_utils_with(FakeQi())
|
||||||
|
|
||||||
|
monkeypatch.setattr(sys, "argv", ["pytest"])
|
||||||
|
|
||||||
|
session = qi_utils.get_qi_session()
|
||||||
|
assert session is None
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_qi_session_runtime_error(monkeypatch):
|
||||||
|
"""
|
||||||
|
Tests the 'exept RuntineError' path.
|
||||||
|
"""
|
||||||
|
class FakeApp:
|
||||||
|
def start(self):
|
||||||
|
raise RuntimeError("boom")
|
||||||
|
|
||||||
|
class FakeQi:
|
||||||
|
Application = lambda self=None: FakeApp()
|
||||||
|
|
||||||
|
reload_qi_utils_with(FakeQi())
|
||||||
|
|
||||||
|
monkeypatch.setattr(sys, "argv", ["pytest", "--qi-url", "tcp://localhost"])
|
||||||
|
|
||||||
|
session = qi_utils.get_qi_session()
|
||||||
|
assert session is None
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_qi_session_success(monkeypatch):
|
||||||
|
"""
|
||||||
|
Tests a valid path.
|
||||||
|
"""
|
||||||
|
class FakeSession:
|
||||||
|
pass
|
||||||
|
|
||||||
|
class FakeApp:
|
||||||
|
def __init__(self):
|
||||||
|
self.session = FakeSession()
|
||||||
|
|
||||||
|
def start(self):
|
||||||
|
return True
|
||||||
|
|
||||||
|
class FakeQi:
|
||||||
|
Application = lambda self=None: FakeApp()
|
||||||
|
|
||||||
|
reload_qi_utils_with(FakeQi())
|
||||||
|
|
||||||
|
monkeypatch.setattr(sys, "argv", ["pytest", "--qi-url", "tcp://localhost"])
|
||||||
|
|
||||||
|
session = qi_utils.get_qi_session()
|
||||||
|
assert isinstance(session, FakeSession)
|
||||||
Reference in New Issue
Block a user