test window size script

pyproject.toml

@@ -44,7 +44,6 @@ omit = [
     "src/magic_tiler/utils/filestore.py",
     # top-level scripts
     "src/magic_tiler/magic_tiler.py",
-    "src/magic_tiler/get_window_sizes.py",
 ]
 
 [tool.coverage.report]

src/magic_tiler/get_window_sizes.py

@@ -2,10 +2,15 @@
 
 import click
 
+from magic_tiler.utils import interfaces
 from magic_tiler.utils import sway
 
 
 @click.command()
 def main() -> None:
     window_manager = sway.Sway()
+    print_window_sizes(window_manager)
+
+
+def print_window_sizes(window_manager: interfaces.TilingWindowManager) -> None:
     pprint.pprint(window_manager.get_window_sizes())

src/magic_tiler/utils/interfaces.py

@@ -40,6 +40,10 @@ def num_workspace_windows(self) -> int:
     def get_tree(self) -> List:
         pass
 
+    @abc.abstractmethod
+    def get_window_sizes(self) -> Dict:
+        pass
+
 
 class ConfigReader(object):
     @abc.abstractmethod

src/magic_tiler/utils/sway.py

@@ -15,7 +15,7 @@
 SLEEP_TIME = 0.30
 
 
-class Sway(interfaces.TilingWindowManager):  # pragma: nocover
+class Sway(interfaces.TilingWindowManager):
     def __init__(self) -> None:
         self._sway = i3ipc.Connection()
 

tests/fakes.py

@@ -54,9 +54,13 @@ class FakeNode(NamedTuple):
 
 
 class FakeWindowManager(interfaces.TilingWindowManager):
-    def __init__(self, tree: Optional[List[FakeNode]] = None):
+    def __init__(
+        self, tree: Optional[List[FakeNode]] = None, window_sizes: Optional[Dict] = None
+    ):
         if tree:
             self._tree = tree
+        if window_sizes:
+            self._window_sizes = window_sizes
 
     def make_window(self, window_details: dtos.WindowDetails) -> None:
         pass
@@ -84,3 +88,6 @@ def num_workspace_windows(self) -> int:
 
     def get_tree(self) -> List[FakeNode]:
         return self._tree
+
+    def get_window_sizes(self):
+        return self._window_sizes

tests/test_get_window_sizes.py

@@ -0,0 +1,37 @@
+from unittest import mock
+
+import pytest
+
+from magic_tiler import get_window_sizes
+
+
+@pytest.fixture
+def mock_window_manager(mocker):
+    return mocker.patch("magic_tiler.utils.sway.Sway")
+
+
+@pytest.fixture
+def mock_print_window_sizes(mocker):
+    return mocker.patch("magic_tiler.get_window_sizes.print_window_sizes")
+
+
+@pytest.mark.e2e
+def test_get_window_sizes(click_runner):
+    result = click_runner.invoke(get_window_sizes.main)
+    assert result.exit_code == 0
+
+
+def test_click_handles_options(
+    click_runner, mock_window_manager, mock_print_window_sizes
+):
+    """Click should handle the user's input and specified flags to pass in the right
+    details to the print_window_sizes function
+    """
+    click_runner.invoke(get_window_sizes.main)
+    mock_print_window_sizes.assert_called_once_with(mock_window_manager())
+
+
+def test_calls_window_manager():
+    window_manager = mock.MagicMock()
+    get_window_sizes.print_window_sizes(window_manager)
+    window_manager.get_window_sizes.assert_called_once_with()

tests/test_layout.py

@@ -55,6 +55,9 @@ def split(self, split_type: str) -> None:
     def get_tree(self):
         pass
 
+    def get_window_sizes(self) -> Dict:
+        pass
+
 
 class LayoutTestCase(NamedTuple):
     config: Dict