Changes to fit better with the needs of the download manager.

components/offline_pages/background/request_queue_store_unittest.cc

 // Copyright 2016 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 "components/offline_pages/background/request_queue_store.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/test/test_simple_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
+#include "components/offline_pages/background/request_queue.h"
 #include "components/offline_pages/background/request_queue_in_memory_store.h"
 #include "components/offline_pages/background/request_queue_store_sql.h"
 #include "components/offline_pages/background/save_page_request.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace offline_pages {
 
 using UpdateStatus = RequestQueueStore::UpdateStatus;
 
 namespace {
@@ skipping 36 matching lines @@
 
   void PumpLoop();
   void ClearResults();
 
   // Callback used for get requests.
   void GetRequestsDone(bool result,
                        const std::vector<SavePageRequest>& requests);
   // Callback used for add/update request.
   void AddOrUpdateDone(UpdateStatus result);
   // Callback used for remove requests.
-  void RemoveDone(bool result, int count);
+  void RemoveDone(
+      const std::vector<int64_t>& request_ids,
+      const std::vector<RequestQueue::UpdateRequestResult>& results);
   // Callback used for reset.
   void ResetDone(bool result);
 
   LastResult last_result() const { return last_result_; }
   UpdateStatus last_update_status() const { return last_update_status_; }
-  int last_remove_count() const { return last_remove_count_; }
+  const std::vector<int64_t>& last_remove_request_ids() {
+    return last_remove_request_ids_;
+  }
+  const std::vector<RequestQueue::UpdateRequestResult>& last_remove_results()
+      const {
+    return last_remove_results_;
+  }
   const std::vector<SavePageRequest>& last_requests() const {
     return last_requests_;
   }
 
  protected:
   base::ScopedTempDir temp_directory_;
 
  private:
   LastResult last_result_;
   UpdateStatus last_update_status_;
-  int last_remove_count_;
+  std::vector<int64_t> last_remove_request_ids_;
+  std::vector<RequestQueue::UpdateRequestResult> last_remove_results_;
   std::vector<SavePageRequest> last_requests_;
 
   scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
   base::ThreadTaskRunnerHandle task_runner_handle_;
 };
 
 RequestQueueStoreTestBase::RequestQueueStoreTestBase()
     : last_result_(LastResult::kNone),
       last_update_status_(UpdateStatus::FAILED),
-      last_remove_count_(0),
       task_runner_(new base::TestSimpleTaskRunner),
       task_runner_handle_(task_runner_) {
   EXPECT_TRUE(temp_directory_.CreateUniqueTempDir());
 }
 
 void RequestQueueStoreTestBase::TearDown() {
   // Wait for all the pieces of the store to delete itself properly.
   PumpLoop();
 }
 
 void RequestQueueStoreTestBase::PumpLoop() {
   task_runner_->RunUntilIdle();
 }
 
 void RequestQueueStoreTestBase::ClearResults() {
   last_result_ = LastResult::kNone;
   last_update_status_ = UpdateStatus::FAILED;
-  last_remove_count_ = 0;
+  last_remove_request_ids_.clear();
+  last_remove_results_.clear();
   last_requests_.clear();
 }
 
 void RequestQueueStoreTestBase::GetRequestsDone(
     bool result,
     const std::vector<SavePageRequest>& requests) {
   last_result_ = result ? LastResult::kTrue : LastResult::kFalse;
   last_requests_ = requests;
 }
 
 void RequestQueueStoreTestBase::AddOrUpdateDone(UpdateStatus status) {
   last_update_status_ = status;
 }
 
-void RequestQueueStoreTestBase::RemoveDone(bool result, int count) {
-  last_result_ = result ? LastResult::kTrue : LastResult::kFalse;
-  last_remove_count_ = count;
+void RequestQueueStoreTestBase::RemoveDone(
+    const std::vector<int64_t>& request_ids,
+    const std::vector<RequestQueue::UpdateRequestResult>& results) {
+  last_remove_request_ids_ = request_ids;
+  last_remove_results_ = results;
 }
 
 void RequestQueueStoreTestBase::ResetDone(bool result) {
   last_result_ = result ? LastResult::kTrue : LastResult::kFalse;
 }
 
 // Defines interface for the store factory.
 class RequestQueueStoreFactory {
  public:
   virtual RequestQueueStore* BuildStore(const base::FilePath& path) = 0;
@@ skipping 107 matching lines @@
   this->ClearResults();
   store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone,
                                 base::Unretained(this)));
   ASSERT_EQ(LastResult::kNone, this->last_result());
   this->PumpLoop();
   ASSERT_EQ(LastResult::kTrue, this->last_result());
   ASSERT_EQ(1ul, this->last_requests().size());
   ASSERT_TRUE(updated_request == this->last_requests()[0]);
 }
 
