Psharma/fix tests group2a

tests/admin_docs/models.py

@@ -41,7 +41,7 @@ class Person(models.Model):
     last_name = models.CharField(max_length=200, help_text="The person's last name")
     company = models.ForeignKey(Company, models.CASCADE, help_text="place of work")
     family = models.ForeignKey(Family, models.SET_NULL, related_name="+", null=True)
-    groups = models.ManyToManyField(Group, help_text="has membership")
+    groups = models.ManyToManyField("Group", help_text="has membership", through="PersonGroup")
 
     def _get_full_name(self):
         return "%s %s" % (self.first_name, self.last_name)
@@ -88,3 +88,12 @@ def get_status_count(self):
 
     def get_groups_list(self):
         return []
+
+
+class PersonGroup(models.Model):
+    person = models.ForeignKey(Person, on_delete=models.CASCADE)
+    group = models.ForeignKey(Group, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('person', 'group'),)
+        db_table = "admin_docs_person_group"

tests/async/models.py

@@ -12,4 +12,13 @@ class SimpleModel(models.Model):
 
 
 class ManyToManyModel(models.Model):
-    simples = models.ManyToManyField("SimpleModel")
+    simples = models.ManyToManyField("SimpleModel", through="ManyToManyModelSimpleModel")
+
+
+class ManyToManyModelSimpleModel(models.Model):
+    manytomanymodel = models.ForeignKey(ManyToManyModel, on_delete=models.CASCADE)
+    simplemodel = models.ForeignKey(SimpleModel, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('manytomanymodel', 'simplemodel'),)
+        db_table = "async_manytomanymodel_simplemodel"

tests/defer_regress/models.py

@@ -3,6 +3,7 @@
 """
 
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
 
 
 class Item(models.Model):
@@ -61,6 +62,7 @@ class SpecialFeature(models.Model):
 class OneToOneItem(models.Model):
     item = models.OneToOneField(Item, models.CASCADE, related_name="one_to_one_item")
     name = models.CharField(max_length=15)
+    objects = ModelStorageManager("REFERENCE")
 
 
 class ItemAndSimpleItem(models.Model):
@@ -79,14 +81,23 @@ class Location(models.Model):
 class Request(models.Model):
     profile = models.ForeignKey(Profile, models.SET_NULL, null=True, blank=True)
     location = models.ForeignKey(Location, models.CASCADE)
-    items = models.ManyToManyField(Item)
+    items = models.ManyToManyField("Item", through="RequestItem")
 
     request1 = models.CharField(default="request1", max_length=255)
     request2 = models.CharField(default="request2", max_length=255)
     request3 = models.CharField(default="request3", max_length=255)
     request4 = models.CharField(default="request4", max_length=255)
 
 
+class RequestItem(models.Model):
+    request = models.ForeignKey(Request, on_delete=models.CASCADE)
+    item = models.ForeignKey(Item, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('request', 'item'),)
+        db_table = "defer_regress_request_item"
+
+
 class Base(models.Model):
     text = models.TextField()
 

tests/generic_inline_admin/models.py

@@ -1,6 +1,7 @@
 from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
 from django.contrib.contenttypes.models import ContentType
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
 
 
 class Episode(models.Model):
@@ -39,6 +40,8 @@ class PhoneNumber(models.Model):
     phone_number = models.CharField(max_length=30)
     category = models.ForeignKey(Category, models.SET_NULL, null=True, blank=True)
 
+    objects = ModelStorageManager("ROWSTORE REFERENCE")
+
     class Meta:
         unique_together = (
             (

tests/get_object_or_404/models.py

@@ -28,8 +28,17 @@ def get_queryset(self):
 
 
 class Article(models.Model):
-    authors = models.ManyToManyField(Author)
+    authors = models.ManyToManyField("Author", through="ArticleAuthor")
     title = models.CharField(max_length=50)
     objects = models.Manager()
     by_a_sir = ArticleManager()
     attribute_error_objects = AttributeErrorManager()
+
+
+class ArticleAuthor(models.Model):
+    article = models.ForeignKey(Article, on_delete=models.CASCADE)
+    author = models.ForeignKey(Author, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('article', 'author'),)
+        db_table = "get_object_or_404_article_author"