# -*- coding: utf-8; -*- from unittest import TestCase from unittest.mock import MagicMock, patch from wuttasync.cli import import_csv as mod, ImportCommandHandler class TestImportCsv(TestCase): def test_basic(self): params = { "models": [], "create": True, "update": True, "delete": False, "dry_run": True, } ctx = MagicMock(params=params) with patch.object(ImportCommandHandler, "run") as run: mod.import_csv(ctx) run.assert_called_once_with(params)