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