Changes in file multicast/exceptions.py:

 - fix for PYL-W1113 in class ShutdownCommandReceived

Changes in file tests/test_exceptions.py:
 - added additional test cases for related changes

Author: reactive-firewall

multicast/exceptions.py

@@ -9,7 +9,7 @@
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 # ..........................................
-# http://www.github.com/reactive-firewall/python-repo/LICENSE.md
+# https://www.github.com/reactive-firewall/multicast/LICENSE.md
 # ..........................................
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
@@ -284,7 +284,7 @@ class ShutdownCommandReceived(RuntimeError):
 
 	__name__ = """multicast.exceptions.ShutdownCommandReceived"""
 
-	def __init__(self, message="SHUTDOWN", *args, **kwargs):
+	def __init__(self, *args, **kwargs):
 		"""
 		Initialize the ShutdownCommandReceived exception.
 
@@ -345,10 +345,37 @@ def __init__(self, message="SHUTDOWN", *args, **kwargs):
 				... except ShutdownCommandReceived as e:
 				...     str(e) == "Custom message"
 				True
+
+			Testcase 7: Initialization with different call to init:
+				A. - Initializes a ShutdownCommandReceived with a specific exit code.
+				B. - Checks the message is still set, as normally would.
+				C. - Checks the cause is set, as super class would.
+				D. - Checks the specific exit code is NOT sixty-five (65) but rather still 143.
+
+				>>> pre_made_args = [ValueError("inner Cause"), "A Pre-made SHUTDOWN", int(65)]
+				>>> error = ShutdownCommandReceived(*pre_made_args)
+				>>> error.message
+				'A Pre-made SHUTDOWN'
+				>>> error.__cause__  #doctest: +ELLIPSIS
+				ValueError...inner Cause...
+				>>> error.exit_code
+				143
 		"""
+		if len(args) > 0 and isinstance(args[0], BaseException):
+			cause = args[0]
+			args = args[1:]
+		else:
+			cause = kwargs.pop("__cause__", None)
+		if len(args) > 0 and isinstance(args[0], str):
+			message = args[0]
+			args = args[1:]
+		else:
+			message = kwargs.pop("message", "SHUTDOWN")
 		if not isinstance(message, str):
 			raise TypeError("[CWE-573] message must be a string")
 		super().__init__(message, *args, **kwargs)
+		if cause is not None:
+			self.__cause__ = cause
 		self.message = message
 		self.exit_code = 143  # Use SIGTERM exit code for graceful shutdown
 

tests/test_exceptions.py

@@ -1,15 +1,15 @@
-#! /usr/bin/env python
+#! /usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-# Python Test Repo Template
+# Python Multicast Repo
 # ..................................
 # Copyright (c) 2017-2024, Mr. Walls
 # ..................................
 # Licensed under MIT (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 # ..........................................
-# http://www.github.com/reactive-firewall/python-repo/LICENSE.md
+# https://www.github.com/reactive-firewall/multicast/LICENSE.md
 # ..........................................
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
@@ -20,16 +20,26 @@
 __module__ = """tests"""
 
 try:
+	"""Handle imports with CWE-758 mitigation.
+
+	This implementation uses a nested try-except pattern to:
+	1. Attempt direct context import
+	2. Fallback to relative import
+	3. Validate context module integrity
+	4. Import required dependencies
+
+	References:
+	- CWE-758: Reliance on Undefined, Unspecified, or Implementation-Defined Behavior
+	"""
 	try:
 		import context
-	except Exception as ImportErr:  # pragma: no branch
-		ImportErr = None
-		del ImportErr  # skipcq - cleanup any error leaks early
+	except ImportError as _:  # pragma: no branch
+		del _  # skipcq - cleanup any error vars early
 		from . import context
 	if context.__name__ is None:
-		raise ImportError("[CWE-758] Failed to import context") from None
+		raise ModuleNotFoundError("[CWE-758] Failed to import context") from None
 	else:
-		from context import multicast
+		from context import multicast  # pylint: disable=cyclic-import - skipcq: PYL-R0401
 		from context import unittest
 		from context import BasicUsageTestSuite
 except Exception as _cause:  # pragma: no branch
@@ -82,6 +92,52 @@ def test_command_execution_error_with_cause(self):
 		self.assertEqual(error.message, "Test with cause")
 		self.assertEqual(error.exit_code, 77)
 
+	def test_shutdown_received_error_with_args(self):
+		"""
+		Test ShutdownCommandReceived initialization with custom message and exit code.
+
+		Verifies that both the message and exit code are correctly assigned when
+		explicitly provided during initialization.
+		"""
+		error = multicast.exceptions.ShutdownCommandReceived("Test Shutdown", 42)
+		self.assertEqual(error.message, "Test Shutdown")
+		self.assertIsNotEqual(error.exit_code, 42, "Unexpectedly was able to overide exit code!")
+		self.assertEqual(error.exit_code, 143)
+
+	def test_shutdown_received_error_default_exit_code(self):
+		"""Test ShutdownCommandReceived initialization with default exit code.
+
+		Verifies that the exit code defaults to 143 when only a message is provided.
+		"""
+		error = multicast.exceptions.ShutdownCommandReceived("Test Shutdown")
+		self.assertEqual(error.exit_code, 143)
+
+	def test_shutdown_received_error_GIVEN_no_args(self):
+		"""Test ShutdownCommandReceived initialization with default exit code.
+
+		Verifies that the exit code defaults to 143 when no arguments are provided.
+		Verifies that the message defaults to 'SHUTDOWN' when no arguments are provided.
+		"""
+		error = multicast.exceptions.ShutdownCommandReceived()
+		self.assertEqual(error.message, "SHUTDOWN")
+		self.assertEqual(error.exit_code, 143)
+
+	def test_shutdown_received_error_with_cause(self):
+		"""Test ShutdownCommandReceived initialization with a cause.
+
+		Verifies that the error properly chains exceptions when initialized with a
+		cause, maintaining both the cause reference and custom attributes.
+		"""
+		test_cause = RuntimeError("test")
+		self.assertIsNotNone(test_cause)
+		error = multicast.exceptions.ShutdownCommandReceived(test_cause, "Shutdown with cause", 77)
+		self.assertIsNotNone(error)
+		self.assertIsNotNone(error.__cause__)
+		self.assertEqual(error.__cause__, test_cause)
+		self.assertEqual(error.message, "Shutdown with cause")
+		self.assertIsNotEqual(error.exit_code, 77, "Unexpectedly was able to overide exit code!")
+		self.assertEqual(error.exit_code, 143)
+
 
 # leave this part
 if __name__ == '__main__':