"""Unit tests for CLI functions to run decider, worker and exec processes."""

from unittest.mock import MagicMock
from unittest.mock import patch

from accounting import cli
from accounting.flows import base


def test_run_activity_worker(mocker):
    """Test run activity worker."""
    worker = mocker.patch('garcon.activity.ActivityWorker')
    cli.garcon('worker', 'reserve_payouts')
    assert worker.called


@patch('boto.swf.layer1.Layer1')
@patch('accounting.cli.swf.WorkflowType')
def test_exec_reserve_payouts(mock_wf_type, mock_layer1, monkeypatch):
    """Test exec."""
    monkeypatch.setattr(base, 'get_domain', MagicMock(return_value='domain'))

    # test reserve_payouts exec
    cli.garcon('exec', 'reserve_payouts', '-c', '{\"period_id\": 214}')
    mock_wf_type.assert_called_with(
        task_list='reserve_payouts-accounting',
        name='reserve_payouts-accounting',
        version='1.0',
        domain='domain')