ansible/test/units/modules/network/icx/test_icx_copy.py

204 lines
8.5 KiB
Python
Raw Normal View History

# 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)