b47669403b
This commit: 1. Formats all of our existing code using `black`. 2. Adds a note about using `black` to `CONTRIBUTING.rst`. 3. Runs `black --check` as part of CI (failing builds that aren't properly formatted).
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
from mongoengine import *
|
|
|
|
from tests.utils import MongoDBTestCase
|
|
|
|
|
|
class TestURLField(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()
|