"""Tests for indexing logic."""

from unittest.mock import patch


@patch('src.logic.indexing.get_os_connector_read')
def test_get_all_items_os(mock_os):
    """Test get all items."""
    from src.logic.indexing import get_all_items
    mock_os.return_value.os.count.return_value = {'count': 1}
    mock_os.return_value.os.search.return_value = {'hits': {
        'hits': [{'_source': {'review_queue_id': 1, 'product_id': 1}}]}
    }
    response = get_all_items()
    assert response == [{'reviewQueueId': 1, 'productId': 1}]
    mock_os.return_value.os.count.assert_called_once_with(index='review_read')
    mock_os.return_value.os.search.assert_called_once_with(
        index='review_read',
        _source=True,
        _source_includes=['review_queue_id', 'product_id'],
        size=1
    )