mirror of
https://github.com/Hopiu/django.git
synced 2026-05-04 21:54:48 +00:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from django.db import models
|
||
|
|
from django.test.testcases import SimpleTestCase
|
||
|
|
from django.utils import six
|
||
|
|
from django.utils.unittest import skipIf
|
||
|
|
|
||
|
|
from .models import CustomBaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class CustomBaseTest(SimpleTestCase):
|
||
|
|
|
||
|
|
@skipIf(six.PY3, 'test metaclass definition under Python 2')
|
||
|
|
def test_py2_custom_base(self):
|
||
|
|
"""
|
||
|
|
Make sure models.Model can be subclassed with a valid custom base
|
||
|
|
using __metaclass__
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
class MyModel(models.Model):
|
||
|
|
__metaclass__ = CustomBaseModel
|
||
|
|
except Exception:
|
||
|
|
self.fail("models.Model couldn't be subclassed with a valid "
|
||
|
|
"custom base using __metaclass__.")
|
||
|
|
|
||
|
|
def test_six_custom_base(self):
|
||
|
|
"""
|
||
|
|
Make sure models.Model can be subclassed with a valid custom base
|
||
|
|
using `six.with_metaclass`.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
class MyModel(six.with_metaclass(CustomBaseModel, models.Model)):
|
||
|
|
pass
|
||
|
|
except Exception:
|
||
|
|
self.fail("models.Model couldn't be subclassed with a valid "
|
||
|
|
"custom base using `six.with_metaclass`.")
|