# -*- coding:utf-8 -*-
from unittest.mock import MagicMock, patch

import pytest

from dw_base.datax import cli


@patch('dw_base.datax.cli.JobConfigGenerator')
def test_gen_json_invokes_generator(MockGen):
    mock_inst = MagicMock()
    MockGen.return_value = mock_inst
    cli.main(['gen-json', 'some.ini',
              '-start-date', '20260422',
              '-stop-date', '20260423'])
    MockGen.assert_called_once()
    call_args = MockGen.call_args[0]
    # 位置参数：(base_dir, ini, start, stop, output)
    assert call_args[1] == 'some.ini'
    assert call_args[2] == '20260422'
    assert call_args[3] == '20260423'
    assert call_args[4].endswith('some.json')
    mock_inst.run.assert_called_once()


def test_gen_json_missing_args_exits():
    with pytest.raises(SystemExit):
        cli.main(['gen-json', 'some.ini'])  # 缺 -start-date/-stop-date


def test_unknown_subcommand_exits():
    with pytest.raises(SystemExit):
        cli.main(['bogus-cmd'])