mirror of
https://github.com/saltstack/salt.git
synced 2025-04-16 17:50:20 +00:00
saltstack/salt#58412 mock prepare_connection since it depends on netmiko
This commit is contained in:
parent
6ff0a1391a
commit
e9afe9168e
1 changed files with 3 additions and 6 deletions
|
@ -2,7 +2,6 @@
|
|||
import logging
|
||||
|
||||
import salt.modules.netmiko_mod as netmiko_mod
|
||||
from salt.utils.args import get_function_argspec
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.mock import patch
|
||||
from tests.support.unit import TestCase
|
||||
|
@ -34,8 +33,8 @@ def mock_file_apply_template_on_contents(*args):
|
|||
return args[0]
|
||||
|
||||
|
||||
def mock_config_get(key, default):
|
||||
return default
|
||||
def mock_prepare_connection(**kwargs):
|
||||
return MockNetmikoConnection(**kwargs), {}
|
||||
|
||||
|
||||
class NetmikoTestCase(TestCase, LoaderModuleMockMixin):
|
||||
|
@ -44,14 +43,12 @@ class NetmikoTestCase(TestCase, LoaderModuleMockMixin):
|
|||
netmiko_mod: {
|
||||
"__salt__": {
|
||||
"file.apply_template_on_contents": mock_file_apply_template_on_contents,
|
||||
"config.get": mock_config_get,
|
||||
},
|
||||
"__proxy__": {
|
||||
"netmiko.conn": mock_netmiko_conn,
|
||||
"netmiko.args": mock_netmiko_args,
|
||||
},
|
||||
"__utils__": {"args.get_function_argspec": get_function_argspec},
|
||||
"ConnectHandler": MockNetmikoConnection,
|
||||
"_prepare_connection": mock_prepare_connection,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue