catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Adding support for UPSERT'ing route53 records

Browse source
This commit is contained in:
Min-Young Wu 2016-04-12 10:37:01 -07:00
commit 826ba82f6e
3 changed files with 37 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
moto/route53/models.py
View file

@ -135,6 +135,16 @@ class FakeZone(object):
self.rrsets.append(record_set)
return record_set
def upsert_rrset(self, record_set):
new_rrset = RecordSet(record_set)
for i, rrset in enumerate(self.rrsets):
if rrset.name == new_rrset.name:
self.rrsets[i] = new_rrset
break
else:
self.rrsets.append(new_rrset)
return new_rrset
def delete_rrset_by_name(self, name):
self.rrsets = [record_set for record_set in self.rrsets if record_set.name != name]