204 lines
8.5 KiB
Python
204 lines
8.5 KiB
Python
|
# Copyright: (c) 2019, Ansible Project
|
||
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||
|
|
from __future__ import (absolute_import, division, print_function)
|
||
|
|
__metaclass__ = type
|
||
|
|
|
||
|
|
from units.compat.mock import patch
|
||
|
|
from ansible.modules.network.icx import icx_copy
|
||
|
|
from units.modules.utils import set_module_args
|
||
|
|
from .icx_module import TestICXModule, load_fixture
|
||
|
|
|
||
|
|
|
||
|
|
class TestICXSCPModule(TestICXModule):
|
||
|
|
|
||
|
|
module = icx_copy
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(TestICXSCPModule, self).setUp()
|
||
|
|
self.mock_exec_scp = patch('ansible.modules.network.icx.icx_copy.exec_scp')
|
||
|
|
self.mock_run_commands = patch('ansible.modules.network.icx.icx_copy.run_commands')
|
||
|
|
self.exec_command = self.mock_exec_scp.start()
|
||
|
|
self.run_commands = self.mock_run_commands.start()
|
||
|
|
self.mock_exec_command = patch('ansible.modules.network.icx.icx_copy.exec_command')
|
||
|
|
self.exec_commands = self.mock_exec_command.start()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super(TestICXSCPModule, self).tearDown()
|
||
|
|
self.mock_exec_scp.stop()
|
||
|
|
self.mock_run_commands.stop()
|
||
|
|
self.mock_exec_command.stop()
|
||
|
|
|
||
|
|
def load_fixtures(self, commands=None):
|
||
|
|
self.exec_commands.return_value = (0, load_fixture('icx_copy.txt').strip(), None)
|
||
|
|
# self.exec_command.return_value = (0, load_fixture('icx_banner_show_banner.txt').strip(), None)
|
||
|
|
if(commands is not None):
|
||
|
|
fixtureName = commands[0].replace(" ", "_") + ".txt"
|
||
|
|
# print("loading fixture: ",load_fixture(fixtureName).strip())
|
||
|
|
self.mock_exec_scp.return_value = load_fixture("icx_copy.txt").strip()
|
||
|
|
self.mock_run_commands.return_value = load_fixture("icx_copy.txt").strip()
|
||
|
|
else:
|
||
|
|
self.exec_command.return_value = ""
|
||
|
|
|
||
|
|
def test_icx_scp_upload_running(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='running-config',
|
||
|
|
protocol='scp',
|
||
|
|
remote_server='172.16.10.49',
|
||
|
|
remote_filename='running.conf',
|
||
|
|
remote_user='alethea',
|
||
|
|
remote_pass='alethea123'))
|
||
|
|
commands = ['copy running-config scp 172.16.10.49 running.conf']
|
||
|
|
self.execute_module(commands=commands)
|
||
|
|
|
||
|
|
def test_icx_scp_download_running(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
download='running-config',
|
||
|
|
protocol='scp',
|
||
|
|
remote_server='172.16.10.49',
|
||
|
|
remote_filename='running.conf',
|
||
|
|
remote_user='alethea',
|
||
|
|
remote_pass='alethea123'))
|
||
|
|
commands = ['copy scp running-config 172.16.10.49 running.conf']
|
||
|
|
self.execute_module(commands=commands, changed=True)
|
||
|
|
|
||
|
|
def test_icx_scp_upload_startup(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='startup-config',
|
||
|
|
protocol='scp',
|
||
|
|
remote_server='172.16.10.49',
|
||
|
|
remote_filename='running.conf',
|
||
|
|
remote_user='alethea',
|
||
|
|
remote_pass='alethea123'))
|
||
|
|
commands = ['copy startup-config scp 172.16.10.49 running.conf']
|
||
|
|
self.execute_module(commands=commands, changed=False)
|
||
|
|
|
||
|
|
def test_icx_scp_download_startup(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
download='startup-config',
|
||
|
|
protocol='scp',
|
||
|
|
remote_server='172.16.10.49',
|
||
|
|
remote_filename='running.conf',
|
||
|
|
remote_user='alethea',
|
||
|
|
remote_pass='alethea123'))
|
||
|
|
commands = ['copy scp startup-config 172.16.10.49 running.conf']
|
||
|
|
self.execute_module(commands=commands, changed=True)
|
||
|
|
|
||
|
|
def test_icx_scp_upload_primary(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='flash_primary',
|
||
|
|
protocol='scp',
|
||
|
|
remote_server='172.16.10.49',
|
||
|
|
remote_filename='SPS08080b.bin',
|
||
|
|
remote_user='alethea',
|
||
|
|
remote_pass='alethea123'))
|
||
|
|
commands = ['copy flash scp 172.16.10.49 SPS08080b.bin primary']
|
||
|
|
self.execute_module(commands=commands, changed=False)
|
||
|
|
|
||
|
|
def test_icx_scp_download_primary(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
download='flash_primary',
|
||
|
|
protocol='scp',
|
||
|
|
remote_server='172.16.10.49',
|
||
|
|
remote_filename='SPS08080b.bin',
|
||
|
|
remote_user='alethea',
|
||
|
|
remote_pass='alethea123'))
|
||
|
|
commands = ['copy scp flash 172.16.10.49 SPS08080b.bin primary']
|
||
|
|
self.execute_module(commands=commands, changed=True)
|
||
|
|
|
||
|
|
# HTTPS tests
|
||
|
|
|
||
|
|
def test_icx_https_upload_running(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='running-config',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/upload_running'))
|
||
|
|
commands = ['copy running-config https fileserver.alethea.in filestorage/test/upload_running']
|
||
|
|
self.execute_module(commands=commands)
|
||
|
|
|
||
|
|
def test_icx_https_download_running(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
download='running-config',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/running.conf'))
|
||
|
|
commands = ['copy https running-config fileserver.alethea.in filestorage/test/running.conf']
|
||
|
|
self.execute_module(failed=True)
|
||
|
|
|
||
|
|
def test_icx_https_upload_startup(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='startup-config',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/upload_startup'))
|
||
|
|
commands = ['copy startup-config https fileserver.alethea.in filestorage/test/upload_startup']
|
||
|
|
self.execute_module(commands=commands)
|
||
|
|
|
||
|
|
def test_icx_https_download_startup(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
download='startup-config',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/startup.conf'))
|
||
|
|
commands = ['copy https startup-config fileserver.alethea.in filestorage/test/startup.conf']
|
||
|
|
self.execute_module(commands=commands, changed=True)
|
||
|
|
|
||
|
|
def test_icx_https_upload_primary(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='flash_primary',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/upload_primary'))
|
||
|
|
commands = ['copy startup-config https fileserver.alethea.in filestorage/test/upload_primary']
|
||
|
|
self.execute_module(failed=True)
|
||
|
|
|
||
|
|
def test_icx_https_download_primary(self):
|
||
|
|
set_module_args(dict(download='flash_primary', protocol='https', remote_server='fileserver.alethea.in', remote_filename='filestorage/test/primary.bin'))
|
||
|
|
commands = ['copy https flash fileserver.alethea.in filestorage/test/primary.bin primary']
|
||
|
|
self.execute_module(commands=commands, changed=True)
|
||
|
|
|
||
|
|
def test_icx_https_upload_secondary(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='flash_secondary',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/upload_secondary'))
|
||
|
|
commands = ['copy flash https fileserver.alethea.in filestorage/test/upload_secondary secondary']
|
||
|
|
self.execute_module(failed=True)
|
||
|
|
|
||
|
|
def test_icx_https_download_secondary(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
download='flash_secondary',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/secondary.bin'))
|
||
|
|
commands = ['copy https flash fileserver.alethea.in filestorage/test/secondary.bin secondary']
|
||
|
|
self.execute_module(commands=commands, changed=True)
|
||
|
|
|
||
|
|
def test_icx_https_upload_download(self):
|
||
|
|
set_module_args(
|
||
|
|
dict(
|
||
|
|
upload='flash_secondary',
|
||
|
|
download='flash_secondary',
|
||
|
|
protocol='https',
|
||
|
|
remote_server='fileserver.alethea.in',
|
||
|
|
remote_filename='filestorage/test/secondary.bin'))
|
||
|
|
self.execute_module(failed=True)
|
||
|
|
|
||
|
|
def test_icx_scp_no_user(self):
|
||
|
|
set_module_args(dict(upload='running-config', protocol='scp', remote_server='172.16.10.49', remote_filename='running.conf'))
|
||
|
|
self.execute_module(failed=True)
|