From e811cafd81da12b759d290293faa69702275a42e Mon Sep 17 00:00:00 2001
From: Tony Narlock <tony@git-pull.com>
Date: Sat, 1 Jun 2019 15:42:24 -0500
Subject: [PATCH] :white_check_mark: Add test for tmuxp freeze

---
 tests/test_cli.py | 49 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/tests/test_cli.py b/tests/test_cli.py
index e54db679e0..e05986c75b 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -472,6 +472,55 @@ def test_import_tmuxinator(cli_args, inputs, tmpdir, monkeypatch):
         assert tmpdir.join('la.yaml').check()
 
 
+@pytest.mark.parametrize(
+    "cli_args,inputs",
+    [
+        (['freeze', 'mysession'], ['\n', 'y\n', './la.yaml\n', './la.yaml\n', 'y\n']),
+        (  # Exists
+            ['freeze', 'mysession'],
+            [
+                '\n',
+                'y\n',
+                './exists.yaml\n',
+                './exists.yaml\n',
+                './la.yaml\n',
+                './la.yaml\n',
+                'y\n',
+            ],
+        ),
+        (  # Imply current session if not entered
+            ['freeze'],
+            ['\n', 'y\n', './la.yaml\n', './la.yaml\n', 'y\n'],
+        ),
+        (
+            ['freeze'],
+            [
+                '\n',
+                'y\n',
+                './exists.yaml\n',
+                './exists.yaml\n',
+                './la.yaml\n',
+                './la.yaml\n',
+                'y\n',
+            ],
+        ),  # Exists
+    ],
+)
+def test_freeze(server, cli_args, inputs, tmpdir, monkeypatch):
+    monkeypatch.setenv('HOME', str(tmpdir))
+    tmpdir.join('exists.yaml').ensure()
+
+    server.new_session(session_name='mysession')
+
+    with tmpdir.as_cwd():
+        runner = CliRunner()
+        # Use tmux server (socket name) used in the test
+        cli_args = cli_args + ['-L', server.socket_name]
+        out = runner.invoke(cli.cli, cli_args, input=''.join(inputs))
+        print(out.output)
+        assert tmpdir.join('la.yaml').check()
+
+
 def test_get_abs_path(tmpdir):
     expect = str(tmpdir)
     with tmpdir.as_cwd():