# -*- coding: utf-8 -*- from __future__ import absolute_import, unicode_literals from django.forms import EmailField, IntegerField from django.http import HttpResponse from django.template.loader import render_to_string from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature from django.utils import six from django.utils.unittest import skip from .models import Person class SkippingTestCase(TestCase): def test_skip_unless_db_feature(self): "A test that might be skipped is actually called." # Total hack, but it works, just want an attribute that's always true. @skipUnlessDBFeature("__class__") def test_func(): raise ValueError self.assertRaises(ValueError, test_func) class AssertNumQueriesTests(TestCase): urls = 'regressiontests.test_utils.urls' def test_assert_num_queries(self): def test_func(): raise ValueError self.assertRaises(ValueError, self.assertNumQueries, 2, test_func ) def test_assert_num_queries_with_client(self): person = Person.objects.create(name='test') self.assertNumQueries( 1, self.client.get, "/test_utils/get_person/%s/" % person.pk ) self.assertNumQueries( 1, self.client.get, "/test_utils/get_person/%s/" % person.pk ) def test_func(): self.client.get("/test_utils/get_person/%s/" % person.pk) self.client.get("/test_utils/get_person/%s/" % person.pk) self.assertNumQueries(2, test_func) class AssertNumQueriesContextManagerTests(TestCase): urls = 'regressiontests.test_utils.urls' def test_simple(self): with self.assertNumQueries(0): pass with self.assertNumQueries(1): Person.objects.count() with self.assertNumQueries(2): Person.objects.count() Person.objects.count() def test_failure(self): with self.assertRaises(AssertionError) as exc_info: with self.assertNumQueries(2): Person.objects.count() self.assertIn("1 queries executed, 2 expected", str(exc_info.exception)) with self.assertRaises(TypeError): with self.assertNumQueries(4000): raise TypeError def test_with_client(self): person = Person.objects.create(name="test") with self.assertNumQueries(1): self.client.get("/test_utils/get_person/%s/" % person.pk) with self.assertNumQueries(1): self.client.get("/test_utils/get_person/%s/" % person.pk) with self.assertNumQueries(2): self.client.get("/test_utils/get_person/%s/" % person.pk) self.client.get("/test_utils/get_person/%s/" % person.pk) class AssertTemplateUsedContextManagerTests(TestCase): def test_usage(self): with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/base.html') with self.assertTemplateUsed(template_name='template_used/base.html'): render_to_string('template_used/base.html') with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/include.html') with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/extends.html') with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/base.html') render_to_string('template_used/base.html') def test_nested_usage(self): with self.assertTemplateUsed('template_used/base.html'): with self.assertTemplateUsed('template_used/include.html'): render_to_string('template_used/include.html') with self.assertTemplateUsed('template_used/extends.html'): with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/extends.html') with self.assertTemplateUsed('template_used/base.html'): with self.assertTemplateUsed('template_used/alternative.html'): render_to_string('template_used/alternative.html') render_to_string('template_used/base.html') with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/extends.html') with self.assertTemplateNotUsed('template_used/base.html'): render_to_string('template_used/alternative.html') render_to_string('template_used/base.html') def test_not_used(self): with self.assertTemplateNotUsed('template_used/base.html'): pass with self.assertTemplateNotUsed('template_used/alternative.html'): pass def test_error_message(self): with six.assertRaisesRegex(self, AssertionError, r'^template_used/base\.html'): with self.assertTemplateUsed('template_used/base.html'): pass with six.assertRaisesRegex(self, AssertionError, r'^template_used/base\.html'): with self.assertTemplateUsed(template_name='template_used/base.html'): pass with six.assertRaisesRegex(self, AssertionError, r'^template_used/base\.html.*template_used/alternative\.html$'): with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/alternative.html') def test_failure(self): with self.assertRaises(TypeError): with self.assertTemplateUsed(): pass with self.assertRaises(AssertionError): with self.assertTemplateUsed(''): pass with self.assertRaises(AssertionError): with self.assertTemplateUsed(''): render_to_string('template_used/base.html') with self.assertRaises(AssertionError): with self.assertTemplateUsed(template_name=''): pass with self.assertRaises(AssertionError): with self.assertTemplateUsed('template_used/base.html'): render_to_string('template_used/alternative.html') class SaveRestoreWarningState(TestCase): def test_save_restore_warnings_state(self): """ Ensure save_warnings_state/restore_warnings_state work correctly. """ # In reality this test could be satisfied by many broken implementations # of save_warnings_state/restore_warnings_state (e.g. just # warnings.resetwarnings()) , but it is difficult to test more. import warnings self.save_warnings_state() class MyWarning(Warning): pass # Add a filter that causes an exception to be thrown, so we can catch it warnings.simplefilter("error", MyWarning) self.assertRaises(Warning, lambda: warnings.warn("warn", MyWarning)) # Now restore. self.restore_warnings_state() # After restoring, we shouldn't get an exception. But we don't want a # warning printed either, so we have to silence the warning. warnings.simplefilter("ignore", MyWarning) warnings.warn("warn", MyWarning) # Remove the filter we just added. self.restore_warnings_state() class HTMLEqualTests(TestCase): def test_html_parser(self): from django.test.html import parse_html element = parse_html('

