mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-05-07 22:02:50 +00:00
Reformat everything.
This commit is contained in:
Felix Fontein
@@ -1,4 +1,3 @@
|
||||
|
||||
# Copyright (c) 2020, Ansible Project
|
||||
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||||
# SPDX-License-Identifier: GPL-3.0-or-later
|
||||
@@ -9,7 +8,12 @@ import unittest
|
||||
from contextlib import contextmanager
|
||||
from unittest.mock import call, patch
|
||||
|
||||
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import AnsibleExitJson, AnsibleFailJson, ModuleTestCase, set_module_args
|
||||
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
|
||||
AnsibleExitJson,
|
||||
AnsibleFailJson,
|
||||
ModuleTestCase,
|
||||
set_module_args,
|
||||
)
|
||||
|
||||
from ansible_collections.community.general.plugins.modules import ipa_otptoken
|
||||
|
||||
@@ -30,8 +34,8 @@ def patch_ipa(**kwargs):
|
||||
...
|
||||
"""
|
||||
obj = ipa_otptoken.OTPTokenIPAClient
|
||||
with patch.object(obj, 'login') as mock_login:
|
||||
with patch.object(obj, '_post_json', **kwargs) as mock_post:
|
||||
with patch.object(obj, "login") as mock_login:
|
||||
with patch.object(obj, "_post_json", **kwargs) as mock_post:
|
||||
yield mock_login, mock_post
|
||||
|
||||
|
||||
@@ -81,29 +85,19 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
mock_post.assert_not_called()
|
||||
|
||||
# Verify that the module's changed status matches what is expected
|
||||
self.assertIs(exec_info.exception.args[0]['changed'], changed)
|
||||
self.assertIs(exec_info.exception.args[0]["changed"], changed)
|
||||
|
||||
def test_add_new_all_default(self):
|
||||
"""Add a new OTP with all default values"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1'
|
||||
}
|
||||
module_args = {"uniqueid": "NewToken1"}
|
||||
return_value = {}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_add',
|
||||
'name': 'NewToken1',
|
||||
'item': {'ipatokendisabled': False,
|
||||
'all': True}
|
||||
}
|
||||
{"method": "otptoken_add", "name": "NewToken1", "item": {"ipatokendisabled": False, "all": True}},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -111,25 +105,15 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
|
||||
def test_add_new_all_default_with_aliases(self):
|
||||
"""Add a new OTP with all default values using alias values"""
|
||||
module_args = {
|
||||
'name': 'NewToken1'
|
||||
}
|
||||
module_args = {"name": "NewToken1"}
|
||||
return_value = {}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_add',
|
||||
'name': 'NewToken1',
|
||||
'item': {'ipatokendisabled': False,
|
||||
'all': True}
|
||||
}
|
||||
{"method": "otptoken_add", "name": "NewToken1", "item": {"ipatokendisabled": False, "all": True}},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -138,54 +122,53 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
def test_add_new_all_specified(self):
|
||||
"""Add a new OTP with all default values"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'otptype': 'hotp',
|
||||
'secretkey': 'VGVzdFNlY3JldDE=',
|
||||
'description': 'Test description',
|
||||
'owner': 'pinky',
|
||||
'enabled': True,
|
||||
'notbefore': '20200101010101',
|
||||
'notafter': '20900101010101',
|
||||
'vendor': 'Acme',
|
||||
'model': 'ModelT',
|
||||
'serial': 'Number1',
|
||||
'state': 'present',
|
||||
'algorithm': 'sha256',
|
||||
'digits': 6,
|
||||
'offset': 10,
|
||||
'interval': 30,
|
||||
'counter': 30,
|
||||
"uniqueid": "NewToken1",
|
||||
"otptype": "hotp",
|
||||
"secretkey": "VGVzdFNlY3JldDE=",
|
||||
"description": "Test description",
|
||||
"owner": "pinky",
|
||||
"enabled": True,
|
||||
"notbefore": "20200101010101",
|
||||
"notafter": "20900101010101",
|
||||
"vendor": "Acme",
|
||||
"model": "ModelT",
|
||||
"serial": "Number1",
|
||||
"state": "present",
|
||||
"algorithm": "sha256",
|
||||
"digits": 6,
|
||||
"offset": 10,
|
||||
"interval": 30,
|
||||
"counter": 30,
|
||||
}
|
||||
return_value = {}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_add',
|
||||
'name': 'NewToken1',
|
||||
'item': {'type': 'HOTP',
|
||||
'ipatokenotpkey': 'KRSXG5CTMVRXEZLUGE======',
|
||||
'description': 'Test description',
|
||||
'ipatokenowner': 'pinky',
|
||||
'ipatokendisabled': False,
|
||||
'ipatokennotbefore': '20200101010101Z',
|
||||
'ipatokennotafter': '20900101010101Z',
|
||||
'ipatokenvendor': 'Acme',
|
||||
'ipatokenmodel': 'ModelT',
|
||||
'ipatokenserial': 'Number1',
|
||||
'ipatokenotpalgorithm': 'sha256',
|
||||
'ipatokenotpdigits': '6',
|
||||
'ipatokentotpclockoffset': '10',
|
||||
'ipatokentotptimestep': '30',
|
||||
'ipatokenhotpcounter': '30',
|
||||
'all': True}
|
||||
}
|
||||
"method": "otptoken_add",
|
||||
"name": "NewToken1",
|
||||
"item": {
|
||||
"type": "HOTP",
|
||||
"ipatokenotpkey": "KRSXG5CTMVRXEZLUGE======",
|
||||
"description": "Test description",
|
||||
"ipatokenowner": "pinky",
|
||||
"ipatokendisabled": False,
|
||||
"ipatokennotbefore": "20200101010101Z",
|
||||
"ipatokennotafter": "20900101010101Z",
|
||||
"ipatokenvendor": "Acme",
|
||||
"ipatokenmodel": "ModelT",
|
||||
"ipatokenserial": "Number1",
|
||||
"ipatokenotpalgorithm": "sha256",
|
||||
"ipatokenotpdigits": "6",
|
||||
"ipatokentotpclockoffset": "10",
|
||||
"ipatokentotptimestep": "30",
|
||||
"ipatokenhotpcounter": "30",
|
||||
"all": True,
|
||||
},
|
||||
},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -194,48 +177,47 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
def test_already_existing_no_change_all_specified(self):
|
||||
"""Add a new OTP with all values specified but needing no change"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'otptype': 'hotp',
|
||||
'secretkey': 'VGVzdFNlY3JldDE=',
|
||||
'description': 'Test description',
|
||||
'owner': 'pinky',
|
||||
'enabled': True,
|
||||
'notbefore': '20200101010101',
|
||||
'notafter': '20900101010101',
|
||||
'vendor': 'Acme',
|
||||
'model': 'ModelT',
|
||||
'serial': 'Number1',
|
||||
'state': 'present',
|
||||
'algorithm': 'sha256',
|
||||
'digits': 6,
|
||||
'offset': 10,
|
||||
'interval': 30,
|
||||
'counter': 30,
|
||||
"uniqueid": "NewToken1",
|
||||
"otptype": "hotp",
|
||||
"secretkey": "VGVzdFNlY3JldDE=",
|
||||
"description": "Test description",
|
||||
"owner": "pinky",
|
||||
"enabled": True,
|
||||
"notbefore": "20200101010101",
|
||||
"notafter": "20900101010101",
|
||||
"vendor": "Acme",
|
||||
"model": "ModelT",
|
||||
"serial": "Number1",
|
||||
"state": "present",
|
||||
"algorithm": "sha256",
|
||||
"digits": 6,
|
||||
"offset": 10,
|
||||
"interval": 30,
|
||||
"counter": 30,
|
||||
}
|
||||
return_value = {
|
||||
"ipatokenuniqueid": "NewToken1",
|
||||
"type": "HOTP",
|
||||
"ipatokenotpkey": [{"__base64__": "VGVzdFNlY3JldDE="}],
|
||||
"description": ["Test description"],
|
||||
"ipatokenowner": ["pinky"],
|
||||
"ipatokendisabled": [False],
|
||||
"ipatokennotbefore": ["20200101010101Z"],
|
||||
"ipatokennotafter": ["20900101010101Z"],
|
||||
"ipatokenvendor": ["Acme"],
|
||||
"ipatokenmodel": ["ModelT"],
|
||||
"ipatokenserial": ["Number1"],
|
||||
"ipatokenotpalgorithm": ["sha256"],
|
||||
"ipatokenotpdigits": ["6"],
|
||||
"ipatokentotpclockoffset": ["10"],
|
||||
"ipatokentotptimestep": ["30"],
|
||||
"ipatokenhotpcounter": ["30"],
|
||||
}
|
||||
return_value = {'ipatokenuniqueid': 'NewToken1',
|
||||
'type': 'HOTP',
|
||||
'ipatokenotpkey': [{'__base64__': 'VGVzdFNlY3JldDE='}],
|
||||
'description': ['Test description'],
|
||||
'ipatokenowner': ['pinky'],
|
||||
'ipatokendisabled': [False],
|
||||
'ipatokennotbefore': ['20200101010101Z'],
|
||||
'ipatokennotafter': ['20900101010101Z'],
|
||||
'ipatokenvendor': ['Acme'],
|
||||
'ipatokenmodel': ['ModelT'],
|
||||
'ipatokenserial': ['Number1'],
|
||||
'ipatokenotpalgorithm': ['sha256'],
|
||||
'ipatokenotpdigits': ['6'],
|
||||
'ipatokentotpclockoffset': ['10'],
|
||||
'ipatokentotptimestep': ['30'],
|
||||
'ipatokenhotpcounter': ['30']}
|
||||
mock_calls = [
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
}
|
||||
]
|
||||
changed = False
|
||||
@@ -245,62 +227,63 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
def test_already_existing_one_change_all_specified(self):
|
||||
"""Modify an existing OTP with one value specified needing change"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'otptype': 'hotp',
|
||||
'secretkey': 'VGVzdFNlY3JldDE=',
|
||||
'description': 'Test description',
|
||||
'owner': 'brain',
|
||||
'enabled': True,
|
||||
'notbefore': '20200101010101',
|
||||
'notafter': '20900101010101',
|
||||
'vendor': 'Acme',
|
||||
'model': 'ModelT',
|
||||
'serial': 'Number1',
|
||||
'state': 'present',
|
||||
'algorithm': 'sha256',
|
||||
'digits': 6,
|
||||
'offset': 10,
|
||||
'interval': 30,
|
||||
'counter': 30,
|
||||
"uniqueid": "NewToken1",
|
||||
"otptype": "hotp",
|
||||
"secretkey": "VGVzdFNlY3JldDE=",
|
||||
"description": "Test description",
|
||||
"owner": "brain",
|
||||
"enabled": True,
|
||||
"notbefore": "20200101010101",
|
||||
"notafter": "20900101010101",
|
||||
"vendor": "Acme",
|
||||
"model": "ModelT",
|
||||
"serial": "Number1",
|
||||
"state": "present",
|
||||
"algorithm": "sha256",
|
||||
"digits": 6,
|
||||
"offset": 10,
|
||||
"interval": 30,
|
||||
"counter": 30,
|
||||
}
|
||||
return_value = {
|
||||
"ipatokenuniqueid": "NewToken1",
|
||||
"type": "HOTP",
|
||||
"ipatokenotpkey": [{"__base64__": "VGVzdFNlY3JldDE="}],
|
||||
"description": ["Test description"],
|
||||
"ipatokenowner": ["pinky"],
|
||||
"ipatokendisabled": [False],
|
||||
"ipatokennotbefore": ["20200101010101Z"],
|
||||
"ipatokennotafter": ["20900101010101Z"],
|
||||
"ipatokenvendor": ["Acme"],
|
||||
"ipatokenmodel": ["ModelT"],
|
||||
"ipatokenserial": ["Number1"],
|
||||
"ipatokenotpalgorithm": ["sha256"],
|
||||
"ipatokenotpdigits": ["6"],
|
||||
"ipatokentotpclockoffset": ["10"],
|
||||
"ipatokentotptimestep": ["30"],
|
||||
"ipatokenhotpcounter": ["30"],
|
||||
}
|
||||
return_value = {'ipatokenuniqueid': 'NewToken1',
|
||||
'type': 'HOTP',
|
||||
'ipatokenotpkey': [{'__base64__': 'VGVzdFNlY3JldDE='}],
|
||||
'description': ['Test description'],
|
||||
'ipatokenowner': ['pinky'],
|
||||
'ipatokendisabled': [False],
|
||||
'ipatokennotbefore': ['20200101010101Z'],
|
||||
'ipatokennotafter': ['20900101010101Z'],
|
||||
'ipatokenvendor': ['Acme'],
|
||||
'ipatokenmodel': ['ModelT'],
|
||||
'ipatokenserial': ['Number1'],
|
||||
'ipatokenotpalgorithm': ['sha256'],
|
||||
'ipatokenotpdigits': ['6'],
|
||||
'ipatokentotpclockoffset': ['10'],
|
||||
'ipatokentotptimestep': ['30'],
|
||||
'ipatokenhotpcounter': ['30']}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_mod',
|
||||
'name': 'NewToken1',
|
||||
'item': {'description': 'Test description',
|
||||
'ipatokenowner': 'brain',
|
||||
'ipatokendisabled': False,
|
||||
'ipatokennotbefore': '20200101010101Z',
|
||||
'ipatokennotafter': '20900101010101Z',
|
||||
'ipatokenvendor': 'Acme',
|
||||
'ipatokenmodel': 'ModelT',
|
||||
'ipatokenserial': 'Number1',
|
||||
'all': True}
|
||||
}
|
||||
"method": "otptoken_mod",
|
||||
"name": "NewToken1",
|
||||
"item": {
|
||||
"description": "Test description",
|
||||
"ipatokenowner": "brain",
|
||||
"ipatokendisabled": False,
|
||||
"ipatokennotbefore": "20200101010101Z",
|
||||
"ipatokennotafter": "20900101010101Z",
|
||||
"ipatokenvendor": "Acme",
|
||||
"ipatokenmodel": "ModelT",
|
||||
"ipatokenserial": "Number1",
|
||||
"all": True,
|
||||
},
|
||||
},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -309,62 +292,63 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
def test_already_existing_all_valid_change_all_specified(self):
|
||||
"""Modify an existing OTP with all valid values specified needing change"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'otptype': 'hotp',
|
||||
'secretkey': 'VGVzdFNlY3JldDE=',
|
||||
'description': 'New Test description',
|
||||
'owner': 'pinky',
|
||||
'enabled': False,
|
||||
'notbefore': '20200101010102',
|
||||
'notafter': '20900101010102',
|
||||
'vendor': 'NewAcme',
|
||||
'model': 'NewModelT',
|
||||
'serial': 'Number2',
|
||||
'state': 'present',
|
||||
'algorithm': 'sha256',
|
||||
'digits': 6,
|
||||
'offset': 10,
|
||||
'interval': 30,
|
||||
'counter': 30,
|
||||
"uniqueid": "NewToken1",
|
||||
"otptype": "hotp",
|
||||
"secretkey": "VGVzdFNlY3JldDE=",
|
||||
"description": "New Test description",
|
||||
"owner": "pinky",
|
||||
"enabled": False,
|
||||
"notbefore": "20200101010102",
|
||||
"notafter": "20900101010102",
|
||||
"vendor": "NewAcme",
|
||||
"model": "NewModelT",
|
||||
"serial": "Number2",
|
||||
"state": "present",
|
||||
"algorithm": "sha256",
|
||||
"digits": 6,
|
||||
"offset": 10,
|
||||
"interval": 30,
|
||||
"counter": 30,
|
||||
}
|
||||
return_value = {
|
||||
"ipatokenuniqueid": "NewToken1",
|
||||
"type": "HOTP",
|
||||
"ipatokenotpkey": [{"__base64__": "VGVzdFNlY3JldDE="}],
|
||||
"description": ["Test description"],
|
||||
"ipatokenowner": ["pinky"],
|
||||
"ipatokendisabled": [False],
|
||||
"ipatokennotbefore": ["20200101010101Z"],
|
||||
"ipatokennotafter": ["20900101010101Z"],
|
||||
"ipatokenvendor": ["Acme"],
|
||||
"ipatokenmodel": ["ModelT"],
|
||||
"ipatokenserial": ["Number1"],
|
||||
"ipatokenotpalgorithm": ["sha256"],
|
||||
"ipatokenotpdigits": ["6"],
|
||||
"ipatokentotpclockoffset": ["10"],
|
||||
"ipatokentotptimestep": ["30"],
|
||||
"ipatokenhotpcounter": ["30"],
|
||||
}
|
||||
return_value = {'ipatokenuniqueid': 'NewToken1',
|
||||
'type': 'HOTP',
|
||||
'ipatokenotpkey': [{'__base64__': 'VGVzdFNlY3JldDE='}],
|
||||
'description': ['Test description'],
|
||||
'ipatokenowner': ['pinky'],
|
||||
'ipatokendisabled': [False],
|
||||
'ipatokennotbefore': ['20200101010101Z'],
|
||||
'ipatokennotafter': ['20900101010101Z'],
|
||||
'ipatokenvendor': ['Acme'],
|
||||
'ipatokenmodel': ['ModelT'],
|
||||
'ipatokenserial': ['Number1'],
|
||||
'ipatokenotpalgorithm': ['sha256'],
|
||||
'ipatokenotpdigits': ['6'],
|
||||
'ipatokentotpclockoffset': ['10'],
|
||||
'ipatokentotptimestep': ['30'],
|
||||
'ipatokenhotpcounter': ['30']}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_mod',
|
||||
'name': 'NewToken1',
|
||||
'item': {'description': 'New Test description',
|
||||
'ipatokenowner': 'pinky',
|
||||
'ipatokendisabled': True,
|
||||
'ipatokennotbefore': '20200101010102Z',
|
||||
'ipatokennotafter': '20900101010102Z',
|
||||
'ipatokenvendor': 'NewAcme',
|
||||
'ipatokenmodel': 'NewModelT',
|
||||
'ipatokenserial': 'Number2',
|
||||
'all': True}
|
||||
}
|
||||
"method": "otptoken_mod",
|
||||
"name": "NewToken1",
|
||||
"item": {
|
||||
"description": "New Test description",
|
||||
"ipatokenowner": "pinky",
|
||||
"ipatokendisabled": True,
|
||||
"ipatokennotbefore": "20200101010102Z",
|
||||
"ipatokennotafter": "20900101010102Z",
|
||||
"ipatokenvendor": "NewAcme",
|
||||
"ipatokenmodel": "NewModelT",
|
||||
"ipatokenserial": "Number2",
|
||||
"all": True,
|
||||
},
|
||||
},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -372,39 +356,32 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
|
||||
def test_delete_existing_token(self):
|
||||
"""Delete an existing OTP"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'state': 'absent'
|
||||
module_args = {"uniqueid": "NewToken1", "state": "absent"}
|
||||
return_value = {
|
||||
"ipatokenuniqueid": "NewToken1",
|
||||
"type": "HOTP",
|
||||
"ipatokenotpkey": [{"__base64__": "KRSXG5CTMVRXEZLUGE======"}],
|
||||
"description": ["Test description"],
|
||||
"ipatokenowner": ["pinky"],
|
||||
"ipatokendisabled": [False],
|
||||
"ipatokennotbefore": ["20200101010101Z"],
|
||||
"ipatokennotafter": ["20900101010101Z"],
|
||||
"ipatokenvendor": ["Acme"],
|
||||
"ipatokenmodel": ["ModelT"],
|
||||
"ipatokenserial": ["Number1"],
|
||||
"ipatokenotpalgorithm": ["sha256"],
|
||||
"ipatokenotpdigits": ["6"],
|
||||
"ipatokentotpclockoffset": ["10"],
|
||||
"ipatokentotptimestep": ["30"],
|
||||
"ipatokenhotpcounter": ["30"],
|
||||
}
|
||||
return_value = {'ipatokenuniqueid': 'NewToken1',
|
||||
'type': 'HOTP',
|
||||
'ipatokenotpkey': [{'__base64__': 'KRSXG5CTMVRXEZLUGE======'}],
|
||||
'description': ['Test description'],
|
||||
'ipatokenowner': ['pinky'],
|
||||
'ipatokendisabled': [False],
|
||||
'ipatokennotbefore': ['20200101010101Z'],
|
||||
'ipatokennotafter': ['20900101010101Z'],
|
||||
'ipatokenvendor': ['Acme'],
|
||||
'ipatokenmodel': ['ModelT'],
|
||||
'ipatokenserial': ['Number1'],
|
||||
'ipatokenotpalgorithm': ['sha256'],
|
||||
'ipatokenotpdigits': ['6'],
|
||||
'ipatokentotpclockoffset': ['10'],
|
||||
'ipatokentotptimestep': ['30'],
|
||||
'ipatokenhotpcounter': ['30']}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_del',
|
||||
'name': 'NewToken1'
|
||||
}
|
||||
{"method": "otptoken_del", "name": "NewToken1"},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -412,42 +389,32 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
|
||||
def test_disable_existing_token(self):
|
||||
"""Disable an existing OTP"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'otptype': 'hotp',
|
||||
'enabled': False
|
||||
module_args = {"uniqueid": "NewToken1", "otptype": "hotp", "enabled": False}
|
||||
return_value = {
|
||||
"ipatokenuniqueid": "NewToken1",
|
||||
"type": "HOTP",
|
||||
"ipatokenotpkey": [{"__base64__": "KRSXG5CTMVRXEZLUGE======"}],
|
||||
"description": ["Test description"],
|
||||
"ipatokenowner": ["pinky"],
|
||||
"ipatokendisabled": [False],
|
||||
"ipatokennotbefore": ["20200101010101Z"],
|
||||
"ipatokennotafter": ["20900101010101Z"],
|
||||
"ipatokenvendor": ["Acme"],
|
||||
"ipatokenmodel": ["ModelT"],
|
||||
"ipatokenserial": ["Number1"],
|
||||
"ipatokenotpalgorithm": ["sha256"],
|
||||
"ipatokenotpdigits": ["6"],
|
||||
"ipatokentotpclockoffset": ["10"],
|
||||
"ipatokentotptimestep": ["30"],
|
||||
"ipatokenhotpcounter": ["30"],
|
||||
}
|
||||
return_value = {'ipatokenuniqueid': 'NewToken1',
|
||||
'type': 'HOTP',
|
||||
'ipatokenotpkey': [{'__base64__': 'KRSXG5CTMVRXEZLUGE======'}],
|
||||
'description': ['Test description'],
|
||||
'ipatokenowner': ['pinky'],
|
||||
'ipatokendisabled': [False],
|
||||
'ipatokennotbefore': ['20200101010101Z'],
|
||||
'ipatokennotafter': ['20900101010101Z'],
|
||||
'ipatokenvendor': ['Acme'],
|
||||
'ipatokenmodel': ['ModelT'],
|
||||
'ipatokenserial': ['Number1'],
|
||||
'ipatokenotpalgorithm': ['sha256'],
|
||||
'ipatokenotpdigits': ['6'],
|
||||
'ipatokentotpclockoffset': ['10'],
|
||||
'ipatokentotptimestep': ['30'],
|
||||
'ipatokenhotpcounter': ['30']}
|
||||
mock_calls = (
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
},
|
||||
{
|
||||
'method': 'otptoken_mod',
|
||||
'name': 'NewToken1',
|
||||
'item': {'ipatokendisabled': True,
|
||||
'all': True}
|
||||
}
|
||||
{"method": "otptoken_mod", "name": "NewToken1", "item": {"ipatokendisabled": True, "all": True}},
|
||||
)
|
||||
changed = True
|
||||
|
||||
@@ -455,20 +422,14 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
|
||||
def test_delete_not_existing_token(self):
|
||||
"""Delete a OTP that does not exist"""
|
||||
module_args = {
|
||||
'uniqueid': 'NewToken1',
|
||||
'state': 'absent'
|
||||
}
|
||||
module_args = {"uniqueid": "NewToken1", "state": "absent"}
|
||||
return_value = {}
|
||||
|
||||
mock_calls = [
|
||||
{
|
||||
'method': 'otptoken_find',
|
||||
'name': None,
|
||||
'item': {'all': True,
|
||||
'ipatokenuniqueid': 'NewToken1',
|
||||
'timelimit': '0',
|
||||
'sizelimit': '0'}
|
||||
"method": "otptoken_find",
|
||||
"name": None,
|
||||
"item": {"all": True, "ipatokenuniqueid": "NewToken1", "timelimit": "0", "sizelimit": "0"},
|
||||
}
|
||||
]
|
||||
|
||||
@@ -478,15 +439,13 @@ class TestIPAOTPToken(ModuleTestCase):
|
||||
|
||||
def test_fail_post(self):
|
||||
"""Fail due to an exception raised from _post_json"""
|
||||
with set_module_args({
|
||||
'uniqueid': 'NewToken1'
|
||||
}):
|
||||
with patch_ipa(side_effect=Exception('ERROR MESSAGE')) as (mock_login, mock_post):
|
||||
with set_module_args({"uniqueid": "NewToken1"}):
|
||||
with patch_ipa(side_effect=Exception("ERROR MESSAGE")) as (mock_login, mock_post):
|
||||
with self.assertRaises(AnsibleFailJson) as exec_info:
|
||||
self.module.main()
|
||||
|
||||
self.assertEqual(exec_info.exception.args[0]['msg'], 'ERROR MESSAGE')
|
||||
self.assertEqual(exec_info.exception.args[0]["msg"], "ERROR MESSAGE")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user