Add a ResourceScheduler to ResourceDispatcherHost.

content/browser/renderer_host/resource_scheduler_unittest.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "base/message_loop.h"
+#include "content/browser/renderer_host/resource_loader.h"
+#include "content/browser/renderer_host/resource_scheduler.h"
+#include "content/public/browser/global_request_id.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_test_util.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+namespace {
+
+class TestResourceLoaderFactory;
+
+class TestResourceLoader : public ResourceLoader {
+ public:
+  virtual ~TestResourceLoader() {}
+
+  virtual void StartRequest() {
+    started_ = true;
+  }
+  virtual net::URLRequest* request() { return request_.get(); }
+
+  int id() { return id_; }
+  bool started() { return started_; }
+
+ protected:
+  friend class TestResourceLoaderFactory;
+
+  TestResourceLoader(int request_id,
+                     const char* url,
+                     net::RequestPriority priority,
+                     net::URLRequestContext* context)
+      : id_(request_id),
+        request_(new net::URLRequest(GURL(url), NULL, context, NULL)),
+        started_(false) {
+    request_->set_priority(priority);
+  }
+
+ private:
+  int id_;
+  scoped_ptr<net::URLRequest> request_;
+  bool started_;
+};
+
+class TestResourceLoaderFactory {
+ public:
+  TestResourceLoaderFactory()
+      : next_request_id_(100) {
+  }
+
+  TestResourceLoader* NewLoader(const char* url,
+                                net::RequestPriority priority) {
+    return new TestResourceLoader(
+        ++next_request_id_, url, priority, &context_);
+  }
+
+ private:
+  int next_request_id_;
+  TestURLRequestContext context_;
+};
+
+const int kChildId = 30;
+const int kRouteId = 75;
+
+}  // unnamed namespace
+
+class ResourceSchedulerTest : public testing::Test {
+ protected:
+  ResourceSchedulerTest()
+      : message_loop_(MessageLoop::TYPE_IO) {
+    scheduler_.OnCreate(kChildId, kRouteId);
+  }
+
+  virtual ~ResourceSchedulerTest() {
+    scheduler_.OnDestroy(kChildId, kRouteId);
+  }
+
+  MessageLoop message_loop_;
+  ResourceScheduler scheduler_;
+  TestResourceLoaderFactory factory_;
+};
+
+TEST_F(ResourceSchedulerTest, OneIsolatedRequest) {
+  scoped_ptr<TestResourceLoader> loader(
+      factory_.NewLoader("http://host/1", net::LOW));
+  scheduler_.ScheduleLoad(kChildId, kRouteId, loader->id(), loader.get());
+  EXPECT_TRUE(loader->started());
+  scheduler_.RemoveLoad(kChildId, kRouteId, loader->id());
+}
+
+TEST_F(ResourceSchedulerTest, LowBlocksUntilIdle) {
+  scoped_ptr<TestResourceLoader> high(
+      factory_.NewLoader("http://host/high", net::HIGHEST));
+  scoped_ptr<TestResourceLoader> low(
+      factory_.NewLoader("http://host/low", net::LOW));
+
+  scheduler_.ScheduleLoad(kChildId, kRouteId, high->id(), high.get());
+  scheduler_.ScheduleLoad(kChildId, kRouteId, low->id(), low.get());
+  EXPECT_TRUE(high->started());
+  EXPECT_FALSE(low->started());
+  scheduler_.RemoveLoad(kChildId, kRouteId, high->id());
+  EXPECT_TRUE(low->started());
+  scheduler_.RemoveLoad(kChildId, kRouteId, low->id());
+}
+
+TEST_F(ResourceSchedulerTest, LowBlocksUntilPaint) {
+  scoped_ptr<TestResourceLoader> high(
+      factory_.NewLoader("http://host/high", net::HIGHEST));
+  scoped_ptr<TestResourceLoader> low(
+      factory_.NewLoader("http://host/low", net::LOW));
+
+  scheduler_.ScheduleLoad(kChildId, kRouteId, high->id(), high.get());
+  scheduler_.ScheduleLoad(kChildId, kRouteId, low->id(), low.get());
+  EXPECT_TRUE(high->started());
+  EXPECT_FALSE(low->started());
+  scheduler_.OnPaint(kChildId, kRouteId);
+  EXPECT_TRUE(low->started());
+  scheduler_.RemoveLoad(kChildId, kRouteId, high->id());
+  scheduler_.RemoveLoad(kChildId, kRouteId, low->id());
+}
+
+TEST_F(ResourceSchedulerTest, NavigationResetsPaintState) {
+  scoped_ptr<TestResourceLoader> high(
+      factory_.NewLoader("http://host/high", net::HIGHEST));
+  scoped_ptr<TestResourceLoader> low(
+      factory_.NewLoader("http://host/low", net::LOW));
+
+  scheduler_.OnPaint(kChildId, kRouteId);
+  scheduler_.OnNavigate(kChildId, kRouteId);
+  scheduler_.ScheduleLoad(kChildId, kRouteId, high->id(), high.get());
+  scheduler_.ScheduleLoad(kChildId, kRouteId, low->id(), low.get());
+  EXPECT_TRUE(high->started());
+  EXPECT_FALSE(low->started());
+  scheduler_.RemoveLoad(kChildId, kRouteId, high->id());
+  scheduler_.RemoveLoad(kChildId, kRouteId, low->id());
+}
+
+}  // namespace content