From f5406ad212953d07d70f8d56cb4ae62c43007697 Mon Sep 17 00:00:00 2001 From: Paul Craciunoiu Date: Mon, 4 Jan 2016 16:29:02 -0700 Subject: [PATCH] Handle delete updates. --- moto/dynamodb2/models.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/moto/dynamodb2/models.py b/moto/dynamodb2/models.py index 5aa870b48..94f726bc2 100644 --- a/moto/dynamodb2/models.py +++ b/moto/dynamodb2/models.py @@ -124,6 +124,9 @@ class Item(object): def update_with_attribute_updates(self, attribute_updates): for attribute_name, update_action in attribute_updates.items(): action = update_action['Action'] + if action == 'DELETE' and not 'Value' in update_action: + del self.attrs[attribute_name] + continue new_value = list(update_action['Value'].values())[0] if action == 'PUT': # TODO deal with other types @@ -133,6 +136,8 @@ class Item(object): self.attrs[attribute_name] = DynamoType({"M": new_value}) elif update_action['Value'].keys() == ['N']: self.attrs[attribute_name] = DynamoType({"N": new_value}) + elif update_action['Value'].keys() == ['NULL']: + del self.attrs[attribute_name] else: self.attrs[attribute_name] = DynamoType({"S": new_value})