mirror of
				https://github.com/ansible-collections/community.general.git
				synced 2025-10-26 21:59:38 -07:00 
			
		
		
		
	* test/: PEP8 compliancy - Make PEP8 compliant * Python3 chokes on casting int to bytes (#24952) But if we tell the formatter that the var is a number, it works
		
			
				
	
	
		
			89 lines
		
	
	
	
		
			3.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			89 lines
		
	
	
	
		
			3.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # -*- coding: utf-8 -*-
 | |
| #
 | |
| # This file is part of Ansible
 | |
| #
 | |
| # Ansible is free software: you can redistribute it and/or modify
 | |
| # it under the terms of the GNU General Public License as published by
 | |
| # the Free Software Foundation, either version 3 of the License, or
 | |
| # (at your option) any later version.
 | |
| #
 | |
| # Ansible is distributed in the hope that it will be useful,
 | |
| # but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
| # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
| # GNU General Public License for more details.
 | |
| #
 | |
| # You should have received a copy of the GNU General Public License
 | |
| # along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
 | |
| 
 | |
| import json
 | |
| import sys
 | |
| 
 | |
| from ansible.compat.tests import unittest
 | |
| from units.mock.procenv import swap_stdin_and_argv, swap_stdout
 | |
| 
 | |
| import ansible.module_utils.basic
 | |
| 
 | |
| 
 | |
| class TestAnsibleModuleWarnDeprecate(unittest.TestCase):
 | |
|     """Test the AnsibleModule Warn Method"""
 | |
| 
 | |
|     def test_warn(self):
 | |
|         args = json.dumps(dict(ANSIBLE_MODULE_ARGS={}))
 | |
|         with swap_stdin_and_argv(stdin_data=args):
 | |
|             with swap_stdout():
 | |
| 
 | |
|                 ansible.module_utils.basic._ANSIBLE_ARGS = None
 | |
|                 am = ansible.module_utils.basic.AnsibleModule(
 | |
|                     argument_spec=dict(),
 | |
|                 )
 | |
|                 am._name = 'unittest'
 | |
| 
 | |
|                 am.warn('warning1')
 | |
| 
 | |
|                 with self.assertRaises(SystemExit):
 | |
|                     am.exit_json(warnings=['warning2'])
 | |
|                 self.assertEquals(json.loads(sys.stdout.getvalue())['warnings'], ['warning1', 'warning2'])
 | |
| 
 | |
|     def test_deprecate(self):
 | |
|         args = json.dumps(dict(ANSIBLE_MODULE_ARGS={}))
 | |
|         with swap_stdin_and_argv(stdin_data=args):
 | |
|             with swap_stdout():
 | |
| 
 | |
|                 ansible.module_utils.basic._ANSIBLE_ARGS = None
 | |
|                 am = ansible.module_utils.basic.AnsibleModule(
 | |
|                     argument_spec=dict(),
 | |
|                 )
 | |
|                 am._name = 'unittest'
 | |
| 
 | |
|                 am.deprecate('deprecation1')
 | |
|                 am.deprecate('deprecation2', '2.3')
 | |
| 
 | |
|                 with self.assertRaises(SystemExit):
 | |
|                     am.exit_json(deprecations=['deprecation3', ('deprecation4', '2.4')])
 | |
|                 output = json.loads(sys.stdout.getvalue())
 | |
|                 self.assertTrue('warnings' not in output or output['warnings'] == [])
 | |
|                 self.assertEquals(output['deprecations'], [
 | |
|                     {u'msg': u'deprecation1', u'version': None},
 | |
|                     {u'msg': u'deprecation2', u'version': '2.3'},
 | |
|                     {u'msg': u'deprecation3', u'version': None},
 | |
|                     {u'msg': u'deprecation4', u'version': '2.4'},
 | |
|                 ])
 | |
| 
 | |
|     def test_deprecate_without_list(self):
 | |
|         args = json.dumps(dict(ANSIBLE_MODULE_ARGS={}))
 | |
|         with swap_stdin_and_argv(stdin_data=args):
 | |
|             with swap_stdout():
 | |
| 
 | |
|                 ansible.module_utils.basic._ANSIBLE_ARGS = None
 | |
|                 am = ansible.module_utils.basic.AnsibleModule(
 | |
|                     argument_spec=dict(),
 | |
|                 )
 | |
|                 am._name = 'unittest'
 | |
| 
 | |
|                 with self.assertRaises(SystemExit):
 | |
|                     am.exit_json(deprecations='Simple deprecation warning')
 | |
|                 output = json.loads(sys.stdout.getvalue())
 | |
|                 self.assertTrue('warnings' not in output or output['warnings'] == [])
 | |
|                 self.assertEquals(output['deprecations'], [
 | |
|                     {u'msg': u'Simple deprecation warning', u'version': None},
 | |
|                 ])
 |