-TYPED_TEST(RequestQueueStoreTest, RemoveRequest) {
+TYPED_TEST(RequestQueueStoreTest, RemoveRequests) {
   std::unique_ptr<RequestQueueStore> store(this->BuildStore());
   base::Time creation_time = base::Time::Now();
-  SavePageRequest original_request(
-      kRequestId, kUrl, kClientId, creation_time, kUserRequested);
+  SavePageRequest request1(kRequestId, kUrl, kClientId, creation_time,
+                           kUserRequested);
   store->AddOrUpdateRequest(
-      original_request, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
-                                   base::Unretained(this)));
+      request1, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
+                           base::Unretained(this)));
+  SavePageRequest request2(kRequestId2, kUrl2, kClientId2, creation_time,
+                           kUserRequested);
+  store->AddOrUpdateRequest(
+      request2, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
+                           base::Unretained(this)));
   this->PumpLoop();
   this->ClearResults();
 
-  std::vector<int64_t> request_ids{kRequestId};
+  std::vector<int64_t> request_ids{kRequestId, kRequestId2};
   store->RemoveRequests(request_ids,
                         base::Bind(&RequestQueueStoreTestBase::RemoveDone,
                                    base::Unretained(this)));
-  ASSERT_EQ(LastResult::kNone, this->last_result());
-  ASSERT_EQ(0, this->last_remove_count());
+  ASSERT_EQ(0ul, this->last_remove_request_ids().size());
+  ASSERT_EQ(0ul, this->last_remove_results().size());
   this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  ASSERT_EQ(1, this->last_remove_count());
+  ASSERT_EQ(2ul, this->last_remove_request_ids().size());
+  ASSERT_EQ(2ul, this->last_remove_results().size());
+  ASSERT_EQ(RequestQueue::UpdateRequestResult::SUCCESS,
+            this->last_remove_results().at(0));
+  ASSERT_EQ(RequestQueue::UpdateRequestResult::SUCCESS,
+            this->last_remove_results().at(1));
   this->ClearResults();
 
   store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone,
                                 base::Unretained(this)));
   this->PumpLoop();
   ASSERT_EQ(LastResult::kTrue, this->last_result());
   ASSERT_TRUE(this->last_requests().empty());
   this->ClearResults();
 
   // Try to remove a request that is not in the queue.
   store->RemoveRequests(request_ids,
                         base::Bind(&RequestQueueStoreTestBase::RemoveDone,
                                    base::Unretained(this)));
   ASSERT_EQ(LastResult::kNone, this->last_result());
-  ASSERT_EQ(0, this->last_remove_count());
   this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  ASSERT_EQ(0, this->last_remove_count());
