test cascade

zuri-training · Nov 14, 2017 · 65b757a · 65b757a
1 parent 805f0da
commit 65b757a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/models/place.py b/models/place.py
@@ -10,9 +10,13 @@
 if models.storage_t == 'db':
     place_amenity = Table('place_amenity', Base.metadata,
                           Column('place_id', String(60),
-                                 ForeignKey('places.id'), primary_key=True),
+                                 ForeignKey('places.id', onupdate= 'CASCADE',
+                                            ondelete='CASCADE'),
+                                 primary_key=True),
                           Column('amenity_id', String(60),
-                                 ForeignKey('amenities.id'), primary_key=True))
+                                 ForeignKey('amenities.id', onupdate= 'CASCADE',
+                                            ondelete='CASCADE'),
+                                 primary_key=True))
 
 
 class Place(BaseModel, Base):