Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cognito:user_status filter in list users (#4120)

Browse Source 
Co-authored-by: Łukasz Nowak <lukasz.nowak@idemia.com>
This commit is contained in:
Łukasz 2021-08-03 08:56:41 +02:00 committed by GitHub
parent 45fa825619
commit 16a4db1ce5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
moto/cognitoidp/responses.py
View File

@ -327,6 +327,11 @@ class CognitoIdpResponse(BaseResponse):
user_pool_id, limit=limit, pagination_token=token
)
if filt:
inherent_attributes = {
"cognito:user_status": lambda u: u.status,
"status": lambda u: u.enabled,
"username": lambda u: u.username,
}
name, value = filt.replace('"', "").replace(" ", "").split("=")
users = [
user
@ -336,6 +341,10 @@ class CognitoIdpResponse(BaseResponse):
for attr in user.attributes
if attr["Name"] == name and attr["Value"] == value
]
or (
name in inherent_attributes
and inherent_attributes[name](user) == value
)
]
response = {"Users": [user.to_json(extended=True) for user in users]}
if token:

7
tests/test_cognitoidp/test_cognitoidp.py
View File

@ -1221,6 +1221,13 @@ def test_list_users():
result["Users"].should.have.length_of(1)
result["Users"][0]["Username"].should.equal(username_bis)
# checking Filter for inherent attributes
result = conn.list_users(
UserPoolId=user_pool_id, Filter='username = "{}"'.format(username_bis)
)
result["Users"].should.have.length_of(1)
result["Users"][0]["Username"].should.equal(username_bis)
@mock_cognitoidp
def test_get_user_unconfirmed():