test
This commit is contained in:
parent
40acb6e4e3
commit
4b6f7433c7
1 changed files with 8 additions and 1 deletions
|
|
@ -8,6 +8,9 @@ import sys
|
||||||
import subprocess
|
import subprocess
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import patch, MagicMock
|
||||||
|
from cognee.shared.logging_utils import get_logger
|
||||||
|
|
||||||
|
logger = get_logger()
|
||||||
|
|
||||||
|
|
||||||
class TestCliIntegration:
|
class TestCliIntegration:
|
||||||
|
|
@ -91,7 +94,11 @@ class TestCliIntegration:
|
||||||
"loggingworker cancelled" in stderr_lower or "cancellederror" in stderr_lower
|
"loggingworker cancelled" in stderr_lower or "cancellederror" in stderr_lower
|
||||||
)
|
)
|
||||||
|
|
||||||
assert not has_error or has_expected_failure or has_litellm_cancellation
|
try:
|
||||||
|
assert not has_error or has_expected_failure or has_litellm_cancellation
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"{str(e)}", exc_info=True)
|
||||||
|
raise e
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
os.unlink(temp_file)
|
os.unlink(temp_file)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue