Skip to content

Commit a3c81e7

Browse files
committed
Linting
Signed-off-by: rsuplina <rsuplina@cloudera.com>
1 parent 8188a17 commit a3c81e7

File tree

4 files changed

+102
-38
lines changed

4 files changed

+102
-38
lines changed

plugins/module_utils/cdp_client.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,11 @@ def _post(
660660
) -> Dict[str, Any]:
661661
"""Execute HTTP POST request."""
662662
return self._make_request(
663-
"POST", path, data=data, json_data=json_data, squelch=squelch
663+
"POST",
664+
path,
665+
data=data,
666+
json_data=json_data,
667+
squelch=squelch,
664668
)
665669

666670
def _put(
@@ -672,7 +676,11 @@ def _put(
672676
) -> Dict[str, Any]:
673677
"""Execute HTTP PUT request."""
674678
return self._make_request(
675-
"PUT", path, data=data, json_data=json_data, squelch=squelch
679+
"PUT",
680+
path,
681+
data=data,
682+
json_data=json_data,
683+
squelch=squelch,
676684
)
677685

678686
def _delete(self, path: str, squelch: Dict[int, Any] = {}) -> Dict[str, Any]:

plugins/module_utils/cdp_iam.py

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def get_group_details(self, group_name: str) -> Optional[Dict[str, Any]]:
7575

7676
# Get assigned resource roles
7777
resource_roles_response = self.list_group_assigned_resource_roles(
78-
group_name=group_name
78+
group_name=group_name,
7979
)
8080
resource_assignments = resource_roles_response.get("resourceAssignments", [])
8181

@@ -119,7 +119,8 @@ def manage_group_users(
119119
for user_crn in users_to_remove:
120120
if self._is_machine_user(user_crn):
121121
self.remove_machine_user_from_group(
122-
machine_user_name=user_crn, group_name=group_name
122+
machine_user_name=user_crn,
123+
group_name=group_name,
123124
)
124125
else:
125126
self.remove_user_from_group(user_id=user_crn, group_name=group_name)
@@ -130,7 +131,8 @@ def manage_group_users(
130131
for user_crn in users_to_add:
131132
if self._is_machine_user(user_crn):
132133
self.add_machine_user_to_group(
133-
machine_user_name=user_crn, group_name=group_name
134+
machine_user_name=user_crn,
135+
group_name=group_name,
134136
)
135137
else:
136138
self.add_user_to_group(user_id=user_crn, group_name=group_name)
@@ -837,7 +839,9 @@ def add_user_to_group(self, group_name: str, user_id: str) -> Dict[str, Any]:
837839
)
838840

839841
def add_machine_user_to_group(
840-
self, group_name: str, machine_user_name: str
842+
self,
843+
group_name: str,
844+
machine_user_name: str,
841845
) -> Dict[str, Any]:
842846
"""
843847
Add a machine user to a group.
@@ -881,7 +885,9 @@ def remove_user_from_group(self, group_name: str, user_id: str) -> Dict[str, Any
881885
)
882886

883887
def remove_machine_user_from_group(
884-
self, group_name: str, machine_user_name: str
888+
self,
889+
group_name: str,
890+
machine_user_name: str,
885891
) -> Dict[str, Any]:
886892
"""
887893
Remove a machine user from a group.
@@ -927,7 +933,10 @@ def assign_group_role(self, group_name: str, role: str) -> Dict[str, Any]:
927933
)
928934

929935
def assign_group_resource_role(
930-
self, group_name: str, resource_crn: str, resource_role_crn: str
936+
self,
937+
group_name: str,
938+
resource_crn: str,
939+
resource_role_crn: str,
931940
) -> Dict[str, Any]:
932941
"""
933942
Assign a resource role to a group.
@@ -973,7 +982,10 @@ def unassign_group_role(self, group_name: str, role: str) -> Dict[str, Any]:
973982
)
974983

975984
def unassign_group_resource_role(
976-
self, group_name: str, resource_crn: str, resource_role_crn: str
985+
self,
986+
group_name: str,
987+
resource_crn: str,
988+
resource_role_crn: str,
977989
) -> Dict[str, Any]:
978990
"""
979991
Unassign a resource role from a group.

plugins/modules/iam_group.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -246,10 +246,14 @@ def __init__(self):
246246
elements="dict",
247247
options=dict(
248248
resource=dict(
249-
required=True, type="str", aliases=["resource_crn"]
249+
required=True,
250+
type="str",
251+
aliases=["resource_crn"],
250252
),
251253
role=dict(
252-
required=True, type="str", aliases=["resource_role_crn"]
254+
required=True,
255+
type="str",
256+
aliases=["resource_role_crn"],
253257
),
254258
),
255259
),
@@ -289,7 +293,8 @@ def process(self):
289293
self.group = {"groupName": self.name}
290294
else:
291295
response = self.client.create_group(
292-
group_name=self.name, sync_membership_on_user_login=self.sync
296+
group_name=self.name,
297+
sync_membership_on_user_login=self.sync,
293298
)
294299
self.group = response.get("group", {})
295300
self.changed = True
@@ -299,7 +304,8 @@ def process(self):
299304

