"""Unit tests for utility functions in utils module."""

import pytest
from src import utils


class TestUtils:
    """Test suite for utility functions in utils module."""

    @pytest.fixture
    def get_treatment_mock(self, mocker):
        """Fixture for get_treatment."""
        splitio_client_mock = mocker.patch('src.utils.config.splitio_client')
        return splitio_client_mock.get_treatment

    def test_is_triggered_send_enabled_for_business_unit_id_on(self, get_treatment_mock):
        """Should return True when get_treatment returns 'on'."""
        get_treatment_mock.return_value = 'on'
        assert utils.is_triggered_send_enabled_for_business_unit_id('12345') is True

    def test_is_triggered_send_enabled_for_business_unit_id_off(self, get_treatment_mock):
        """Should return False when get_treatment returns 'off'."""
        get_treatment_mock.return_value = 'off'
        assert utils.is_triggered_send_enabled_for_business_unit_id('12345') is False