exclude db table name

theonion · Jul 29, 2016 · f80b0a8 · f80b0a8
1 parent 19d63bf
commit f80b0a8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/bulbs/utils/test/__init__.py b/bulbs/utils/test/__init__.py
@@ -19,6 +19,10 @@
 import six
 
 from bulbs.content.models import Content
+from bulbs.super_features.utils import get_superfeature_model
+
+
+SUPERFEATURE_MODEL = get_superfeature_model()
 
 
 def make_content(*args, **kwargs):
@@ -31,7 +35,7 @@ def make_content(*args, **kwargs):
         models = indexable_registry.families[Content]
         model_keys = []
         for key in models.keys():
-            if key not in ['content_content', 'poll_poll', 'super_features_basesuperfeature']:
+            if key not in ['content_content', 'poll_poll', SUPERFEATURE_MODEL._meta.db_table]:
                 model_keys.append(key)
         key = random.choice(model_keys)
         klass = indexable_registry.all_models[key]