Create the BackgroundFetchBatchManager and initiate a download.

content/browser/background_fetch/background_fetch_data_manager_unittest.cc

 // Copyright 2017 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 "content/browser/background_fetch/background_fetch_data_manager.h"
 
 #include "base/files/file_path.h"
 #include "content/browser/background_fetch/background_fetch_context.h"
+#include "content/browser/background_fetch/batch_request.h"
 #include "content/browser/background_fetch/fetch_request.h"
 #include "content/browser/service_worker/embedded_worker_test_helper.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
 #include "content/public/test/test_browser_context.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 const char kOrigin[] = "https://example.com/";
 const char kResource[] = "https://example.com/resource.html";
 const char kTag[] = "TestRequestTag";
 constexpr int64_t kServiceWorkerRegistrationId = 9001;
 
 }  // namespace
 
 namespace content {
 
 class BackgroundFetchDataManagerTest : public testing::Test {
  protected:
   BackgroundFetchDataManagerTest()
       : helper_(base::FilePath()),
-        background_fetch_context_(
-            new BackgroundFetchContext(&browser_context_,
-                                       helper_.context_wrapper())) {}
+        background_fetch_context_(new BackgroundFetchContext(
+            &browser_context_,
+            BrowserContext::GetDefaultStoragePartition(&browser_context_),
+            helper_.context_wrapper())) {}
 
   BackgroundFetchDataManager* GetDataManager() const {
     return background_fetch_context_->GetDataManagerForTesting();
   }
 
  private:
   TestBrowserThreadBundle browser_thread_bundle_;
   TestBrowserContext browser_context_;
   EmbeddedWorkerTestHelper helper_;
 
   scoped_refptr<BackgroundFetchContext> background_fetch_context_;
 };
 
 TEST_F(BackgroundFetchDataManagerTest, AddRequest) {
+  // Create a BatchRequest and a set of FetchRequests.
+  std::vector<FetchRequest> fetch_requests;
+  std::vector<std::string> fetch_guids;
+  for (int i = 0; i < 10; i++) {
+    FetchRequest fetch_request(GURL(kResource), kTag);
+    fetch_guids.push_back(fetch_request.guid());
+    fetch_requests.push_back(std::move(fetch_request));
+  }
+  BatchRequest batch_request(kTag, url::Origin(GURL(kOrigin)),
+                             kServiceWorkerRegistrationId);
+
+  // Initialize a BatchData with the constructed requests.
   BackgroundFetchDataManager* data_manager = GetDataManager();
-  FetchRequest request(url::Origin(GURL(kOrigin)), GURL(kResource),
-                       kServiceWorkerRegistrationId, kTag);
+  BackgroundFetchDataManager::BatchData* batch_data =
+      data_manager->CreateRequest(batch_request, fetch_requests);
 
-  data_manager->CreateRequest(request);
+  // Get all of the fetch requests from the BatchData.
+  for (int i = 0; i < 10; i++) {
+    EXPECT_FALSE(batch_data->IsComplete());
+    ASSERT_TRUE(batch_data->HasRequestsRemaining());
+    const FetchRequest& fetch_request = batch_data->GetNextFetchRequest();
+    EXPECT_EQ(fetch_request.tag(), kTag);
+  }
 
-  // TODO(harkness) There's no output to test yet. Once there is, check that the
-  // request was actually added.
+  // At this point, all the fetches have been started, but none finished.
+  EXPECT_FALSE(batch_data->HasRequestsRemaining());
+  EXPECT_FALSE(batch_data->IsComplete());
+
+  // Complete all buy one of the fetch requests.
+  for (int i = 0; i < 9; i++) {
+    EXPECT_FALSE(batch_data->FetchRequestComplete(fetch_guids[i]));
+    EXPECT_FALSE(batch_data->IsComplete());
+  }
+
+  // Complete the final fetch request.
+  EXPECT_FALSE(batch_data->FetchRequestComplete(fetch_guids[9]));
+  EXPECT_TRUE(batch_data->IsComplete());
 }
 
 }  // namespace content