add test for workspaces

Author: Tony Crisci

test/aio/test_window.py

@@ -8,7 +8,7 @@ class TestWindow(IpcTest):
     event = None
 
     def on_window(self, i3, e):
-        self.event = e
+        TestWindow.event = e
         i3.main_quit()
 
     @pytest.mark.asyncio
@@ -21,7 +21,7 @@ async def test_window_event(self, i3):
 
         await i3.main()
 
-        assert self.event
+        assert TestWindow.event
 
     @pytest.mark.asyncio
     async def test_marks(self, i3):

test/aio/test_workspace.py

@@ -0,0 +1,28 @@
+from .ipctest import IpcTest
+
+import pytest
+from i3ipc import Event
+
+
+class TestWorkspace(IpcTest):
+    event = None
+
+    def on_workspace(self, i3, e):
+        TestWorkspace.event = e
+        i3.main_quit()
+
+    @pytest.mark.asyncio
+    async def test_workspace(self, i3):
+        await i3.subscribe([Event.WORKSPACE])
+        i3.on(Event.WORKSPACE_INIT, self.on_workspace)
+        await i3.command('workspace 12')
+        await i3.main()
+        workspaces = await i3.get_workspaces()
+
+        assert len(workspaces) == 1
+        ws = workspaces[0]
+        assert ws.name == '12'
+
+        e = TestWorkspace.event
+        assert e is not None
+        assert e.current.name == '12'