add tests for if conditions

Signed-off-by: Tin Lai <[email protected]>

Author: soraxas

examples/if-conditions-test.yaml

@@ -0,0 +1,48 @@
+session_name: if conditions test conditions
+environment:
+  foo: 'false'
+  bar: '1'
+  F: '0'
+windows:
+  - window_name: window all false
+    panes:
+      - if: ${foo}
+        shell_command:
+          - echo pane 1
+      - if:
+          shell: '[ 1 -gt 2 ]'
+        shell_command:
+          - echo pane 2
+      - if:
+          shell_var: ${F}
+  - window_name: window 2 of 3 true
+    panes:
+      - if:
+          shell: '[ "foo" = "bar" ]'
+        shell_command:
+          - echo pane 3
+      - if:
+          shell: '[ "hello" != "byte" ]'
+        shell_command:
+          - echo pane 4
+      - if:
+          python: '2**4 == 16'
+        shell_command:
+          - echo pane 5
+  - window_name: window 2 of 4 true
+    panes:
+      - if:
+          shell_var: 'FALSE'
+        shell_command:
+          - echo pane 6
+      - if:
+          shell_var: ${bar}
+        shell_command:
+          - echo pane 7
+      - if:
+          python: import os; not os.path.isdir('/a/very/random/path')
+        shell_command:
+          - echo pane 8
+      - if: ${non_existing_var}
+        shell_command:
+          - echo pane 9

examples/if-conditions.yaml

@@ -19,7 +19,7 @@ windows:
           - echo the above is a false statement
       # python condition
       - if:
-          python: import os; os.path.isdir('${PWD}')
+          python: import os; os.path.isdir(os.path.expandvars('${PWD}'))
         shell_command:
           - echo "checking for PWD (${PWD}) is a directory in python"
           - python -m http.server

tests/workspace/test_builder.py

@@ -563,6 +563,34 @@ def test_blank_pane_spawn(
     assert len(window4.panes) == 2
 
 
+def test_if_conditions(
+    session: Session,
+) -> None:
+    """
+    Test various ways of spawning panes with conditions from a tmuxp configuration.
+    """
+    yaml_workspace_file = EXAMPLE_PATH / "if-conditions-test.yaml"
+    test_config = ConfigReader._from_file(yaml_workspace_file)
+
+    test_config = loader.expand(test_config)
+    builder = WorkspaceBuilder(session_config=test_config, server=session.server)
+    builder.build(session=session)
+
+    assert session == builder.session
+
+    with pytest.raises(ObjectDoesNotExist):
+        window1 = session.windows.get(window_name="window all false")
+        assert window1 is None
+
+    window2 = session.windows.get(window_name="window 2 of 3 true")
+    assert window2 is not None
+    assert len(window2.panes) == 2
+
+    window3 = session.windows.get(window_name="window 2 of 4 true")
+    assert window3 is not None
+    assert len(window3.panes) == 2
+
+
 def test_start_directory(session: Session, tmp_path: pathlib.Path) -> None:
     """Test workspace builder setting start_directory relative to current directory."""
     test_dir = tmp_path / "foo bar"