def testGetDocumentUrl(self): trace = test_utils.LoadingTraceFromEvents([self._REQUEST], self._PAGE_EVENTS) lens = ContentClassificationLens(trace, [], []) self.assertEquals(self._DOCUMENT_URL, lens._GetDocumentUrl()) # Don't be fooled by redirects. request = copy.deepcopy(self._REQUEST) request.status = 302 request.document_url = 'http://www.bla.com' trace = test_utils.LoadingTraceFromEvents([request, self._REQUEST], self._PAGE_EVENTS) lens = ContentClassificationLens(trace, [], []) self.assertEquals(self._DOCUMENT_URL, lens._GetDocumentUrl())
def testGetDocumentUrl(self): trace = test_utils.LoadingTraceFromEvents( [self._REQUEST], self._PAGE_EVENTS) lens = ContentClassificationLens(trace, [], []) self.assertEquals(self._DOCUMENT_URL, lens._GetDocumentUrl()) # Don't be fooled by redirects. request = copy.deepcopy(self._REQUEST) request.status = 302 request.document_url = 'http://www.bla.com' trace = test_utils.LoadingTraceFromEvents( [request, self._REQUEST], self._PAGE_EVENTS) lens = ContentClassificationLens(trace, [], []) self.assertEquals(self._DOCUMENT_URL, lens._GetDocumentUrl())
def testGetDocumentUrlSeveralChanges(self): request = copy.deepcopy(self._REQUEST) request.status = 200 request.document_url = 'http://www.blabla.com' request2 = copy.deepcopy(request) request2.document_url = 'http://www.blablabla.com' trace = test_utils.LoadingTraceFromEvents( [self._REQUEST, request, request2], self._PAGE_EVENTS) lens = ContentClassificationLens(trace, [], []) self.assertEquals(request2.document_url, lens._GetDocumentUrl())
def testGetDocumentUrlSeveralChanges(self): request = copy.deepcopy(self._REQUEST) request.status = 200 request.document_url = 'http://www.blabla.com' request2 = copy.deepcopy(request) request2.document_url = 'http://www.blablabla.com' trace = test_utils.LoadingTraceFromEvents( [self._REQUEST, request, request2], self._PAGE_EVENTS) lens = ContentClassificationLens(trace, [], []) self.assertEquals(request2.document_url, lens._GetDocumentUrl())