From 3534daf37d4d721740e115b8b6425834b8e7943d Mon Sep 17 00:00:00 2001 From: Stefan Wojcik Date: Fri, 24 Feb 2017 16:15:03 -0500 Subject: [PATCH] fix tests structure --- tests/fields/fields.py | 33 ++++++++++++++++++--------------- 1 file changed, 18 insertions(+), 15 deletions(-) diff --git a/tests/fields/fields.py b/tests/fields/fields.py index d900d706..318c0c59 100644 --- a/tests/fields/fields.py +++ b/tests/fields/fields.py @@ -341,15 +341,30 @@ class FieldTest(MongoDBTestCase): person.validate() def test_url_validation(self): - """Ensure that URLFields validate urls properly. - """ + """Ensure that URLFields validate urls properly.""" class Link(Document): url = URLField() + Link.drop_collection() + + link = Link() + link.url = 'google' + self.assertRaises(ValidationError, link.validate) + + link.url = 'http://www.google.com:8080' + link.validate() + + def test_unicode_url_validation(self): + """Ensure unicode URLs are validated properly.""" + class Link(Document): + url = URLField() + + Link.drop_collection() + link = Link() link.url = u'http://привет.com' - # TODO fix URL validation - this IS a valid URL + # TODO fix URL validation - this *IS* a valid URL # For now we just want to make sure that the error message is correct try: link.validate() @@ -360,18 +375,6 @@ class FieldTest(MongoDBTestCase): u"ValidationError (Link:None) (Invalid URL: http://\u043f\u0440\u0438\u0432\u0435\u0442.com: ['url'])" ) - def test_unicode_url_validation(self): - """Ensure unicode URLs are validated properly.""" - class Link(Document): - url = URLField() - - link = Link() - link.url = 'google' - self.assertRaises(ValidationError, link.validate) - - link.url = 'http://www.google.com:8080' - link.validate() - def test_url_scheme_validation(self): """Ensure that URLFields validate urls with specific schemes properly. """