Nekmo/pip-rating

import unittest
from unittest.mock import MagicMock, patch

from pip_rating.exceptions import (
    RequirementsRatingError,
    RequirementsRatingMissingReqFile,
    catch,
)


class TestRequirementsRatingError(unittest.TestCase):
    """Tests for the RequirementsRatingError exception."""

    def test_init(self):
        """Test the __init__ method of RequirementsRatingError."""
        with self.subTest("Test with no arguments"):
            exception = RequirementsRatingError()
            self.assertEqual(exception.extra_body, "")
        with self.subTest("Test with extra_body argument"):
            exception = RequirementsRatingError("foo")
            self.assertEqual(exception.extra_body, "foo")

    def test_str(self):
        """Test the __str__ method of RequirementsRatingError."""
        with self.subTest("Test with no arguments"):
            exception = RequirementsRatingError()
            self.assertEqual(str(exception), "RequirementsRatingError")
        with self.subTest("Test with body argument"):
            exception = RequirementsRatingError()
            exception.body = "bar"
            self.assertEqual(str(exception), "RequirementsRatingError: bar")
        with self.subTest("Test with extra_body argument"):
            exception = RequirementsRatingError(extra_body="foo")
            self.assertEqual(str(exception), "RequirementsRatingError: foo")
        with self.subTest("Test with body and extra_body arguments"):
            exception = RequirementsRatingError("foo")
            exception.body = "bar"
            self.assertEqual(str(exception), "RequirementsRatingError: bar. foo")


class TestRequirementsRatingMissingReqFile(unittest.TestCase):
    """Tests for the RequirementsRatingMissingReqFile exception."""

    def test_init(self):
        """Test the __init__ method of RequirementsRatingMissingReqFile."""
        directory = "directory"
        exception = RequirementsRatingMissingReqFile(directory)
        self.assertEqual(exception.directory, directory)
        self.assertEqual(
            f"Missing requirements file in {directory}", exception.extra_body
        )


class TestCatch(unittest.TestCase):
    """Tests for the catch decorator."""

    @patch("pip_rating.exceptions.Console")
    def test_catch(self, mock_console: MagicMock):
        """Test the catch decorator."""

        @catch
        def requirements_rating_error():
            raise RequirementsRatingError("foo")

        @catch
        def requirements_rating_missing_req_file():
            raise RequirementsRatingMissingReqFile("foo")

        with self.subTest("Test with RequirementsRatingError"):
            with self.assertRaises(SystemExit) as context:
                requirements_rating_error()
            self.assertEqual(RequirementsRatingError.exit_code, context.exception.code)
            mock_console.return_value.print.assert_called()

        mock_console.reset_mock()
        with self.subTest("Test with RequirementsRatingMissingReqFile"):
            with self.assertRaises(SystemExit) as context:
                requirements_rating_missing_req_file()
            self.assertEqual(context.exception.code, 13)
            mock_console.return_value.print.assert_called()