Add test to expose problem with DeclarativeColumnsMetaclass inheritence

bradleyayers · bradleyayers · commit b2358b14cc93 · 2012-03-08T13:39:48.000+10:00
diff --git a/tests/core.py b/tests/core.py
@@ -6,6 +6,7 @@
 from django.core.paginator import Paginator
 from django.http import Http404
 import django_tables2 as tables
+from django_tables2.tables import DeclarativeColumnsMetaclass
 from haystack.query import SearchQuerySet
 
 
@@ -58,6 +59,38 @@ class CityTable(GeoAreaTable, AddedMixin):
     assert 'added' in CityTable.base_columns
 
 
+@core.test
+def metaclass_inheritance():
+    class Tweaker(type):
+        """Adds an attribute "tweaked" to all classes"""
+        def __new__(cls, name, bases, attrs):
+            attrs['tweaked'] = True
+            return super(Tweaker, cls).__new__(cls, name, bases, attrs)
+
+    class Meta(Tweaker, DeclarativeColumnsMetaclass):
+        pass
+
+    class TweakedTable(tables.Table):
+        __metaclass__ = Meta
+        name = tables.Column()
+
+    table = TweakedTable([])
+    assert 'name' in table.columns
+    assert table.tweaked
+
+    # now flip the order
+    class FlippedMeta(DeclarativeColumnsMetaclass, Tweaker):
+        pass
+
+    class FlippedTweakedTable(tables.Table):
+        __metaclass__ = FlippedMeta
+        name = tables.Column()
+
+    table = FlippedTweakedTable([])
+    assert 'name' in table.columns
+    assert table.tweaked
+
+
 @core.test
 def attrs():
     class TestTable(tables.Table):
