more polymorphic-related tests

Author: smcoll

polymorphic_tree/tests/test_models.py

@@ -4,7 +4,7 @@
 from django.db.models import Q
 from django.test import TestCase
 
-from polymorphic import ShowFieldContent, ShowFieldType, ShowFieldTypeAndContent
+from .utils import ShowFieldType
 
 from polymorphic_tree.models import PolymorphicMPTTModel, PolymorphicTreeForeignKey
 
@@ -42,9 +42,44 @@ class One2OneRelatingModelDerived(One2OneRelatingModel):
  field2 = models.CharField(max_length=10)
 
 
+class Base(ShowFieldType, PolymorphicMPTTModel):
+ parent = PolymorphicTreeForeignKey('self', blank=True, null=True, related_name='children', verbose_name='parent')
+ field_b = models.CharField(max_length=10)
+
+class ModelX(Base):
+ field_x = models.CharField(max_length=10)
+
+class ModelY(Base):
+ field_y = models.CharField(max_length=10)
+
+
 class PolymorphicTreeTests(TestCase):
  """
  Test Suite, largely derived from django-polymorphic tests
+
+ TODO: potentially port these tests from django_polymorphic.tests
+
+ test_foreignkey_field()
+ test_onetoone_field()
+ test_manytomany_field()
+ test_extra_method()
+ test_instance_of_filter()
+ test_polymorphic___filter()
+ test_delete()
+ test_combine_querysets()
+ test_multiple_inheritance()
+ test_relation_base()
+ test_user_defined_manager()
+ test_manager_inheritance()
+ test_queryset_assignment()
+ test_proxy_models()
+ test_proxy_get_real_instance_class()
+ test_content_types_for_proxy_models()
+ test_proxy_model_inheritance()
+ test_custom_pk()
+ test_fix_getattribute()
+ test_parent_link_and_related_name()
+
  """
 
  def create_model2abcd(self):
@@ -124,3 +159,26 @@ def show_base_manager(model):
 
  self.assertEqual(show_base_manager(One2OneRelatingModel), "<class 'polymorphic_tree.managers.PolymorphicMPTTModelManager'> <class 'polymorphic_tree.tests.test_models.One2OneRelatingModel'>")
  self.assertEqual(show_base_manager(One2OneRelatingModelDerived), "<class 'django.db.models.manager.Manager'> <class 'polymorphic_tree.tests.test_models.One2OneRelatingModelDerived'>")
+
+ def test_instance_default_manager(self):
+ def show_default_manager(instance):
+ return "{0} {1}".format(
+ repr(type(instance._default_manager)),
+ repr(instance._default_manager.model)
+ )
+
+ plain_a = PlainA(field1='C1')
+ plain_b = PlainB(field2='C1')
+ plain_c = PlainC(field3='C1')
+
+ model_2a = Model2A(field1='C1')
+ model_2b = Model2B(field2='C1')
+ model_2c = Model2C(field3='C1')
+
+ self.assertEqual(show_default_manager(plain_a), "<class 'django.db.models.manager.Manager'> <class 'polymorphic_tree.tests.test_models.PlainA'>")
+ self.assertEqual(show_default_manager(plain_b), "<class 'django.db.models.manager.Manager'> <class 'polymorphic_tree.tests.test_models.PlainB'>")
+ self.assertEqual(show_default_manager(plain_c), "<class 'django.db.models.manager.Manager'> <class 'polymorphic_tree.tests.test_models.PlainC'>")
+
+ self.assertEqual(show_default_manager(model_2a), "<class 'polymorphic_tree.managers.PolymorphicMPTTModelManager'> <class 'polymorphic_tree.tests.test_models.Model2A'>")
+ self.assertEqual(show_default_manager(model_2b), "<class 'polymorphic_tree.managers.PolymorphicMPTTModelManager'> <class 'polymorphic_tree.tests.test_models.Model2B'>")
+ self.assertEqual(show_default_manager(model_2c), "<class 'polymorphic_tree.managers.PolymorphicMPTTModelManager'> <class 'polymorphic_tree.tests.test_models.Model2C'>")

polymorphic_tree/tests/utils.py

