1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from .ipctest import IpcTest
import pytest
import asyncio
class TestShutdownEvent(IpcTest):
events = []
def restart_func(self, i3):
asyncio.ensure_future(i3.command('restart'))
def on_shutdown(self, i3, e):
self.events.append(e)
if len(self.events) == 1:
i3._loop.call_later(0.1, self.restart_func, i3)
elif len(self.events) == 2:
i3.main_quit()
@pytest.mark.asyncio
async def test_shutdown_event_reconnect(self, i3):
i3._auto_reconnect = True
self.events = []
i3.on('shutdown::restart', self.on_shutdown)
i3._loop.call_later(0.1, self.restart_func, i3)
await i3.main()
assert len(self.events) == 2
|
