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 <arkadii.yakovets@owasp.org>

Author: pUrGe12

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)