"""Test tax report logic."""

from unittest.mock import patch

from royalties.tax_report import execute_tax_report_query


@patch('royalties.tax_report.SnowflakeSQLExecutor')
def test_tax_report_execute(mock_sf):
    """Test executing the tax report logic."""
    execute_tax_report_query(None, '1', 'all', 'month', 'bar')
    mock_sf.return_value.__enter__.return_value.fetchall.assert_called_once()
    execute_tax_report_query(None, '1', 'domestic', 'month', 'bar')
    execute_tax_report_query(None, '1', 'foreign', 'month', 'bar')
    assert mock_sf.return_value.__enter__.return_value.fetchall.call_count == 3