35 lines
1 KiB
Python
35 lines
1 KiB
Python
|
import unittest
|
||
|
|
from unittest import mock
|
||
|
|
from unittest.mock import call
|
||
|
|
|
||
|
|
from click.testing import CliRunner
|
||
|
|
|
||
|
|
from ofx_processor.processors.bpvf import BpvfProcessor
|
||
|
|
from ofx_processor.processors.ce import CeProcessor
|
||
|
|
from ofx_processor.processors.revolut import RevolutProcessor
|
||
|
|
from ofx_processor.utils.ynab import config
|
||
|
|
|
||
|
|
|
||
|
|
class UtilsTestCase(unittest.TestCase):
|
||
|
|
@staticmethod
|
||
|
|
def test_discover_processors():
|
||
|
|
ce_main = CeProcessor.main
|
||
|
|
bpvf_main = BpvfProcessor.main
|
||
|
|
revolut_main = RevolutProcessor.main
|
||
|
|
runner = CliRunner()
|
||
|
|
with mock.patch("click.core.Group.add_command") as add_command:
|
||
|
|
from ofx_processor.main import cli
|
||
|
|
|
||
|
|
runner.invoke(cli, ["--help"])
|
||
|
|
calls = [
|
||
|
|
call(ce_main),
|
||
|
|
call(bpvf_main),
|
||
|
|
call(revolut_main),
|
||
|
|
call(config, name="config"),
|
||
|
|
]
|
||
|
|
add_command.assert_has_calls(calls, any_order=True)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main() # pragma: nocover
|