25 lines
586 B
Python
25 lines
586 B
Python
import pytest
|
|
|
|
import mock
|
|
|
|
from onedns import cli
|
|
from onedns.tests import test_shell
|
|
|
|
|
|
def test_cli_help():
|
|
with pytest.raises(SystemExit):
|
|
cli.main(args=['--help'])
|
|
|
|
|
|
@mock.patch.object(cli, 'logger', mock.MagicMock())
|
|
def test_cli_subcmd_daemon(vms):
|
|
cli.main(args=['daemon'], testing=True, vms=vms)
|
|
|
|
|
|
@mock.patch.dict('sys.modules', test_shell.IPY_MODULES)
|
|
@mock.patch.object(cli, 'logger', mock.MagicMock())
|
|
def test_cli_subcmd_shell():
|
|
test_shell.IPY.embed.reset_mock()
|
|
cli.main(args=['shell'], testing=True)
|
|
test_shell.IPY.embed.assert_called_once()
|