From e419d227c274b47e02fc7c1bf54e908df75821fd Mon Sep 17 00:00:00 2001 From: Achintya Jai <153343775+pUrGe12@users.noreply.github.com> Date: Thu, 12 Jun 2025 05:50:17 +0530 Subject: [PATCH] Add die.py tests (#1042) * created tests for die.py * updated * migrate to pytest * Update deps * Revert poetry.lock --------- Signed-off-by: Achintya Jai <153343775+pUrGe12@users.noreply.github.com> Co-authored-by: Arkadii Yakovets <2201626+arkid15r@users.noreply.github.com> Co-authored-by: Arkadii Yakovets --- tests/core/test_die.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 tests/core/test_die.py diff --git a/tests/core/test_die.py b/tests/core/test_die.py new file mode 100644 index 00000000..3caa1ba2 --- /dev/null +++ b/tests/core/test_die.py @@ -0,0 +1,25 @@ +from io import StringIO +from unittest.mock import patch + +from nettacker.core.die import die_success, die_failure +from nettacker.logger import TerminalCodes + + +@patch("sys.stdout", new_callable=StringIO) +@patch("sys.exit") +def test_die_success(mock_exit, mock_stdout): + reset_code = TerminalCodes.RESET.value + die_success() + success_message = mock_stdout.getvalue() + assert reset_code in success_message + mock_exit.assert_called_once_with(0) + + +@patch("sys.stdout", new_callable=StringIO) +@patch("sys.exit") +def test_die_failure(mock_exit, mock_stdout): + test_message = "Test error message" + die_failure(test_message) + error_message = mock_stdout.getvalue() + assert test_message in error_message + mock_exit.assert_called_once_with(1)