def testAdFrame(self):
request = copy.deepcopy(self._REQUEST)
request.frame_id = '123.123'
trace = test_utils.LoadingTraceFromEvents(
[request] * 10 + [self._REQUEST] * 5, self._PAGE_EVENTS)
lens = ContentClassificationLens(trace, self._RULES, [])
self.assertTrue(lens.IsAdFrame(request.frame_id, .5))
def testMainFrameIsNotAdFrame(self):
trace = test_utils.LoadingTraceFromEvents([self._REQUEST] * 10,
self._PAGE_EVENTS)
lens = ContentClassificationLens(trace, self._RULES, [])
self.assertFalse(lens.IsAdFrame(self._MAIN_FRAME_ID, .5))