-}
-
-TYPED_TEST(RequestQueueStoreTest, RemoveRequestByClientId) {
-  std::unique_ptr<RequestQueueStore> store(this->BuildStore());
-  base::Time creation_time = base::Time::Now();
-
-  // Create requests and add them to the queue.
-  SavePageRequest request1(kRequestId, kUrl, kClientId, creation_time,
-                                   kUserRequested);
-  SavePageRequest request2(kRequestId2, kUrl2, kClientId2, creation_time,
-                                   kUserRequested);
-  store->AddOrUpdateRequest(
-      request1, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
-                           base::Unretained(this)));
-  store->AddOrUpdateRequest(
-      request2, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
-                           base::Unretained(this)));
-  this->PumpLoop();
-  this->ClearResults();
-
-  // Remove a request.
-  std::vector<ClientId> client_ids_to_delete{kClientId};
-  store->RemoveRequestsByClientId(
-      client_ids_to_delete,
-      base::Bind(
-          &RequestQueueStoreTestBase::RemoveDone, base::Unretained(this)));
-  ASSERT_EQ(LastResult::kNone, this->last_result());
-  ASSERT_EQ(0, this->last_remove_count());
-  this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  ASSERT_EQ(1, this->last_remove_count());
-  this->ClearResults();
-
-  // Check to see what remains in the queue.  Removed request should be gone.
-  store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone,
-                                base::Unretained(this)));
-  this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  // The other request should still be in the queue.
-  ASSERT_EQ(1UL, this->last_requests().size());
-  ASSERT_EQ(kClientId2, this->last_requests().at(0).client_id());
-  this->ClearResults();
-
-  // Try to remove a request that is not in the queue.
-  store->RemoveRequestsByClientId(
-      client_ids_to_delete,
-      base::Bind(
-          &RequestQueueStoreTestBase::RemoveDone, base::Unretained(this)));
-  this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  ASSERT_EQ(0, this->last_remove_count());
-}
-
-TYPED_TEST(RequestQueueStoreTest, RemoveRequestWithSameClientId) {
-  std::unique_ptr<RequestQueueStore> store(this->BuildStore());
-  base::Time creation_time = base::Time::Now();
-
-  // Create requests and add them to the queue.
-  SavePageRequest request1(kRequestId, kUrl, kClientId, creation_time,
-                                   kUserRequested);
-  SavePageRequest request2(kRequestId2, kUrl2, kClientId, creation_time,
-                                   kUserRequested);
-  store->AddOrUpdateRequest(
-      request1, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
-                           base::Unretained(this)));
-  store->AddOrUpdateRequest(
-      request2, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
-                           base::Unretained(this)));
-  this->PumpLoop();
-  this->ClearResults();
-
-  // Remove a request.
-  std::vector<ClientId> client_ids_to_delete{kClientId};
-  store->RemoveRequestsByClientId(
-      client_ids_to_delete,
-      base::Bind(
-          &RequestQueueStoreTestBase::RemoveDone, base::Unretained(this)));
-  ASSERT_EQ(LastResult::kNone, this->last_result());
-  ASSERT_EQ(0, this->last_remove_count());
-  this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  ASSERT_EQ(2, this->last_remove_count());
-  this->ClearResults();
-
-  // Check to see what remains in the queue.  Removed request should be gone.
-  store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone,
-                                base::Unretained(this)));
-  this->PumpLoop();
-  ASSERT_EQ(LastResult::kTrue, this->last_result());
-  // The other request should still be in the queue.
-  ASSERT_TRUE(this->last_requests().empty());
-  this->ClearResults();
+  ASSERT_EQ(2ul, this->last_remove_results().size());
+  // Since the SQL statement returns true on a delete of an item that isn't
+  // present, SQL is returning SUCCESS, but the memory is returning
+  // REQUEST_DOES_NOT_EXIST, so we just check that the result is not failure.
+  ASSERT_NE(RequestQueue::UpdateRequestResult::STORE_FAILURE,
+            this->last_remove_results().at(0));
+  ASSERT_NE(RequestQueue::UpdateRequestResult::STORE_FAILURE,
+            this->last_remove_results().at(1));
 }
 
 TYPED_TEST(RequestQueueStoreTest, ResetStore) {
   std::unique_ptr<RequestQueueStore> store(this->BuildStore());
   base::Time creation_time = base::Time::Now();
   SavePageRequest original_request(
       kRequestId, kUrl, kClientId, creation_time, kUserRequested);
   store->AddOrUpdateRequest(
       original_request, base::Bind(&RequestQueueStoreTestBase::AddOrUpdateDone,
                                    base::Unretained(this)));
@@ skipping 37 matching lines @@
   store->GetRequests(base::Bind(&RequestQueueStoreTestBase::GetRequestsDone,
                                 base::Unretained(this)));
   ASSERT_EQ(LastResult::kNone, this->last_result());
   this->PumpLoop();
   ASSERT_EQ(LastResult::kTrue, this->last_result());
   ASSERT_EQ(1ul, this->last_requests().size());
   ASSERT_TRUE(original_request == this->last_requests()[0]);
 }
 
 }  // offline_pages