Index: appengine/monorail/sitewide/test/userupdates_test.py |
diff --git a/appengine/monorail/sitewide/test/userupdates_test.py b/appengine/monorail/sitewide/test/userupdates_test.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c662abaabba4d6981990c59c1af626c0384786b0 |
--- /dev/null |
+++ b/appengine/monorail/sitewide/test/userupdates_test.py |
@@ -0,0 +1,116 @@ |
+# 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 monorail.sitewide.userupdates.""" |
+ |
+import unittest |
+ |
+import mox |
+ |
+from features import activities |
+from services import service_manager |
+from sitewide import sitewide_helpers |
+from sitewide import userupdates |
+from testing import fake |
+from testing import testing_helpers |
+ |
+ |
+class ProjectUpdatesTest(unittest.TestCase): |
+ |
+ def setUp(self): |
+ self.services = service_manager.Services( |
+ project=fake.ProjectService(), |
+ user_star=fake.UserStarService()) |
+ |
+ self.user_id = 2 |
+ self.project_id = 987 |
+ self.project = self.services.project.TestAddProject( |
+ 'proj', project_id=self.project_id) |
+ |
+ self.mr = testing_helpers.MakeMonorailRequest( |
+ services=self.services, project=self.project) |
+ self.mr.cnxn = 'fake cnxn' |
+ self.mr.viewed_user_auth.user_id = 100 |
+ |
+ self.mox = mox.Mox() |
+ |
+ def tearDown(self): |
+ self.mox.UnsetStubs() |
+ self.mox.ResetAll() |
+ |
+ def testUserUpdatesProjects(self): |
+ uup = userupdates.UserUpdatesProjects(None, None, self.services) |
+ |
+ self.mox.StubOutWithMock(sitewide_helpers, 'GetViewableStarredProjects') |
+ sitewide_helpers.GetViewableStarredProjects( |
+ self.mr.cnxn, self.services, self.mr.viewed_user_auth.user_id, |
+ self.mr.auth.effective_ids, self.mr.auth.user_pb).AndReturn( |
+ [self.project]) |
+ |
+ self.mox.StubOutWithMock(activities, 'GatherUpdatesData') |
+ activities.GatherUpdatesData( |
+ self.services, self.mr, mox.IgnoreArg(), |
+ user_ids=None, |
+ project_ids=[self.project_id], |
+ ending=uup._ENDING, |
+ updates_page_url=uup._UPDATES_PAGE_URL, |
+ highlight=uup._HIGHLIGHT).AndReturn({}) |
+ |
+ self.mox.ReplayAll() |
+ |
+ page_data = uup.GatherPageData(self.mr) |
+ self.mox.VerifyAll() |
+ self.assertEqual(3, len(page_data)) |
+ self.assertEqual('st5', page_data['user_tab_mode']) |
+ self.assertEqual('yes', page_data['viewing_user_page']) |
+ self.assertEqual(uup._TAB_MODE, page_data['user_updates_tab_mode']) |
+ |
+ def testUserUpdatesDevelopers(self): |
+ uud = userupdates.UserUpdatesDevelopers(None, None, self.services) |
+ |
+ self.mox.StubOutWithMock(self.services.user_star, 'LookupStarredItemIDs') |
+ self.services.user_star.LookupStarredItemIDs( |
+ self.mr.cnxn, self.mr.viewed_user_auth.user_id).AndReturn( |
+ [self.user_id]) |
+ |
+ self.mox.StubOutWithMock(activities, 'GatherUpdatesData') |
+ activities.GatherUpdatesData( |
+ self.services, self.mr, mox.IgnoreArg(), |
+ user_ids=[self.user_id], |
+ project_ids=None, |
+ ending=uud._ENDING, |
+ updates_page_url=uud._UPDATES_PAGE_URL, |
+ highlight=uud._HIGHLIGHT).AndReturn({}) |
+ |
+ self.mox.ReplayAll() |
+ |
+ page_data = uud.GatherPageData(self.mr) |
+ self.mox.VerifyAll() |
+ self.assertEqual(3, len(page_data)) |
+ self.assertEqual('st5', page_data['user_tab_mode']) |
+ self.assertEqual('yes', page_data['viewing_user_page']) |
+ self.assertEqual(uud._TAB_MODE, page_data['user_updates_tab_mode']) |
+ |
+ def testUserUpdatesIndividual(self): |
+ uui = userupdates.UserUpdatesIndividual(None, None, self.services) |
+ |
+ self.mox.StubOutWithMock(activities, 'GatherUpdatesData') |
+ activities.GatherUpdatesData( |
+ self.services, self.mr, mox.IgnoreArg(), |
+ user_ids=[self.mr.viewed_user_auth.user_id], |
+ project_ids=None, |
+ ending=uui._ENDING, |
+ updates_page_url=uui._UPDATES_PAGE_URL, |
+ highlight=uui._HIGHLIGHT).AndReturn({}) |
+ |
+ self.mox.ReplayAll() |
+ |
+ page_data = uui.GatherPageData(self.mr) |
+ self.mox.VerifyAll() |
+ self.assertEqual(3, len(page_data)) |
+ self.assertEqual('st5', page_data['user_tab_mode']) |
+ self.assertEqual('yes', page_data['viewing_user_page']) |
+ self.assertEqual(uui._TAB_MODE, page_data['user_updates_tab_mode']) |
+ |