test(Session.new_window): Test with direction

Author: tony

tests/test_session.py

@@ -330,6 +330,37 @@ def test_new_window_with_environment_logs_warning_for_old_tmux(
     ), "Warning missing"
 
 
+@pytest.mark.skipif(
+    has_lt_version("3.2"),
+    reason="Only 3.2+ has the -a and -b flag on new-window",
+)
+def test_session_new_window_with_direction(
+    session: Session,
+) -> None:
+    """Verify new window with direction."""
+    window = session.active_window
+    window.refresh()
+
+    window_initial = session.new_window(window_name="Example")
+    assert window_initial.window_index == "2"
+
+    window_before = session.new_window(
+        window_name="Window before", direction=NewWindowDirection.Before
+    )
+    window_initial.refresh()
+    assert window_before.window_index == "1"
+    assert window_initial.window_index == "3"
+
+    window_after = session.new_window(
+        window_name="Window after", direction=NewWindowDirection.After
+    )
+    window_initial.refresh()
+    window_after.refresh()
+    assert window_after.window_index == "3"
+    assert window_initial.window_index == "4"
+    assert window_before.window_index == "1"
+
+
 @pytest.mark.skipif(
     has_gte_version("3.1"),
     reason="Only 3.1 has the -a and -b flag on new-window",