From 137954de6b59428264de461500a8b9076bfa68de Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 28 Aug 2023 15:47:57 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(=5F=5Fmain=5F=5F.py):=20rena?=
 =?UTF-8?q?me=20serve=20command=20to=20run=20command=20for=20better=20sema?=
 =?UTF-8?q?ntics=20and=20clarity=20=F0=9F=94=A7=20chore(test=5Fcli.py):=20?=
 =?UTF-8?q?update=20test=20to=20use=20the=20new=20run=20command=20instead?=
 =?UTF-8?q?=20of=20serve=20command=20=E2=9C=A8=20feat(test=5Fcli.py):=20ad?=
 =?UTF-8?q?d=20test=20for=20creating=20a=20superuser=20with=20the=20superu?=
 =?UTF-8?q?ser=20command?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/__main__.py | 11 ++++++++++-
 tests/test_cli.py                |  8 +++++++-
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py
index b362d2fa9..c2b7b0fd6 100644
--- a/src/backend/langflow/__main__.py
+++ b/src/backend/langflow/__main__.py
@@ -95,7 +95,7 @@ def serve_on_jcloud():
 
 
 @app.command()
-def serve(
+def run(
     host: str = typer.Option(
         "127.0.0.1", help="Host to bind the server to.", envvar="LANGFLOW_HOST"
     ),
@@ -326,7 +326,16 @@ def superuser(
         from langflow.services.auth.utils import create_super_user
 
         if create_super_user(session, username, password):
+            # Verify that the superuser was created
+            from langflow.services.database.models.user.user import User
+
+            user = session.query(User).filter(User.username == username).first()
+            if user is None:
+                typer.echo("Superuser creation failed.")
+                return
+
             typer.echo("Superuser created successfully.")
+
         else:
             typer.echo("Superuser creation failed.")
 
diff --git a/tests/test_cli.py b/tests/test_cli.py
index c5fba1da6..4ed00893e 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -23,8 +23,14 @@ def test_components_path(runner, client, default_settings):
 
     result = runner.invoke(
         app,
-        ["serve", "--components-path", str(temp_dir), *default_settings],
+        ["run", "--components-path", str(temp_dir), *default_settings],
     )
     assert result.exit_code == 0, result.stdout
     settings_manager = utils.get_settings_manager()
     assert str(temp_dir) in settings_manager.settings.COMPONENTS_PATH
+
+
+def test_superuser(runner, client, session):
+    result = runner.invoke(app, ["superuser"], input="admin\nadmin\n")
+    assert result.exit_code == 0, result.stdout
+    assert "Superuser created successfully." in result.stdout