From 544e941b271add2bc4a3f43988f174db9d0b624f Mon Sep 17 00:00:00 2001
From: int3l <int3l@users.noreply.github.com>
Date: Wed, 16 Sep 2020 23:18:38 +0300
Subject: [PATCH] Update the error handling tests with exit codes

---
 tests/error_handler_test.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/error_handler_test.py b/tests/error_handler_test.py
index 833bb8f8..5f57ae3a 100644
--- a/tests/error_handler_test.py
+++ b/tests/error_handler_test.py
@@ -32,6 +32,7 @@ def test_error_handler_fatal_error(mocked_log_and_exit):
         exc,
         # Tested below
         mock.ANY,
+        1,
     )
 
     assert re.match(
@@ -56,6 +57,7 @@ def test_error_handler_uncaught_error(mocked_log_and_exit):
         exc,
         # Tested below
         mock.ANY,
+        3,
     )
     assert re.match(
         r'Traceback \(most recent call last\):\n'
@@ -79,6 +81,7 @@ def test_error_handler_keyboardinterrupt(mocked_log_and_exit):
         exc,
         # Tested below
         mock.ANY,
+        130,
     )
     assert re.match(
         r'Traceback \(most recent call last\):\n'