Open Source Monorail

appengine/monorail/sitewide/test/projectsearch_test.py

Index: appengine/monorail/sitewide/test/projectsearch_test.py
diff --git a/appengine/monorail/sitewide/test/projectsearch_test.py b/appengine/monorail/sitewide/test/projectsearch_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..1e71076ca29354ce2fe1c987e991be3a8b96c274
--- /dev/null
+++ b/appengine/monorail/sitewide/test/projectsearch_test.py
@@ -0,0 +1,101 @@
+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is govered by a BSD-style
+# license that can be found in the LICENSE file or at
+# https://developers.google.com/open-source/licenses/bsd
+
+"""Unittests for the projectsearch module."""
+
+import unittest
+
+import mox
+
+from framework import profiler
+from proto import project_pb2
+from services import service_manager
+from sitewide import projectsearch
+from testing import fake
+from testing import testing_helpers
+
+
+class ProjectSearchTest(unittest.TestCase):
+
+  def setUp(self):
+    self.services = service_manager.Services(
+        project=fake.ProjectService())
+
+    for idx, letter in enumerate('abcdefghijklmnopqrstuvwxyz'):
+      self.services.project.TestAddProject(letter, project_id=idx + 1)
+
+    self.mox = mox.Mox()
+    self.mox.StubOutWithMock(self.services.project, 'GetVisibleLiveProjects')
+
+  def tearDown(self):
+    self.mox.UnsetStubs()
+    self.mox.ResetAll()
+
+  def TestPipeline(
+      self, mr=None, expected_start=1, expected_last=None,
+      expected_len=None):
+    if not mr:
+      mr = testing_helpers.MakeMonorailRequest()
+
+    if expected_last is None and expected_len is not None:
+      expected_last = expected_len
+
+    mr.can = 1
+    prof = profiler.Profiler()
+
+    pipeline = projectsearch.ProjectSearchPipeline(mr, self.services, prof)
+    pipeline.SearchForIDs()
+    pipeline.GetProjectsAndPaginate('fake cnxn', '/hosting/search')
+    self.assertEqual(expected_start, pipeline.pagination.start)
+    if expected_last is not None:
+      self.assertEqual(expected_last, pipeline.pagination.last)
+    if expected_len is not None:
+      self.assertEqual(expected_len, len(pipeline.visible_results))
+
+    return pipeline
+
+  def SetUpZeroResults(self):
+    self.services.project.GetVisibleLiveProjects(
+        mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg()).AndReturn([])
+
+  def testZeroResults(self):
+    self.SetUpZeroResults()
+    self.mox.ReplayAll()
+    pipeline = self.TestPipeline(
+        expected_last=0, expected_len=0)
+    self.mox.VerifyAll()
+    self.assertListEqual([], pipeline.visible_results)
+
+  def SetUpNonzeroResults(self):
+    self.services.project.GetVisibleLiveProjects(
+        mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg()).AndReturn([1, 2, 3])
+
+  def testNonzeroResults(self):
+    self.SetUpNonzeroResults()
+    self.mox.ReplayAll()
+    pipeline = self.TestPipeline(
+        expected_last=3, expected_len=3)
+    self.mox.VerifyAll()
+    self.assertListEqual(
+        [1, 2, 3], [p.project_id for p in pipeline.visible_results])
+
+  def SetUpTwoPageResults(self):
+    self.services.project.GetVisibleLiveProjects(
+        mox.IgnoreArg(), mox.IgnoreArg(), mox.IgnoreArg()).AndReturn(
+            range(1, 16))
+
+  def testTwoPageResults(self):
+    """Test more than one pagination page of results."""
+    self.SetUpTwoPageResults()
+    self.mox.ReplayAll()
+    pipeline = self.TestPipeline(
+        expected_last=10, expected_len=10)
+    self.mox.VerifyAll()
+    self.assertEqual(
+        '/hosting/search?num=10&start=10', pipeline.pagination.next_url)
+
+
+if __name__ == '__main__':
+  unittest.main()