diff options
author | Peter Sprygada <privateip@users.noreply.github.com> | 2017-02-13 20:22:10 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-02-13 20:22:10 -0500 |
commit | 14b942f3fb17d00be66a734fd2933169bd5978cb (patch) | |
tree | 21b82b98e5384c87b897cf6eb65d8fea742147ee /test/units/modules/network/eos/test_eos_user.py | |
parent | 9937e604f512746478e24724c3cdfc1326e5afd9 (diff) | |
download | ansible-14b942f3fb17d00be66a734fd2933169bd5978cb.tar.gz |
updates eos modules to use socket (#21197)
* updates eos modules to use persistent connection socket
* removes split eos shared module and combines into one
* adds singular eos doc frag (eos_local to be removed after module updates)
* updates unit test cases
Diffstat (limited to 'test/units/modules/network/eos/test_eos_user.py')
-rw-r--r-- | test/units/modules/network/eos/test_eos_user.py | 65 |
1 files changed, 9 insertions, 56 deletions
diff --git a/test/units/modules/network/eos/test_eos_user.py b/test/units/modules/network/eos/test_eos_user.py index 583a8c31e3..413a3c1363 100644 --- a/test/units/modules/network/eos/test_eos_user.py +++ b/test/units/modules/network/eos/test_eos_user.py @@ -17,43 +17,16 @@ from __future__ import (absolute_import, division, print_function) __metaclass__ = type -import os import json -from ansible.compat.tests import unittest -from ansible.compat.tests.mock import patch, MagicMock -from ansible.errors import AnsibleModuleExit +from ansible.compat.tests.mock import patch from ansible.modules.network.eos import eos_user -from ansible.module_utils import basic -from ansible.module_utils._text import to_bytes +from .eos_module import TestEosModule, load_fixture, set_module_args -def set_module_args(args): - args = json.dumps({'ANSIBLE_MODULE_ARGS': args}) - basic._ANSIBLE_ARGS = to_bytes(args) +class TestEosUserModule(TestEosModule): -fixture_path = os.path.join(os.path.dirname(__file__), 'fixtures') -fixture_data = {} - -def load_fixture(name): - path = os.path.join(fixture_path, name) - - if path in fixture_data: - return fixture_data[path] - - with open(path) as f: - data = f.read() - - try: - data = json.loads(data) - except: - pass - - fixture_data[path] = data - return data - - -class TestEosUserModule(unittest.TestCase): + module = eos_user def setUp(self): self.mock_get_config = patch('ansible.modules.network.eos.eos_user.get_config') @@ -66,29 +39,10 @@ class TestEosUserModule(unittest.TestCase): self.mock_get_config.stop() self.mock_load_config.stop() - def execute_module(self, failed=False, changed=False, commands=None, sort=True): - + def load_fixtures(self, commands=None): self.get_config.return_value = load_fixture('eos_user_config.cfg') self.load_config.return_value = dict(diff=None, session='session') - with self.assertRaises(AnsibleModuleExit) as exc: - eos_user.main() - - result = exc.exception.result - - if failed: - self.assertTrue(result['failed'], result) - else: - self.assertEqual(result['changed'], changed, result) - - if commands: - if sort: - self.assertEqual(sorted(commands), sorted(result['commands']), result['commands']) - else: - self.assertEqual(commands, result['commands']) - - return result - def test_eos_user_create(self): set_module_args(dict(username='test', nopassword=True)) commands = ['username test nopassword'] @@ -101,7 +55,7 @@ class TestEosUserModule(unittest.TestCase): def test_eos_user_password(self): set_module_args(dict(username='ansible', password='test')) - commands = ['username ansible secret ********'] + commands = ['username ansible secret test'] self.execute_module(changed=True, commands=commands) def test_eos_user_privilege(self): @@ -130,17 +84,16 @@ class TestEosUserModule(unittest.TestCase): def test_eos_user_update_password_changed(self): set_module_args(dict(username='test', password='test', update_password='on_create')) - commands = ['username ******** secret ********'] + commands = ['username test secret test'] self.execute_module(changed=True, commands=commands) def test_eos_user_update_password_on_create_ok(self): set_module_args(dict(username='ansible', password='test', update_password='on_create')) - commands = [] - self.execute_module(commands=commands) + self.execute_module() def test_eos_user_update_password_always(self): set_module_args(dict(username='ansible', password='test', update_password='always')) - commands = ['username ansible secret ********'] + commands = ['username ansible secret test'] self.execute_module(changed=True, commands=commands) |