prerender_tracker_unittest.cc revision 9ab5563a3196760eb381d102cbb2bc0f7abc6a50
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <set> 62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <utility> 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 99ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chrome/browser/prerender/prerender_contents.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/prerender/prerender_manager.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/prerender/prerender_tracker.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/test/base/testing_browser_process.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/test/test_browser_thread.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::BrowserThread; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace prerender { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class TestPrerenderContents : public PrerenderContents { 242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) public: 252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestPrerenderContents(PrerenderManager* prerender_manager, 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int child_id, int route_id) 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : PrerenderContents(prerender_manager, static_cast<Profile*>(NULL), 282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GURL(), content::Referrer(), ORIGIN_NONE, 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) PrerenderManager::kNoExperiment), 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) child_id_(child_id), 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) route_id_(route_id) { 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual ~TestPrerenderContents() { 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (final_status() == FINAL_STATUS_MAX) 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SetFinalStatus(FINAL_STATUS_USED); 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual bool GetChildId(int* child_id) const OVERRIDE { 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) *child_id = child_id_; 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual bool GetRouteId(int* route_id) const OVERRIDE { 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) *route_id = route_id_; 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void Start() { 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) AddObserver(prerender_manager()->prerender_tracker()); 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerendering_has_started_ = true; 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NotifyPrerenderStart(); 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void Cancel() { 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) Destroy(FINAL_STATUS_CANCELLED); 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void Use() { 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SetFinalStatus(FINAL_STATUS_USED); 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) PrepareForUse(); 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) private: 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int child_id_; 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int route_id_; 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestPrerenderManager : public PrerenderManager { 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit TestPrerenderManager(PrerenderTracker* prerender_tracker) : 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrerenderManager(NULL, prerender_tracker) { 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mutable_config().rate_limit_enabled = false; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void DestroyPrerenderForRenderView( 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int process_id, int view_id, FinalStatus final_status) OVERRIDE { 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cancelled_id_pairs_.insert(std::make_pair(process_id, view_id)); 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // We never allocate our PrerenderContents in PrerenderManager, so we don't 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // ever want the default pending delete behaviour. 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void MoveEntryToPendingDelete(PrerenderContents* entry, 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) FinalStatus final_status) OVERRIDE { 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool WasPrerenderCancelled(int child_id, int route_id) { 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::pair<int, int> child_route_id_pair(child_id, route_id); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return cancelled_id_pairs_.count(child_route_id_pair) != 0; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set of all the RenderViews that have been cancelled. 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::set<std::pair<int, int> > cancelled_id_pairs_; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class PrerenderTrackerTest : public testing::Test { 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) static const int kDefaultChildId = 0; 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) static const int kDefaultRouteId = 100; 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrerenderTrackerTest() : 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui_thread_(BrowserThread::UI, &message_loop_), 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io_thread_(BrowserThread::IO, &message_loop_), 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) prerender_manager_(prerender_tracker()), 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents_(&prerender_manager_, kDefaultChildId, kDefaultRouteId) { 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) PrerenderTracker* prerender_tracker() { 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return g_browser_process->prerender_tracker(); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestPrerenderManager* prerender_manager() { 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return &prerender_manager_; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestPrerenderContents* test_contents() { 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return &test_contents_; 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int GetCurrentStatus(int child_id, int route_id) { 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FinalStatus final_status; 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!prerender_tracker()->GetFinalStatus(child_id, route_id, 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &final_status)) { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return -1; 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return final_status; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Runs any tasks queued on either thread. 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RunEvents() { 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) message_loop_.RunUntilIdle(); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 13790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop message_loop_; 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::TestBrowserThread ui_thread_; 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::TestBrowserThread io_thread_; 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestPrerenderManager prerender_manager_; 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestPrerenderContents test_contents_; 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that a non-existant RenderView is handled correctly. 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(PrerenderTrackerTest, PrerenderTrackerNull) { 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryUse(kDefaultChildId, kDefaultRouteId)); 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancel( 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancelOnIOThread( 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) FinalStatus final_status; 1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, &final_status)); 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_manager()->WasPrerenderCancelled( 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that a page that is used is handled correctly. 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(PrerenderTrackerTest, PrerenderTrackerUsed) { 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Start(); 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_MAX, 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls AddPrerenderOnIOThreadTask(). 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->IsPrerenderingOnIOThread( 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_MAX, 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Display the prerendered RenderView. 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryUse(kDefaultChildId, kDefaultRouteId)); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure the page can't be destroyed or claim it was destroyed after 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // it's been used. 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancel( 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancelOnIOThread( 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_USED, 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This would call DestroyPrerenderForChildRouteIdPair(), if the prerender 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // were cancelled. 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These functions should all behave as before. 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancel( 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancelOnIOThread( 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_USED, GetCurrentStatus( 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls DestroyPrerenderForChildRouteIdPair(). 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Use(); 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->IsPrerenderingOnIOThread( 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls RemovePrerenderOnIOThreadTask(). 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FinalStatus final_status; 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, &final_status)); 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancel( 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_manager()->WasPrerenderCancelled( 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that a prerendered page cancelled by TryCancel() is handled correctly. 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(PrerenderTrackerTest, PrerenderTrackerCancelled) { 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Start(); 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_MAX, 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls AddPrerenderOnIOThreadTask(). 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancel the prerender. 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryUse(kDefaultChildId, kDefaultRouteId)); 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_CANCELLED, 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls DestroyPrerenderForChildRouteIdPair(). 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_manager()->WasPrerenderCancelled( 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These should all work until the prerendering RenderViewHost is destroyed. 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_CANCELLED, 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Cancel(); 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->IsPrerenderingOnIOThread( 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls RemovePrerenderOnIOThreadTask(). 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FinalStatus final_status; 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, &final_status)); 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that a prerendered page cancelled on the IO thread by 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TryCancelOnIOThread() is handled correctly. 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(PrerenderTrackerTest, PrerenderTrackerCancelledOnIO) { 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Start(); 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_MAX, 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls AddPrerenderOnIOThreadTask(). 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancel the prerender. 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryUse(kDefaultChildId, kDefaultRouteId)); 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel(kDefaultChildId, kDefaultRouteId, 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) FINAL_STATUS_CANCELLED)); 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_TIMED_OUT, 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls DestroyPrerenderForChildRouteIdPair(). 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_manager()->WasPrerenderCancelled( 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These should all work until the prerendering RenderViewHost is destroyed. 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_TIMED_OUT, 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Cancel(); 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->IsPrerenderingOnIOThread( 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls RemovePrerenderOnIOThreadTask(). 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FinalStatus final_status; 3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, &final_status)); 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that a prerendered page cancelled before it reaches the IO thread is 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// handled correctly. 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(PrerenderTrackerTest, PrerenderTrackerCancelledFast) { 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Start(); 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cancel the prerender. 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryUse(kDefaultChildId, kDefaultRouteId)); 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls AddPrerenderOnIOThreadTask() and 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // DestroyPrerenderForChildRouteIdPair(). 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_manager()->WasPrerenderCancelled( 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_TIMED_OUT)); 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_CANCELLED, GetCurrentStatus( 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Cancel(); 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls RemovePrerenderOnIOThreadTask(). 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FinalStatus final_status; 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, &final_status)); 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Check that handling two pages at once works. 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(PrerenderTrackerTest, PrerenderTrackerMultiple) { 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Start(); 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls AddPrerenderOnIOThreadTask(). 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->IsPrerenderingOnIOThread( 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1)); 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryUse( 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1)); 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancel( 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1, FINAL_STATUS_CANCELLED)); 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Start second prerender. 3702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestPrerenderContents second_test_contents(prerender_manager(), 3712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, 3722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultRouteId + 1); 3732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) second_test_contents.Start(); 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls AddPrerenderOnIOThreadTask(). 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Use (kDefaultChildId, kDefaultRouteId). 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryUse(kDefaultChildId, kDefaultRouteId)); 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_USED, GetCurrentStatus( 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 3822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_MAX, 3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId + 1, kDefaultRouteId + 1)); 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Cancel (kDefaultChildId + 1, kDefaultRouteId + 1). 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancelOnIOThread( 3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1, FINAL_STATUS_CANCELLED)); 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryCancel( 3902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, FINAL_STATUS_CANCELLED)); 3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_USED, 3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId, kDefaultRouteId)); 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->TryUse( 3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1)); 3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_tracker()->TryCancel( 3972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1, FINAL_STATUS_CANCELLED)); 3982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(FINAL_STATUS_CANCELLED, 3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetCurrentStatus(kDefaultChildId + 1, kDefaultRouteId + 1)); 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls DestroyPrerenderForChildRouteIdPair(). 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_manager()->WasPrerenderCancelled(kDefaultChildId, 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultRouteId)); 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(prerender_manager()->WasPrerenderCancelled(kDefaultChildId + 1, 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultRouteId + 1)); 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) test_contents()->Cancel(); 4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) second_test_contents.Cancel(); 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This calls RemovePrerenderOnIOThreadTask(). 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunEvents(); 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FinalStatus final_status; 4152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId, &final_status)); 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId, kDefaultRouteId)); 4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->GetFinalStatus( 4212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1, &final_status)); 4222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(prerender_tracker()->IsPrerenderingOnIOThread( 4232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kDefaultChildId + 1, kDefaultRouteId + 1)); 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace prerender 427