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))