Allow doing an ADD update of a string set

Fix test
2017-09-11 12:06:24 -07:00 · 2017-09-11 12:06:24 -07:00 · 386ac94abe
commit 386ac94abe
parent 1d600011c5
2 changed files with 36 additions and 0 deletions
--- a/moto/dynamodb2/models.py
+++ b/moto/dynamodb2/models.py
@ -171,6 +171,12 @@ class Item(BaseModel):
                        decimal.Decimal(existing.value) +
                        decimal.Decimal(new_value)
                    )})
+                elif set(update_action['Value'].keys()) == set(['SS']):
+                    existing = self.attrs.get(attribute_name, DynamoType({"SS": {}}))
+                    new_set = set(existing.value).union(set(new_value))
+                    self.attrs[attribute_name] = DynamoType({
+                        "SS": list(new_set)
+                    })
                else:
                    # TODO: implement other data types
                    raise NotImplementedError(