Skip to content

Commit

Permalink
Minor linting fixups
Browse files Browse the repository at this point in the history
  • Loading branch information
tremble committed Aug 8, 2022
1 parent 884c242 commit bc85d19
Show file tree
Hide file tree
Showing 10 changed files with 34 additions and 36 deletions.
5 changes: 3 additions & 2 deletions plugins/module_utils/acm.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@
# on behalf of Telstra Corporation Limited
#
# Common functionality to be used by the modules:
# - acm
# - acm_certificate
# - acm_certificate_info

from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
Expand Down Expand Up @@ -213,7 +214,7 @@ def import_certificate(self, client, module, certificate, private_key, arn=None,
module.debug("Attempting to delete the cert we just created, arn=%s" % arn)
try:
self.delete_certificate_with_backoff(client, arn)
except Exception as f:
except Exception:
module.warn("Certificate %s exists, and is not tagged. So Ansible will not see it on the next run.")
module.fail_json_aws(e, msg="Couldn't tag certificate %s, couldn't delete it either" % arn)
module.fail_json_aws(e, msg="Couldn't tag certificate %s" % arn)
Expand Down
4 changes: 2 additions & 2 deletions plugins/module_utils/botocore.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def boto3_conn(module, conn_type=None, resource=None, region=None, endpoint=None
except (botocore.exceptions.ProfileNotFound, botocore.exceptions.PartialCredentialsError,
botocore.exceptions.NoCredentialsError, botocore.exceptions.ConfigParseError) as e:
module.fail_json(msg=to_native(e))
except botocore.exceptions.NoRegionError as e:
except botocore.exceptions.NoRegionError:
module.fail_json(msg="The %s module requires a region and none was found in configuration, "
"environment variables or module parameters" % module._name)

Expand Down Expand Up @@ -155,7 +155,7 @@ def get_aws_region(module, boto3=None):
try:
profile_name = module.params.get('profile')
return botocore.session.Session(profile=profile_name).get_config_variable('region')
except botocore.exceptions.ProfileNotFound as e:
except botocore.exceptions.ProfileNotFound:
return None


Expand Down
2 changes: 1 addition & 1 deletion plugins/module_utils/iam.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def get_aws_account_info(module):
except (botocore.exceptions.BotoCoreError, botocore.exceptions.ClientError):
try:
iam_client = module.client('iam', retry_decorator=AWSRetry.jittered_backoff())
arn, partition, service, reg, account_id, resource = iam_client.get_user(aws_retry=True)['User']['Arn'].split(':')
_arn, partition, _service, _reg, account_id, _resource = iam_client.get_user(aws_retry=True)['User']['Arn'].split(':')
except is_boto3_error_code('AccessDenied') as e:
try:
except_msg = to_native(e.message)
Expand Down
1 change: 0 additions & 1 deletion plugins/module_utils/modules.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,6 @@ def __init__(self, **kwargs):
if not HAS_BOTO3:
self._module.fail_json(
msg=missing_required_lib('botocore or boto3'))
current_versions = self._gather_versions()
if not self.botocore_at_least('1.21.0'):
self.warn('botocore < 1.21.0 is not supported or tested.'
' Some features may not work.')
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def test_fail_client_minimal(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.EXAMPLE_MSG
Expand All @@ -88,7 +88,7 @@ def test_fail_client_msg(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, msg=self.FAIL_MSG)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.FAIL_MSG + ": " + self.EXAMPLE_MSG
Expand All @@ -115,7 +115,7 @@ def test_fail_client_positional_msg(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, self.FAIL_MSG)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.FAIL_MSG + ": " + self.EXAMPLE_MSG
Expand All @@ -142,7 +142,7 @@ def test_fail_client_key(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, extra_key="Some Value")
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.EXAMPLE_MSG
Expand Down Expand Up @@ -170,7 +170,7 @@ def test_fail_client_msg_and_key(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, extra_key="Some Value", msg=self.FAIL_MSG)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.FAIL_MSG + ": " + self.EXAMPLE_MSG
Expand Down Expand Up @@ -198,7 +198,7 @@ def test_fail_botocore_minimal(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.DEFAULT_CORE_MSG
Expand All @@ -225,7 +225,7 @@ def test_fail_botocore_msg(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, msg=self.FAIL_MSG)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.FAIL_MSG + ": " + self.DEFAULT_CORE_MSG
Expand Down Expand Up @@ -253,7 +253,7 @@ def test_fail_botocore_positional_msg(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, self.FAIL_MSG)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.FAIL_MSG + ": " + self.DEFAULT_CORE_MSG
Expand All @@ -280,7 +280,7 @@ def test_fail_botocore_key(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, extra_key="Some Value")
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.DEFAULT_CORE_MSG
Expand Down Expand Up @@ -308,7 +308,7 @@ def test_fail_botocore_msg_and_key(self, monkeypatch, stdin, capfd):
with pytest.raises(SystemExit) as ctx:
module.fail_json_aws(e, extra_key="Some Value", msg=self.FAIL_MSG)
assert ctx.value.code == 1
out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("msg") == self.FAIL_MSG + ": " + self.DEFAULT_CORE_MSG
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,10 +36,10 @@ def test_no_warn(self, monkeypatch, stdin, capfd):
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.exit_json()

out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("exception") is None
Expand All @@ -59,10 +59,10 @@ def test_no_check(self, monkeypatch, stdin, capfd):
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict(), check_boto3=False)

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.exit_json()

out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("exception") is None
Expand All @@ -82,7 +82,7 @@ def test_warn_boto3(self, monkeypatch, stdin, capfd):
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.exit_json()

out, err = capfd.readouterr()
Expand Down Expand Up @@ -115,7 +115,7 @@ def test_warn_botocore(self, monkeypatch, stdin, capfd):
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.exit_json()

out, err = capfd.readouterr()
Expand Down Expand Up @@ -148,7 +148,7 @@ def test_warn_boto3_and_botocore(self, monkeypatch, stdin, capfd):
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.exit_json()

out, err = capfd.readouterr()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -85,11 +85,11 @@ def test_require_botocore_at_least(self, monkeypatch, stdin, desired_version, co
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.require_botocore_at_least(desired_version)
module.exit_json()

out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("exception") is None
Expand Down Expand Up @@ -118,11 +118,11 @@ def test_require_boto3_at_least(self, monkeypatch, stdin, desired_version, compa
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.require_boto3_at_least(desired_version)
module.exit_json()

out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("exception") is None
Expand Down Expand Up @@ -153,11 +153,11 @@ def test_require_botocore_at_least_with_reason(self, monkeypatch, stdin, desired
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.require_botocore_at_least(desired_version, reason=reason)
module.exit_json()

out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("exception") is None
Expand Down Expand Up @@ -189,11 +189,11 @@ def test_require_boto3_at_least_with_reason(self, monkeypatch, stdin, desired_ve
# Create a minimal module that we can call
module = AnsibleAWSModule(argument_spec=dict())

with pytest.raises(SystemExit) as e:
with pytest.raises(SystemExit):
module.require_boto3_at_least(desired_version, reason=reason)
module.exit_json()

out, err = capfd.readouterr()
out, _err = capfd.readouterr()
return_val = json.loads(out)

assert return_val.get("exception") is None
Expand Down
2 changes: 0 additions & 2 deletions tests/unit/module_utils/test_cloud.py
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,6 @@ def _fail():
# the elapsed execution time should be closed to 2sec
for _i in range(3):
start = datetime.now()
raised = False
with self.assertRaises(self.TestException):
_fail()
duration = (datetime.now() - start).seconds
Expand Down Expand Up @@ -265,7 +264,6 @@ def _fail_exception():
duration = expection[2]

start = datetime.now()
raised = False
with self.assertRaises(Exception):
decorator(function)()
_duration = (datetime.now() - start).seconds
Expand Down
4 changes: 2 additions & 2 deletions tests/unit/module_utils/test_elbv2.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ def test_get_elb_ip_address_type(self):
# Test modify_ip_address_type idempotency
def test_modify_ip_address_type_idempotency(self):
# Run module
return_value = self.elbv2obj.modify_ip_address_type("ipv4")
self.elbv2obj.modify_ip_address_type("ipv4")
# check that no method was called and this has been retrieved from elb attributes
self.connection.set_ip_address_type.assert_not_called()
# assert we got the expected value
Expand All @@ -187,7 +187,7 @@ def test_modify_ip_address_type_idempotency(self):
# Test modify_ip_address_type
def test_modify_ip_address_type_update(self):
# Run module
return_value = self.elbv2obj.modify_ip_address_type("dualstack")
self.elbv2obj.modify_ip_address_type("dualstack")
# check that no method was called and this has been retrieved from elb attributes
self.connection.set_ip_address_type.assert_called_once()
# assert we got the expected value
Expand Down
2 changes: 1 addition & 1 deletion tests/unit/module_utils/test_retries.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def test_no_failures(self):
def no_failures():
self.counter += 1

r = no_failures()
no_failures()
self.assertEqual(self.counter, 1)

def test_extend_boto3_failures(self):
Expand Down

0 comments on commit bc85d19

Please sign in to comment.