class PortTest(TestCase): def setUp(self): self.urlValidator = UrlValidator() # input: correct input # expected: non-None def test_basicCorrectSituation(self): urls = [':8000/urls',\ ':65535',\ ':0000000000000000000000000000000000000000000000065535',\ ''] expected = ['/urls',\ ''] for i in range(0, len(expected)): ret = self.urlValidator._isPortNumberValid(urls[i]) self.assertEqual(expected[i], ret) # input: incorrect input # expected: None def test_illegalPort(self): urls = ['::',\ ':abc', \ ':-1',\ ':',\ ':65536',\ ':065536'] for url in urls: ret = self.urlValidator._isPortNumberValid(url) self.assertEqual(None, ret, 'wrong val at url: ' + url) # input: no port # expected: original url def test_noPort(self): urls = ['/url/paths',\ ''] for url in urls: ret = self.urlValidator._isPortNumberValid(url) self.assertEqual(url, ret, 'wrong val at url: ' + url)