"""Custom export flow test cases."""

import pytest

from processing_accounting.flows.custom_export.flow import Flow


@pytest.mark.parametrize(
    'report_version, expected_report_version', [
        (None, 1),
        (1, 1),
        (2, 2)])
def test_flow_workflow_id(report_version, expected_report_version):
    """Test custom export workflow_id generation."""
    flow = Flow()
    initial_context = {
        'user_id': '123',
        'user_type': 'label',
        'period_ids': '217',
        'transaction_types': 'all',
        'locale': 'en_US',
        'file_format': 'xls',
        'report_version': report_version
    }
    expected_workflow_id = (
        'custom_export_processing-accounting-123-label-217-all-en_US-xls-'
        '{}'.format(expected_report_version))
    workflow_id = flow.workflow_id(initial_context)
    assert workflow_id == expected_workflow_id