@@ -0,0 +1,167 @@
+# -*- coding: utf-8 -*-
+""" from https://github.com/chrisglass/django_polymorphic/blob/master/polymorphic/showfields.py
+"""
+
+from django.db import models
+from django.utils import six
+
+
+class ShowFieldBase(object):
+ """ base class for the ShowField... model mixins, does the work """
+
+ polymorphic_query_multiline_output = True # cause nicer multiline PolymorphicQuery output
+
+ polymorphic_showfield_type = False
+ polymorphic_showfield_content = False
+
+ # these may be overridden by the user
+ polymorphic_showfield_max_line_width = None
+ polymorphic_showfield_max_field_width = 20
+ polymorphic_showfield_old_format = False
+
+ def __repr__(self):
+ return self.__unicode__()
+
+ def _showfields_get_content(self, field_name, field_type=type(None)):
+ "helper for __unicode__"
+ content = getattr(self, field_name)
+ if self.polymorphic_showfield_old_format:
+ out = ': '
+ else:
+ out = ' '
+ if issubclass(field_type, models.ForeignKey):
+ if content is None:
+ out += 'None'
+ else:
+ out += content.__class__.__name__
+ elif issubclass(field_type, models.ManyToManyField):
+ out += '%d' % content.count()
+ elif isinstance(content, six.integer_types):
+ out += str(content)
+ elif content is None:
+ out += 'None'
+ else:
+ txt = str(content)
+ if len(txt) > self.polymorphic_showfield_max_field_width:
+ txt = txt[:self.polymorphic_showfield_max_field_width - 2] + '..'
+ out += '"' + txt + '"'
+ return out
+
+ def _showfields_add_regular_fields(self, parts):
+ "helper for __unicode__"
+ done_fields = set()
+ for field in self._meta.fields + self._meta.many_to_many:
+ if field.name in self.polymorphic_internal_model_fields or '_ptr' in field.name:
+ continue
+ if field.name in done_fields:
+ continue # work around django diamond inheritance problem
+ done_fields.add(field.name)
+
+ out = field.name
+
+ # if this is the standard primary key named "id", print it as we did with older versions of django_polymorphic
+ if field.primary_key and field.name == 'id' and type(field) == models.AutoField:
+ out += ' ' + str(getattr(self, field.name))
+
+ # otherwise, display it just like all other fields (with correct type, shortened content etc.)
+ else:
+ if self.polymorphic_showfield_type:
+ out += ' (' + type(field).__name__
+ if field.primary_key:
+ out += '/pk'
+ out += ')'
+
+ if self.polymorphic_showfield_content:
+ out += self._showfields_get_content(field.name, type(field))
+
+ parts.append((False, out, ','))
+
+ def _showfields_add_dynamic_fields(self, field_list, title, parts):
+ "helper for __unicode__"
+ parts.append((True, '- ' + title, ':'))
+ for field_name in field_list:
+ out = field_name
+ content = getattr(self, field_name)
+ if self.polymorphic_showfield_type:
+ out += ' (' + type(content).__name__ + ')'
+ if self.polymorphic_showfield_content:
+ out += self._showfields_get_content(field_name)
+
+ parts.append((False, out, ','))
+
+ def __unicode__(self):
+ # create list ("parts") containing one tuple for each title/field:
+ # ( bool: new section , item-text , separator to use after item )
+
+ # start with model name
+ parts = [(True, self.__class__.__name__, ':')]
+
+ # add all regular fields
+ self._showfields_add_regular_fields(parts)
+
+ # add annotate fields
+ if hasattr(self, 'polymorphic_annotate_names'):
+ self._showfields_add_dynamic_fields(self.polymorphic_annotate_names, 'Ann', parts)
+
+ # add extra() select fields
+ if hasattr(self, 'polymorphic_extra_select_names'):
+ self._showfields_add_dynamic_fields(self.polymorphic_extra_select_names, 'Extra', parts)
+
+ # format result
+
+ indent = len(self.__class__.__name__) + 5
+ indentstr = ''.rjust(indent)
+ out = ''
+ xpos = 0
+ possible_line_break_pos = None
+
+ for i in range(len(parts)):
+ new_section, p, separator = parts[i]
+ final = (i == len(parts) - 1)
+ if not final:
+ next_new_section, _, _ = parts[i + 1]
+
+ if (self.polymorphic_showfield_max_line_width
+ and xpos + len(p) > self.polymorphic_showfield_max_line_width
+ and possible_line_break_pos != None):
+ rest = out[possible_line_break_pos:]
+ out = out[:possible_line_break_pos]
+ out += '\n' + indentstr + rest
+ xpos = indent + len(rest)
+
+ out += p
+ xpos += len(p)
+
+ if not final:
+ if not next_new_section:
+ out += separator
+ xpos += len(separator)
+ out += ' '
+ xpos += 1
+
+ if not new_section:
+ possible_line_break_pos = len(out)
+
+ return '<' + out + '>'
+
+
+class ShowFieldType(ShowFieldBase):
+ """ model mixin that shows the object's class and it's field types """
+ polymorphic_showfield_type = True
+
+
+class ShowFieldContent(ShowFieldBase):
+ """ model mixin that shows the object's class, it's fields and field contents """
+ polymorphic_showfield_content = True
+
+
+class ShowFieldTypeAndContent(ShowFieldBase):
+ """ model mixin, like ShowFieldContent, but also show field types """
+ polymorphic_showfield_type = True
+ polymorphic_showfield_content = True
+
+
+# compatibility with old class names
+ShowFieldTypes = ShowFieldType
+ShowFields = ShowFieldContent
+ShowFieldsAndTypes = ShowFieldTypeAndContent