Skip to content

Commit fbd3f5c

Browse files
committed
✅ test_unregister_all(), refs #45
1 parent 8e8d54d commit fbd3f5c

File tree

1 file changed

+23
-1
lines changed

1 file changed

+23
-1
lines changed

tests/test_magic.py

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,37 @@
1+
from unittest import mock
2+
13
from pycaw.magic import MagicApp, MagicManager, MagicSession
24

35

6+
def patch_atexit_register():
7+
"""Prevent MagicManager.clean_up() call as it seems to misbehave with tests."""
8+
return mock.patch("atexit.register")
9+
10+
411
class TestMagic:
512
def test_init(self):
613
app_execs = {"msedge.exe"}
7-
magic = MagicApp(app_execs)
14+
# prevent MagicManager.clean_up() call as it seems to misbehave with tests
15+
with patch_atexit_register() as m_register:
16+
magic = MagicApp(app_execs)
817
assert magic.app_execs == app_execs
18+
assert m_register.called
919

1020

1121
class TestMagicManager:
1222
def test_magic_session(self):
1323
assert MagicManager.MagicSessionConfigured is None
1424
MagicManager.magic_session(MagicSession)
1525
assert MagicManager.MagicSessionConfigured == (MagicSession, (), {})
26+
27+
def test_unregister_all(self):
28+
assert MagicManager.magic_apps is not None
29+
assert (
30+
MagicManager.str()
31+
== "<MagicManager magic_apps='1' magic_sessions='1' active_mrs='1' trash_mrs='0'/>"
32+
)
33+
MagicManager.unregister_all()
34+
assert MagicManager.str() == "unactive MagicManager"
35+
assert hasattr(MagicManager, "magic_apps") is False
36+
assert hasattr(MagicManager, "magic_sessions") is False
37+
assert MagicManager.magic_activated is None

0 commit comments

Comments
 (0)