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

Allow spaces to if_not_exists

Browse Source
This commit is contained in:
Kosei Kitahara 2018-10-11 18:21:53 +09:00
parent dfa7935e13
commit 2c15d71c2c
No known key found for this signature in database
GPG Key ID: D18883191F02241B
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/dynamodb2/models.py
View File

@ -154,7 +154,7 @@ class Item(BaseModel):
# If not exists, changes value to a default if needed, else its the same as it was # If not exists, changes value to a default if needed, else its the same as it was
if value.startswith('if_not_exists'): if value.startswith('if_not_exists'):
# Function signature # Function signature
match = re.match(r'.*if_not_exists\((?P<path>.+),\s*(?P<default>.+)\).*', value) match = re.match(r'.*if_not_exists\s*\((?P<path>.+),\s*(?P<default>.+)\).*', value)
if not match: if not match:
raise TypeError raise TypeError

2
tests/test_dynamodb2/test_dynamodb.py
View File

@ -1220,6 +1220,7 @@ def test_update_if_not_exists():
'forum_name': 'the-key', 'forum_name': 'the-key',
'subject': '123' 'subject': '123'
}, },
# if_not_exists without space
UpdateExpression='SET created_at=if_not_exists(created_at,:created_at)', UpdateExpression='SET created_at=if_not_exists(created_at,:created_at)',
ExpressionAttributeValues={ ExpressionAttributeValues={
':created_at': 123 ':created_at': 123
@ -1233,6 +1234,7 @@ def test_update_if_not_exists():
'forum_name': 'the-key', 'forum_name': 'the-key',
'subject': '123' 'subject': '123'
}, },
# if_not_exists with space
UpdateExpression='SET created_at = if_not_exists (created_at, :created_at)', UpdateExpression='SET created_at = if_not_exists (created_at, :created_at)',
ExpressionAttributeValues={ ExpressionAttributeValues={
':created_at': 456 ':created_at': 456