# -*- coding: utf-8 -*- from mongoengine import * from tests.utils import MongoDBTestCase class TestFloatField(MongoDBTestCase): def test_validation(self): """Ensure that URLFields validate urls 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_unicode_url_validation(self): """Ensure unicode URLs are validated properly.""" class Link(Document): url = URLField() link = Link() link.url = u'http://привет.com' # TODO fix URL validation - this *IS* a valid URL # For now we just want to make sure that the error message is correct with self.assertRaises(ValidationError) as ctx_err: link.validate() self.assertEqual(unicode(ctx_err.exception), u"ValidationError (Link:None) (Invalid URL: http://\u043f\u0440\u0438\u0432\u0435\u0442.com: ['url'])") def test_url_scheme_validation(self): """Ensure that URLFields validate urls with specific schemes properly. """ class Link(Document): url = URLField() class SchemeLink(Document): url = URLField(schemes=['ws', 'irc']) link = Link() link.url = 'ws://google.com' self.assertRaises(ValidationError, link.validate) scheme_link = SchemeLink() scheme_link.url = 'ws://google.com' scheme_link.validate() def test_underscore_allowed_in_domains_names(self): class Link(Document): url = URLField() link = Link() link.url = 'https://san_leandro-ca.geebo.com' link.validate()