def testAdFrame(self):
request = copy.deepcopy(self._REQUEST)
request.request_id = '1234.2'
request.frame_id = '123.123'
trace = test_utils.LoadingTraceFromEvents([self._REQUEST, request],
self._PAGE_EVENTS)
lens = ContentClassificationLens(trace, self._RULES, [])
self.assertTrue(lens.IsAdOrTrackingFrame(request.frame_id))
def testMainFrameIsNotAnAdFrame(self):
trace = test_utils.LoadingTraceFromEvents([self._REQUEST],
self._PAGE_EVENTS)
lens = ContentClassificationLens(trace, self._RULES, [])
self.assertFalse(lens.IsAdOrTrackingFrame(self._MAIN_FRAME_ID))
