from unittest import TestCase, mock

from xsdata.utils.hooks import load_entry_points


class HooksTests(TestCase):
    @mock.patch("xsdata.utils.hooks.metadata.entry_points")
    def test_load_entry_points_with_mapping_api(self, mock_entry_points) -> None:
        first_ep = mock.Mock()
        second_ep = mock.Mock()
        third_ep = mock.Mock()

        mock_entry_points.return_value = {
            "foo": [first_ep, second_ep],
            "bar": [third_ep],
        }

        load_entry_points("foo")
        first_ep.load.assert_called_once_with()
        second_ep.load.assert_called_once_with()
        self.assertEqual(0, third_ep.load.call_count)

    @mock.patch("xsdata.utils.hooks.metadata.entry_points")
    def test_load_entry_points_with_select_api(self, mock_entry_points) -> None:
        first_ep = mock.Mock()
        second_ep = mock.Mock()

        entry_points = mock.Mock()
        entry_points.select.return_value = [first_ep, second_ep]
        mock_entry_points.return_value = entry_points

        load_entry_points("foo")
        first_ep.load.assert_called_once_with()
        second_ep.load.assert_called_once_with()
        entry_points.select.assert_called_once_with(group="foo")