Switch to set method for modifying manytomany fields

jazzband · Dec 22, 2017 · 1b7ec90 · 1b7ec90
1 parent c16b1ae
commit 1b7ec90
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/project/tests/test_lib/mock_suite.py b/project/tests/test_lib/mock_suite.py
@@ -5,6 +5,7 @@
 import random
 import traceback
 
+import django
 from django.core import management
 from django.utils import timezone
 
@@ -100,7 +101,10 @@ def mock_sql_queries(self, request=None, profile=None, n=1, as_dict=False):
                 for q in queries:
                     profile['queries'].append(q)
             else:
-                profile.queries = queries
+                if django.VERSION >= (1, 9):
+                    profile.queries.set(queries)
+                else:
+                    profile.queries = queries
                 profile.save()
         return queries