From f814601fa17e8d9276f3fc88f475bb301521619b Mon Sep 17 00:00:00 2001
From: Robin Huang <robin.j.huang@gmail.com>
Date: Tue, 9 Jul 2024 21:28:13 -0700
Subject: [PATCH] Add check disk space.

---
 .github/workflows/run-on-gpu.yml |  4 ++++
 comfy_cli/test_cmdline.py        | 15 +++++++++++++++
 2 files changed, 19 insertions(+)
 create mode 100644 comfy_cli/test_cmdline.py

diff --git a/.github/workflows/run-on-gpu.yml b/.github/workflows/run-on-gpu.yml
index ec413c7..e132116 100644
--- a/.github/workflows/run-on-gpu.yml
+++ b/.github/workflows/run-on-gpu.yml
@@ -40,6 +40,10 @@ jobs:
         run: |
           pip install -e .
 
+      - name: Check disk space
+        run: |
+          df -h
+
       - name: Run Core Functionality
         run: |
           comfy --skip-prompt --no-enable-telemetry env
diff --git a/comfy_cli/test_cmdline.py b/comfy_cli/test_cmdline.py
new file mode 100644
index 0000000..bd1d476
--- /dev/null
+++ b/comfy_cli/test_cmdline.py
@@ -0,0 +1,15 @@
+from typer.testing import CliRunner
+
+from .cmdline import app
+
+runner = CliRunner()
+
+
+def test_app():
+    result = runner.invoke(
+        app,
+        ["--here", "install", "--cpu"],
+    )
+    print("Stdout:")
+    print(result.stdout)
+    assert result.exit_code == 0