300305
if self.sync != current_group.get("syncMembershipOnUserLogin"):
301306
self.client.update_group(
302-
group_name=self.name, sync_membership_on_user_login=self.sync
307+
group_name=self.name,
308+
sync_membership_on_user_login=self.sync,
303309
)
304310
self.changed = True
305311

tests/unit/plugins/modules/iam_group/test_iam_group.py

Lines changed: 63 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -244,13 +244,14 @@ def test_create_group(self, mock_api_client, iam_client):
244244
"crn": "crn:cdp:iam:us-west-1:altus:group:new-team",
245245
"creationDate": "2024-02-01T09:00:00.000Z",
246246
"syncMembershipOnUserLogin": True,
247-
}
247+
},
248248
}
249249

250250
mock_api_client._post.return_value = mock_response
251251

252252
response = iam_client.create_group(
253-
group_name="new-team", sync_membership_on_user_login=True
253+
group_name="new-team",
254+
sync_membership_on_user_login=True,
254255
)
255256

256257
assert "group" in response
@@ -296,7 +297,8 @@ def test_update_group(self, mock_api_client, iam_client):
296297
mock_api_client._post.return_value = mock_response
297298

298299
response = iam_client.update_group(
299-
group_name="existing-team", sync_membership_on_user_login=False
300+
group_name="existing-team",
301+
sync_membership_on_user_login=False,
300302
)
301303

302304
assert isinstance(response, dict)
@@ -313,7 +315,11 @@ def test_update_group(self, mock_api_client, iam_client):
313315
)
314316

315317
def test_list_group_members(
316-
self, mock_api_client, iam_client, sample_users, sample_machine_users
318+
self,
319+
mock_api_client,
320+
iam_client,
321+
sample_users,
322+
sample_machine_users,
317323
):
318324
"""Test listing group members."""
319325

@@ -393,7 +399,10 @@ def test_remove_user_from_group(self, mock_api_client, iam_client, sample_users)
393399
)
394400

395401
def test_add_machine_user_to_group(
396-
self, mock_api_client, iam_client, sample_machine_users
402+
self,
403+
mock_api_client,
404+
iam_client,
405+
sample_machine_users,
397406
):
398407
"""Test adding a machine user to a group."""
399408

@@ -421,7 +430,10 @@ def test_add_machine_user_to_group(
421430
)
422431

423432
def test_remove_machine_user_from_group(
424-
self, mock_api_client, iam_client, sample_machine_users
433+
self,
434+
mock_api_client,
435+
iam_client,
436+
sample_machine_users,
425437
):
426438
"""Test removing a machine user from a group."""
427439

@@ -520,7 +532,7 @@ def test_list_group_assigned_resource_roles(self, mock_api_client, iam_client):
520532
mock_api_client._post.return_value = mock_response
521533

522534
response = iam_client.list_group_assigned_resource_roles(
523-
group_name="data-engineers"
535+
group_name="data-engineers",
524536
)
525537

526538
assert "resourceAssignments" in response
@@ -538,7 +550,10 @@ def test_list_group_assigned_resource_roles(self, mock_api_client, iam_client):
538550
)
539551

540552
def test_assign_group_resource_role(
541-
self, mock_api_client, iam_client, sample_resource_roles
553+
self,
554+
mock_api_client,
555+
iam_client,
556+
sample_resource_roles,
542557
):
543558
"""Test assigning a resource role to a group."""
544559

@@ -568,7 +583,10 @@ def test_assign_group_resource_role(
568583
)
569584

570585
def test_unassign_group_resource_role(
571-
self, mock_api_client, iam_client, sample_resource_roles
586+
self,
587+
mock_api_client,
588+
iam_client,
589+
sample_resource_roles,
572590
):
573591
"""Test unassigning a resource role from a group."""
574592