Hello

') self.assertEqual(len(element.children), 1) self.assertEqual(element.children[0].name, 'p') self.assertEqual(element.children[0].children[0], 'Hello') parse_html('

') parse_html('

') dom = parse_html('

foo') self.assertEqual(len(dom.children), 1) self.assertEqual(dom.name, 'p') self.assertEqual(dom[0], 'foo') def test_parse_html_in_script(self): from django.test.html import parse_html parse_html(''); parse_html(''' ''') # script content will be parsed to text dom = parse_html(''' ''') self.assertEqual(len(dom.children), 1) self.assertEqual(dom.children[0], "

foo

'' bar") def test_self_closing_tags(self): from django.test.html import parse_html self_closing_tags = ('br' , 'hr', 'input', 'img', 'meta', 'spacer', 'link', 'frame', 'base', 'col') for tag in self_closing_tags: dom = parse_html('

Hello <%s> world

' % tag) self.assertEqual(len(dom.children), 3) self.assertEqual(dom[0], 'Hello') self.assertEqual(dom[1].name, tag) self.assertEqual(dom[2], 'world') dom = parse_html('

Hello <%s /> world

' % tag) self.assertEqual(len(dom.children), 3) self.assertEqual(dom[0], 'Hello') self.assertEqual(dom[1].name, tag) self.assertEqual(dom[2], 'world') def test_simple_equal_html(self): self.assertHTMLEqual('', '') self.assertHTMLEqual('

', '

') self.assertHTMLEqual('

', '

') self.assertHTMLEqual( '

Hello

', '

Hello

') self.assertHTMLEqual( '

Hello

', '

Hello

') self.assertHTMLEqual( '
\n

Hello

', '

Hello

\n') self.assertHTMLEqual( '

Hello\nWorld !

', '

Hello World\n!

') self.assertHTMLEqual( '

Hello\nWorld !

', '

Hello World\n!

') self.assertHTMLEqual( '

Hello World !

', '

Hello World\n\n!

') self.assertHTMLEqual('

', '

') self.assertHTMLEqual('

', '

') self.assertHTMLEqual('

', '

') self.assertHTMLEqual('', '') self.assertHTMLEqual('

Hello', '

Hello') self.assertHTMLEqual('

Hello

World', '

Hello

World') def test_ignore_comments(self): self.assertHTMLEqual( '
Hello World!
', '
Hello World!
') def test_unequal_html(self): self.assertHTMLNotEqual('

Hello

', '

Hello!

') self.assertHTMLNotEqual('

foobar

', '

foo bar

') self.assertHTMLNotEqual('

foo bar

', '

foo  bar

') self.assertHTMLNotEqual('

foo nbsp

', '

foo  

') self.assertHTMLNotEqual('

foo #20

', '

foo 

') self.assertHTMLNotEqual( '

HelloWorld

', '

HelloWorld

') self.assertHTMLNotEqual( '

HelloWorld

', '

HelloWorld

') def test_attributes(self): self.assertHTMLEqual( '', '') self.assertHTMLEqual( '''''', '') self.assertHTMLNotEqual( '', '') def test_complex_examples(self): self.assertHTMLEqual( """ """, """ """) self.assertHTMLEqual( """ Document

This is a valid paragraph

this is a div AFTER the p
""", """ Document

This is a valid paragraph

this is a div AFTER the p

""") def test_html_contain(self): from django.test.html import parse_html # equal html contains each other dom1 = parse_html('

foo') dom2 = parse_html('

foo

') self.assertTrue(dom1 in dom2) self.assertTrue(dom2 in dom1) dom2 = parse_html('

foo

') self.assertTrue(dom1 in dom2) self.assertTrue(dom2 not in dom1) self.assertFalse('

foo

' in dom2) self.assertTrue('foo' in dom2) # when a root element is used ... dom1 = parse_html('

foo

bar

') dom2 = parse_html('

foo

bar

') self.assertTrue(dom1 in dom2) dom1 = parse_html('

foo

') self.assertTrue(dom1 in dom2) dom1 = parse_html('

bar

') self.assertTrue(dom1 in dom2) def test_count(self): from django.test.html import parse_html # equal html contains each other one time dom1 = parse_html('

foo') dom2 = parse_html('

foo

') self.assertEqual(dom1.count(dom2), 1) self.assertEqual(dom2.count(dom1), 1) dom2 = parse_html('

foo

bar

') self.assertEqual(dom2.count(dom1), 1) dom2 = parse_html('

foo foo

foo

') self.assertEqual(dom2.count('foo'), 3) dom2 = parse_html('

foo

') self.assertEqual(dom2.count('bar'), 0) self.assertEqual(dom2.count('class'), 0) self.assertEqual(dom2.count('p'), 0) self.assertEqual(dom2.count('o'), 2) dom2 = parse_html('

foo

foo

') self.assertEqual(dom2.count(dom1), 2) dom2 = parse_html('

foo

foo

') self.assertEqual(dom2.count(dom1), 1) dom2 = parse_html('

foo

') self.assertEqual(dom2.count(dom1), 1) dom2 = parse_html('

foo

foo

') self.assertEqual(dom2.count(dom1), 1) dom2 = parse_html('

foo

bar

') self.assertEqual(dom2.count(dom1), 0) def test_parsing_errors(self): from django.test.html import HTMLParseError, parse_html with self.assertRaises(AssertionError): self.assertHTMLEqual('

', '') with self.assertRaises(AssertionError): self.assertHTMLEqual('', '

') with self.assertRaises(HTMLParseError): parse_html('

') def test_contains_html(self): response = HttpResponse(''' This is a form:
''') self.assertNotContains(response, "") self.assertContains(response, '
') self.assertContains(response, "", html=True) self.assertNotContains(response, '', html=True) invalid_response = HttpResponse('''>''') with self.assertRaises(AssertionError): self.assertContains(invalid_response, '

') with self.assertRaises(AssertionError): self.assertContains(response, '

') def test_unicode_handling(self): from django.http import HttpResponse response = HttpResponse('

Some help text for the title (with unicode ŠĐĆŽćžšđ)

') self.assertContains(response, '

Some help text for the title (with unicode ŠĐĆŽćžšđ)

', html=True) class SkippingExtraTests(TestCase): fixtures = ['should_not_be_loaded.json'] # HACK: This depends on internals of our TestCase subclasses def __call__(self, result=None): # Detect fixture loading by counting SQL queries, should be zero with self.assertNumQueries(0): super(SkippingExtraTests, self).__call__(result) @skip("Fixture loading should not be performed for skipped tests.") def test_fixtures_are_skipped(self): pass class AssertRaisesMsgTest(SimpleTestCase): def test_special_re_chars(self): """assertRaisesMessage shouldn't interpret RE special chars.""" def func1(): raise ValueError("[.*x+]y?") self.assertRaisesMessage(ValueError, "[.*x+]y?", func1) class AssertFieldOutputTests(SimpleTestCase): def test_assert_field_output(self): error_invalid = ['Enter a valid e-mail address.'] self.assertFieldOutput(EmailField, {'a@a.com': 'a@a.com'}, {'aaa': error_invalid}) self.assertRaises(AssertionError, self.assertFieldOutput, EmailField, {'a@a.com': 'a@a.com'}, {'aaa': error_invalid + ['Another error']}) self.assertRaises(AssertionError, self.assertFieldOutput, EmailField, {'a@a.com': 'Wrong output'}, {'aaa': error_invalid}) self.assertRaises(AssertionError, self.assertFieldOutput, EmailField, {'a@a.com': 'a@a.com'}, {'aaa': ['Come on, gimme some well formatted data, dude.']}) def test_custom_required_message(self): class MyCustomField(IntegerField): default_error_messages = { 'required': 'This is really required.', } self.assertFieldOutput(MyCustomField, {}, {}, empty_value=None) __test__ = {"API_TEST": r""" # Some checks of the doctest output normalizer. # Standard doctests do fairly >>> import json >>> from django.utils.xmlutils import SimplerXMLGenerator >>> from django.utils.six import StringIO >>> def produce_json(): ... return json.dumps(['foo', {'bar': ('baz', None, 1.0, 2), 'whiz': 42}]) >>> def produce_xml(): ... stream = StringIO() ... xml = SimplerXMLGenerator(stream, encoding='utf-8') ... xml.startDocument() ... xml.startElement("foo", {"aaa" : "1.0", "bbb": "2.0"}) ... xml.startElement("bar", {"ccc" : "3.0"}) ... xml.characters("Hello") ... xml.endElement("bar") ... xml.startElement("whiz", {}) ... xml.characters("Goodbye") ... xml.endElement("whiz") ... xml.endElement("foo") ... xml.endDocument() ... return stream.getvalue() >>> def produce_xml_fragment(): ... stream = StringIO() ... xml = SimplerXMLGenerator(stream, encoding='utf-8') ... xml.startElement("foo", {"aaa": "1.0", "bbb": "2.0"}) ... xml.characters("Hello") ... xml.endElement("foo") ... xml.startElement("bar", {"ccc": "3.0", "ddd": "4.0"}) ... xml.endElement("bar") ... return stream.getvalue() # JSON output is normalized for field order, so it doesn't matter # which order json dictionary attributes are listed in output >>> produce_json() '["foo", {"bar": ["baz", null, 1.0, 2], "whiz": 42}]' >>> produce_json() '["foo", {"whiz": 42, "bar": ["baz", null, 1.0, 2]}]' # XML output is normalized for attribute order, so it doesn't matter # which order XML element attributes are listed in output >>> produce_xml() '\nHelloGoodbye' >>> produce_xml() '\nHelloGoodbye' >>> produce_xml_fragment() 'Hello' >>> produce_xml_fragment() 'Hello' """} if not six.PY3: __test__["API_TEST"] += """ >>> def produce_long(): ... return 42L >>> def produce_int(): ... return 42 # Long values are normalized and are comparable to normal integers ... >>> produce_long() 42 # ... and vice versa >>> produce_int() 42L """