DownloadQuery filters and sorts DownloadItems.

content/browser/download/download_query_unittest.cc

Index: content/browser/download/download_query_unittest.cc
diff --git a/content/browser/download/download_query_unittest.cc b/content/browser/download/download_query_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4e50bf9957beccc2694991b862da4f12dbb17940
--- /dev/null
+++ b/content/browser/download/download_query_unittest.cc
@@ -0,0 +1,79 @@
+// Copyright (c) 2011 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 <string>
+
+#include "base/logging.h"
+#include "base/stl_util.h"
+#include "content/browser/download/download_query.h"
+#include "content/browser/download/mock_download_item.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using download_util::DownloadQuery;
+using ::testing::Return;
+using ::testing::ReturnRef;
+typedef DownloadQuery::DownloadVector DownloadVector;
+
+class DownloadQueryTest : public testing::Test {
+ public:
+  DownloadQueryTest() {}
+  virtual ~DownloadQueryTest() {}
+  virtual void TearDown() {
+    STLDeleteElements(&mocks_);
+  }
+  void CreateMocks(int count) {
+    for (int i = 0; i < count; ++i) {
+      mocks_.push_back(new MockDownloadItem());
+      all_items_.push_back(mocks_[i]);
+    }
+  }
+  MockDownloadItem& mock(int index) { return *mocks_[index]; }
+  DownloadQuery* query() { return &query_; }
+  DownloadVector* all_items() { return &all_items_; }
+  DownloadVector* results() { return &results_; }
+  void Search() {
+    query()->Search(all_items_.begin(), all_items_.end(), &results_);
+  }
+
+ private:
+  std::vector<MockDownloadItem*> mocks_;
+  DownloadQuery query_;
+  DownloadVector all_items_;
+  DownloadVector results_;
+
+  DISALLOW_COPY_AND_ASSIGN(DownloadQueryTest);
+};
+
+TEST_F(DownloadQueryTest, EmptyQueryNoItems) {
+  Search();
+  EXPECT_EQ(0U, results()->size());
+}
+
+TEST_F(DownloadQueryTest, EmptyQuerySomeItems) {
+  CreateMocks(3);
+  Search();
+  EXPECT_EQ(3U, results()->size());
+}
+
+TEST_F(DownloadQueryTest, FilterBytesReceived) {
+  CreateMocks(3);
+  query()->Filter(DownloadQuery::FILTER_FIELD_BYTES_RECEIVED, 2);
+  EXPECT_CALL(mock(0), GetReceivedBytes()).WillOnce(Return(1));
+  EXPECT_CALL(mock(1), GetReceivedBytes()).WillOnce(Return(2));
+  EXPECT_CALL(mock(2), GetReceivedBytes()).WillOnce(Return(2));
+  Search();
+  EXPECT_EQ(2U, results()->size());
+}
+
+TEST_F(DownloadQueryTest, FilterRegexes) {
+  CreateMocks(3);
+  query()->Filter(DownloadQuery::FILTER_FIELD_URL_REGEX, std::string("fo{2}"));
+  GURL url0("http://example.com"), url1("http://foo.com"), url2("http://bar.com/foo/bar");
+  EXPECT_CALL(mock(0), GetOriginalUrl()).WillOnce(ReturnRef(url0));
+  EXPECT_CALL(mock(1), GetOriginalUrl()).WillOnce(ReturnRef(url1));
+  EXPECT_CALL(mock(2), GetOriginalUrl()).WillOnce(ReturnRef(url2));
+  Search();
+  EXPECT_EQ(2U, results()->size());
+}