46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
import datetime
|
|
from unittest import skipIf
|
|
|
|
from django.test import TestCase
|
|
from django.utils import six
|
|
|
|
from .models import Article, BrokenArticle, InternationalArticle
|
|
|
|
|
|
class SimpleTests(TestCase):
|
|
|
|
@skipIf(six.PY3, "tests a __str__ method returning unicode under Python 2")
|
|
def test_basic(self):
|
|
a = Article.objects.create(
|
|
headline=b'Area man programs in Python',
|
|
pub_date=datetime.datetime(2005, 7, 28)
|
|
)
|
|
self.assertEqual(str(a), str('Area man programs in Python'))
|
|
self.assertEqual(repr(a), str('<Article: Area man programs in Python>'))
|
|
|
|
@skipIf(six.PY3, "tests Model's default __str__ method under Python 2")
|
|
def test_broken(self):
|
|
# Regression test for #19362.
|
|
a = BrokenArticle.objects.create(
|
|
headline='Girl wins €12.500 in lottery',
|
|
pub_date=datetime.datetime(2005, 7, 28)
|
|
)
|
|
six.assertRaisesRegex(self, RuntimeError, "Did you apply "
|
|
"@python_2_unicode_compatible without defining __str__\?", str, a)
|
|
|
|
def test_international(self):
|
|
a = InternationalArticle.objects.create(
|
|
headline='Girl wins €12.500 in lottery',
|
|
pub_date=datetime.datetime(2005, 7, 28)
|
|
)
|
|
if six.PY3:
|
|
self.assertEqual(str(a), 'Girl wins €12.500 in lottery')
|
|
else:
|
|
# On Python 2, the default str() output will be the UTF-8 encoded
|
|
# output of __unicode__() -- or __str__() when the
|
|
# python_2_unicode_compatible decorator is used.
|
|
self.assertEqual(str(a), b'Girl wins \xe2\x82\xac12.500 in lottery')
|