From 094a80b50a188c7a75b7fab621d05825d313358d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Dec 2023 14:46:37 -0300 Subject: [PATCH] Fix test_superuser assertion --- tests/test_cli.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index 63122d254..ee95a271c 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -1,8 +1,9 @@ from pathlib import Path from tempfile import tempdir -from langflow.__main__ import app + import pytest +from langflow.__main__ import app from langflow.services import deps @@ -33,4 +34,5 @@ def test_components_path(runner, client, default_settings): 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 + assert "Superuser creation failed." not in result.output, result.output + assert "Superuser created successfully." in result.output, result.output