Psharma/fix tests group3a

tests/inline_formsets/models.py

@@ -1,4 +1,5 @@
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
 
 
 class School(models.Model):
@@ -27,6 +28,8 @@ class Poem(models.Model):
     poet = models.ForeignKey(Poet, models.CASCADE)
     name = models.CharField(max_length=100)
 
+    objects = ModelStorageManager("ROWSTORE REFERENCE")
+
     class Meta:
         unique_together = ("poet", "name")
 

tests/m2m_and_m2o/models.py

@@ -12,7 +12,7 @@ class User(models.Model):
 
 class Issue(models.Model):
     num = models.IntegerField()
-    cc = models.ManyToManyField(User, blank=True, related_name="test_issue_cc")
+    cc = models.ManyToManyField("User", blank=True, related_name="test_issue_cc", through="IssueUser")
     client = models.ForeignKey(User, models.CASCADE, related_name="test_issue_client")
 
     class Meta:
@@ -22,5 +22,23 @@ def __str__(self):
         return str(self.num)
 
 
+class IssueUser(models.Model):
+    issue = models.ForeignKey(Issue, on_delete=models.CASCADE)
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('issue', 'user'),)
+        db_table = "m2m_and_m2o_issue_user"
+
+
 class StringReferenceModel(models.Model):
-    others = models.ManyToManyField("StringReferenceModel")
+    others = models.ManyToManyField("StringReferenceModel", through="StringReferenceModelStringReferenceModel")
+
+
+class StringReferenceModelStringReferenceModel(models.Model):
+    from_stringreferencemodel = models.ForeignKey(StringReferenceModel, on_delete=models.CASCADE, related_name="from_stringreferencemodel")
+    to_stringreferencemodel = models.ForeignKey(StringReferenceModel, on_delete=models.CASCADE, related_name="to_stringreferencemodel")
+
+    class Meta:
+        unique_together = (('from_stringreferencemodel', 'to_stringreferencemodel'),)
+        db_table = "m2m_and_m2o_stringreferencemodel_stringreferencemodel"

tests/many_to_one/models.py

@@ -4,6 +4,7 @@
 To define a many-to-one relationship, use ``ForeignKey()``.
 """
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
 
 
 class Reporter(models.Model):
@@ -72,6 +73,7 @@ class Parent(models.Model):
     bestchild = models.ForeignKey(
         "Child", models.SET_NULL, null=True, related_name="favored_by"
     )
+    objects = ModelStorageManager("REFERENCE")
 
 
 class ParentStringPrimaryKey(models.Model):

tests/model_package/models/article.py

@@ -6,6 +6,24 @@ class Site(models.Model):
 
 
 class Article(models.Model):
-    sites = models.ManyToManyField(Site)
+    sites = models.ManyToManyField("Site", through="ArticleSite")
     headline = models.CharField(max_length=100)
-    publications = models.ManyToManyField("model_package.Publication", blank=True)
+    publications = models.ManyToManyField("model_package.Publication", blank=True, through="Articlemodel_package")
+
+
+class ArticleSite(models.Model):
+    article = models.ForeignKey(Article, on_delete=models.CASCADE)
+    site = models.ForeignKey(Site, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('article', 'site'),)
+        db_table = "model_package_article_site"
+
+
+class Articlemodel_package(models.Model):
+    article = models.ForeignKey(Article, on_delete=models.CASCADE)
+    publication = models.ForeignKey("model_package.Publication", on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('article', 'publication'),)
+        db_table = "model_package_article_publications"

tests/model_package/tests.py

@@ -8,7 +8,16 @@
 
 class Advertisement(models.Model):
     customer = models.CharField(max_length=100)
-    publications = models.ManyToManyField("model_package.Publication", blank=True)
+    publications = models.ManyToManyField("model_package.Publication", blank=True, through="Advertisementmodel_package")
+
+
+class Advertisementmodel_package(models.Model):
+    advertisement = models.ForeignKey(Advertisement, on_delete=models.CASCADE)
+    publication = models.ForeignKey("model_package.Publication", on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('advertisement', 'publication'),)
+        db_table = "model_package_advertisement_publications"
 
 
 class ModelPackageTests(TestCase):

tests/select_related/models.py

@@ -68,7 +68,16 @@ class Topping(models.Model):
 
 class Pizza(models.Model):
     name = models.CharField(max_length=100)
-    toppings = models.ManyToManyField(Topping)
+    toppings = models.ManyToManyField("Topping", through="PizzaTopping")
+
+
+class PizzaTopping(models.Model):
+    pizza = models.ForeignKey(Pizza, on_delete=models.CASCADE)
+    topping = models.ForeignKey(Topping, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = (('pizza', 'topping'),)
+        db_table = "select_related_pizza_topping"
 
 
 class TaggedItem(models.Model):

tests/select_related_onetoone/models.py

@@ -1,4 +1,5 @@
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
 
 
 class User(models.Model):
@@ -10,6 +11,7 @@ class UserProfile(models.Model):
     user = models.OneToOneField(User, models.CASCADE)
     city = models.CharField(max_length=100)
     state = models.CharField(max_length=2)
+    objects = ModelStorageManager("REFERENCE")
 
 
 class UserStatResult(models.Model):
@@ -20,11 +22,15 @@ class UserStat(models.Model):
     user = models.OneToOneField(User, models.CASCADE, primary_key=True)
     posts = models.IntegerField()
     results = models.ForeignKey(UserStatResult, models.CASCADE)
+    objects = ModelStorageManager("REFERENCE")
+
 
 
 class StatDetails(models.Model):
     base_stats = models.OneToOneField(UserStat, models.CASCADE)
     comments = models.IntegerField()
+    objects = ModelStorageManager("REFERENCE")
+
 
 
 class AdvancedUserStat(UserStat):
@@ -38,6 +44,8 @@ class Image(models.Model):
 class Product(models.Model):
     name = models.CharField(max_length=100)
     image = models.OneToOneField(Image, models.SET_NULL, null=True)
+    objects = ModelStorageManager("REFERENCE")
+
 
 
 class Parent1(models.Model):
@@ -52,11 +60,14 @@ class Parent2(models.Model):
 
 class Child1(Parent1, Parent2):
     value = models.IntegerField()
+    objects = ModelStorageManager("REFERENCE")
 
 
 class Child2(Parent1):
     parent2 = models.OneToOneField(Parent2, models.CASCADE)
     value = models.IntegerField()
+    objects = ModelStorageManager("REFERENCE")
+
 
 
 class Child3(Child2):
@@ -76,3 +87,4 @@ class LinkedList(models.Model):
         blank=True,
         null=True,
     )
+    objects = ModelStorageManager("REFERENCE")

tests/string_lookup/models.py

@@ -1,5 +1,5 @@
 from django.db import models
-
+from django_singlestore.schema import ModelStorageManager
 
 class Foo(models.Model):
     name = models.CharField(max_length=50)
@@ -20,6 +20,7 @@ class Whiz(models.Model):
 class Child(models.Model):
     parent = models.OneToOneField("Base", models.CASCADE)
     name = models.CharField(max_length=50)
+    objects = ModelStorageManager("REFERENCE")
 
 
 class Base(models.Model):

tests/test_client_regress/models.py

@@ -1,5 +1,6 @@
 from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
 
 
 class CustomUser(AbstractBaseUser):
@@ -8,5 +9,7 @@ class CustomUser(AbstractBaseUser):
 
     USERNAME_FIELD = "email"
 
+    objects = ModelStorageManager("REFERENCE")
+
     class Meta:
         app_label = "test_client_regress"