diff --git a/src/webutils.py b/src/webutils.py index 3398c1573..61a8e587d 100644 --- a/src/webutils.py +++ b/src/webutils.py @@ -36,6 +36,7 @@ import fix import re import socket import urllib2 +import urlparse class WebError(Exception): pass @@ -75,6 +76,9 @@ def getUrl(url, size=None): fd.close() return text +def getDomain(url): + return urlparse.urlparse(url)[1] + # vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: diff --git a/test/test_webutils.py b/test/test_webutils.py new file mode 100644 index 000000000..b26d242ca --- /dev/null +++ b/test/test_webutils.py @@ -0,0 +1,48 @@ +#!/usr/bin/env python + +### +# Copyright (c) 2004, Jeremiah Fincher +# All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are met: +# +# * Redistributions of source code must retain the above copyright notice, +# this list of conditions, and the following disclaimer. +# * Redistributions in binary form must reproduce the above copyright notice, +# this list of conditions, and the following disclaimer in the +# documentation and/or other materials provided with the distribution. +# * Neither the name of the author of this software nor the name of +# contributors to this software may be used to endorse or promote products +# derived from this software without specific prior written consent. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" +# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE +# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF +# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS +# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN +# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) +# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE +# POSSIBILITY OF SUCH DAMAGE. +### + +from testsupport import * + +import webutils + + +class WebutilsTestCase(unittest.TestCase): + def testGetDomain(self): + self.assertEqual(webutils.getDomain('http://slashdot.org/foo/bar.exe'), + 'slashdot.org') + + def testGetUrlWithSize(self): + url = 'http://slashdot.org/' + self.failUnless(len(webutils.getUrl(url, 1024)) == 1024) + + +# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: +