Open Source Monorail

appengine/monorail/framework/test/reap_test.py

Index: appengine/monorail/framework/test/reap_test.py
diff --git a/appengine/monorail/framework/test/reap_test.py b/appengine/monorail/framework/test/reap_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..dc71506001ab8dc97d4475a7fb979229297efc4c
--- /dev/null
+++ b/appengine/monorail/framework/test/reap_test.py
@@ -0,0 +1,117 @@
+# 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
+
+"""Tests for the reap module."""
+
+import unittest
+
+import mox
+
+from framework import reap
+from framework import sql
+from proto import project_pb2
+from services import service_manager
+from testing import fake
+from testing import testing_helpers
+
+
+class ReapTest(unittest.TestCase):
+
+  def setUp(self):
+    self.project_service = fake.ProjectService()
+    self.issue_service = fake.IssueService()
+    self.issue_star_service = fake.IssueStarService()
+    self.config_service = fake.ConfigService()
+    self.features_service = fake.FeaturesService()
+    self.project_star_service = fake.ProjectStarService()
+    self.services = service_manager.Services(
+        project=self.project_service,
+        issue=self.issue_service,
+        issue_star=self.issue_star_service,
+        config=self.config_service,
+        features=self.features_service,
+        project_star=self.project_star_service,
+        user=fake.UserService(),
+        usergroup=fake.UserGroupService())
+
+    self.proj1_id = 1001
+    self.proj1_issue_id = 111
+    self.proj1 = self.project_service.TestAddProject(
+        name='proj1', project_id=self.proj1_id)
+    self.proj2_id = 1002
+    self.proj2_issue_id = 112
+    self.proj2 = self.project_service.TestAddProject(
+        name='proj2', project_id=self.proj2_id)
+
+    self.mox = mox.Mox()
+    self.cnxn = self.mox.CreateMock(sql.MonorailConnection)
+    self.project_service.project_tbl = self.mox.CreateMock(sql.SQLTableManager)
+    self.issue_service.issue_tbl = self.mox.CreateMock(sql.SQLTableManager)
+
+  def tearDown(self):
+    self.mox.UnsetStubs()
+    self.mox.ResetAll()
+
+  def setUpMarkDoomedProjects(self):
+    self.project_service.project_tbl.Select(
+        self.cnxn, cols=['project_id'], limit=1000, state='archived',
+        where=mox.IgnoreArg()).AndReturn([[self.proj1_id]])
+
+  def testMarkDoomedProjects(self):
+    self.setUpMarkDoomedProjects()
+    reaper = reap.Reap('req', 'resp', services=self.services)
+
+    self.mox.ReplayAll()
+    doomed_project_ids = reaper._MarkDoomedProjects(self.cnxn)
+    self.mox.VerifyAll()
+
+    self.assertEquals([self.proj1_id], doomed_project_ids)
+    self.assertEquals(project_pb2.ProjectState.DELETABLE, self.proj1.state)
+    self.assertEquals('DELETABLE_%s' % self.proj1_id , self.proj1.project_name)
+
+  def setUpExpungeParts(self):
+    self.project_service.project_tbl.Select(
+        self.cnxn, cols=['project_id'], limit=100,
+        state='deletable').AndReturn([[self.proj1_id], [self.proj2_id]])
+    self.issue_service.issue_tbl.Select(
+        self.cnxn, cols=['id'], limit=1000,
+        project_id=self.proj1_id).AndReturn([[self.proj1_issue_id]])
+    self.issue_service.issue_tbl.Select(
+        self.cnxn, cols=['id'], limit=1000,
+        project_id=self.proj2_id).AndReturn([[self.proj2_issue_id]])
+
+  def testExpungeDeletableProjects(self):
+    self.setUpExpungeParts()
+    reaper = reap.Reap('req', 'resp', services=self.services)
+
+    self.mox.ReplayAll()
+    expunged_project_ids = reaper._ExpungeDeletableProjects(self.cnxn)
+    self.mox.VerifyAll()
+
+    self.assertEquals([self.proj1_id, self.proj2_id], expunged_project_ids)
+    # Verify all expected expunge methods were called.
+    self.assertEquals([self.proj1_issue_id, self.proj2_issue_id],
+                      self.services.issue_star.expunged_item_ids)
+    self.assertEquals([self.proj1_issue_id, self.proj2_issue_id],
+                      self.services.issue.expunged_issues)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.config.expunged_configs)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.features.expunged_saved_queries)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.features.expunged_filter_rules)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.issue.expunged_former_locations)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.issue.expunged_local_ids)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.features.expunged_quick_edit)
+    self.assertEquals([self.proj1_id, self.proj2_id],
+                      self.services.project_star.expunged_item_ids)
+    self.assertEquals(0, len(self.services.project.test_projects))
+
+
+if __name__ == '__main__':
+  unittest.main()