def testCategoryBase64(self):
categ = Categorizor('querystring:category:base64')
self.assertEquals(
'violence',
categ.categorize('http://isp.example.com/blocked?'
'category=dmlvbGVuY2U=')
)
def testCategoryMissing(self):
categ = Categorizor('querystring:category')
self.assertIsNone(
categ.categorize(
'http://isp.example.com/blocked'
)
)
def testCategory(self):
categ = Categorizor('querystring:category')
self.assertEquals(
'violence',
categ.categorize('http://isp.example.com/blocked?'
'category=violence')
)
def setUp(self):
self.probe = OrgProbe({})
self.probe.rules = [
"re:url:^http://www\\.talktalk\\.co\\.uk/" +
"notice/parental-controls\\?accessurl",
"re:url:^http://www\\.siteblocked\\.org/piratebay\\.html\\?"
]
self.probe.categorizor = Categorizor('querystring:urlclassname:base64')
def testCategoryMissing(self):
categ = Categorizor('querystring:category')
self.assertIsNone(categ.categorize('http://isp.example.com/blocked'))
def testCategoryBase64(self):
categ = Categorizor('querystring:category:base64')
self.assertEquals(
'violence',
categ.categorize('http://isp.example.com/blocked?'
'category=dmlvbGVuY2U='))
def testCategory(self):
categ = Categorizor('querystring:category')
self.assertEquals(
'violence',
categ.categorize('http://isp.example.com/blocked?'
'category=violence'))