Add regression for delete tests

linkyndy · Jun 4, 2019 · 1c54495 · 1c54495
1 parent 0be7588
commit 1c54495
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -353,9 +353,28 @@ def setUp(self):
         super(DeleteTests, self).setUp()
 
         class Artist(Model):
-            pass
+            belongs_to = ('Label',)
+            has_one = ('TrueFan',)
+            has_many = ('Fan',)
+            has_and_belongs_to_many = ('Band',)
         self.Artist = Artist
 
+        class Label(Model):
+            pass
+        self.Label = Label
+
+        class TrueFan(Model):
+            pass
+        self.TrueFan = TrueFan
+
+        class Fan(Model):
+            pass
+        self.Fan = Fan
+
+        class Band(Model):
+            pass
+        self.Band = Band
+
         create_tables()
         create_indexes()