@@ -602,7 +620,10 @@ def test_unassign_group_resource_role(
602620
# ============================================================================
603621

604622
def test_manage_group_users_add_only(
605-
self, mock_api_client, iam_client, sample_users
623+
self,
624+
mock_api_client,
625+
iam_client,
626+
sample_users,
606627
):
607628
"""Test adding users to a group without removing any."""
608629
mock_api_client._post.return_value = {}
@@ -637,7 +658,10 @@ def test_manage_group_users_purge(self, mock_api_client, iam_client, sample_user
637658
assert changed is True
638659

639660
def test_manage_group_users_no_changes(
640-
self, mock_api_client, iam_client, sample_users
661+
self,
662+
mock_api_client,
663+
iam_client,
664+
sample_users,
641665
):
642666
"""Test when no user changes are needed."""
643667
current_members = sample_users
@@ -654,7 +678,10 @@ def test_manage_group_users_no_changes(
654678
mock_api_client._post.assert_not_called()
655679

656680
def test_manage_group_users_machine_user(
657-
self, mock_api_client, iam_client, sample_machine_users
681+
self,
682+
mock_api_client,
683+
iam_client,
684+
sample_machine_users,
658685
):
659686
"""Test adding machine users to a group."""
660687
mock_api_client._post.return_value = {}
@@ -686,7 +713,8 @@ def test_create_update_delete_group_lifecycle(self, api_client):
686713
try:
687714
# 1. Create a new group
688715
create_response = client.create_group(
689-
group_name=test_group_name, sync_membership_on_user_login=True
716+
group_name=test_group_name,
717+
sync_membership_on_user_login=True,
690718
)
691719

692720
assert "group" in create_response
@@ -701,7 +729,8 @@ def test_create_update_delete_group_lifecycle(self, api_client):
701729

702730
# 3. Update the group (change sync setting)
703731
update_response = client.update_group(
704-
group_name=test_group_name, sync_membership_on_user_login=False
732+
group_name=test_group_name,
733+
sync_membership_on_user_login=False,
705734
)
706735
assert isinstance(update_response, dict)
707736

@@ -731,7 +760,8 @@ def test_group_membership_management(self, api_client):
731760
try:
732761
# 1. Create a test group
733762
client.create_group(
734-
group_name=test_group_name, sync_membership_on_user_login=True
763+
group_name=test_group_name,
764+
sync_membership_on_user_login=True,
735765
)
736766

737767
# 2. Get current user to add to group
@@ -741,7 +771,8 @@ def test_group_membership_management(self, api_client):
741771
if user_crn:
742772
# 3. Add user to group
743773
add_response = client.add_user_to_group(
744-
group_name=test_group_name, user_id=user_crn
774+
group_name=test_group_name,
775+
user_id=user_crn,
745776
)
746777
assert isinstance(add_response, dict)
747778

@@ -752,7 +783,8 @@ def test_group_membership_management(self, api_client):
752783

753784
# 5. Remove user from group
754785
remove_response = client.remove_user_from_group(
755-
group_name=test_group_name, user_id=user_crn
786+
group_name=test_group_name,
787+
user_id=user_crn,
756788
)
757789
assert isinstance(remove_response, dict)
758790

@@ -777,7 +809,8 @@ def test_group_role_assignment(self, api_client):
777809
try:
778810
# 1. Create a test group
779811
client.create_group(
780-
group_name=test_group_name, sync_membership_on_user_login=True
812+
group_name=test_group_name,
813+
sync_membership_on_user_login=True,
781814
)
782815

783816
# 2. Get available roles
@@ -788,26 +821,28 @@ def test_group_role_assignment(self, api_client):
788821

789822
# 3. Assign role to group
790823
assign_response = client.assign_group_role(
791-
group_name=test_group_name, role=test_role_crn
824+
group_name=test_group_name,
825+
role=test_role_crn,
792826
)
793827
assert isinstance(assign_response, dict)
794828

795829
# 4. Verify role was assigned
796830
assigned_roles = client.list_group_assigned_roles(
797-
group_name=test_group_name
831+
group_name=test_group_name,
798832
)
799833
assert "roleCrns" in assigned_roles
800834
assert test_role_crn in assigned_roles["roleCrns"]
801835

802836
# 5. Unassign role from group
803837
unassign_response = client.unassign_group_role(
804-
group_name=test_group_name, role=test_role_crn
838+
group_name=test_group_name,
839+
role=test_role_crn,
805840
)
806841
assert isinstance(unassign_response, dict)
807842

808843
# 6. Verify role was unassigned
809844
roles_after = client.list_group_assigned_roles(
810-
group_name=test_group_name
845+
group_name=test_group_name,
811846
)
812847
assert test_role_crn not in roles_after.get("roleCrns", [])
813848

@@ -828,7 +863,8 @@ def test_machine_user_group_membership(self, api_client):
828863
try:
829864
# 1. Create a test group
830865
client.create_group(
831-
group_name=test_group_name, sync_membership_on_user_login=True
866+
group_name=test_group_name,
867+
sync_membership_on_user_login=True,
832868
)
833869

834870
# 2. Get list of machine users
@@ -845,7 +881,8 @@ def test_machine_user_group_membership(self, api_client):
845881

846882
# 3. Add machine user to group
847883
add_response = client.add_machine_user_to_group(
848-
group_name=test_group_name, machine_user_name=machine_user_name
884+
group_name=test_group_name,
885+
machine_user_name=machine_user_name,
849886
)
850887
assert isinstance(add_response, dict)
851888

@@ -855,7 +892,8 @@ def test_machine_user_group_membership(self, api_client):
855892

856893
# 5. Remove machine user from group
857894
remove_response = client.remove_machine_user_from_group(
858-
group_name=test_group_name, machine_user_name=machine_user_name
895+
group_name=test_group_name,
896+
machine_user_name=machine_user_name,
859897
)
860898
assert isinstance(remove_response, dict)
861899

0 commit comments

Comments
 (0)