Changeset 2384 in products for quintagroup.plonegooglesitemaps/trunk/quintagroup/plonegooglesitemaps/tests/testUpgrade.py

Timestamp:

May 25, 2010 6:17:59 PM (14 years ago)

Author:

mylan

Message:

#250: Fixed and added test for reinstallation

File:

• quintagroup.plonegooglesitemaps/trunk/quintagroup/plonegooglesitemaps/tests/testUpgrade.py (modified) (4 diffs)

quintagroup.plonegooglesitemaps/trunk/quintagroup/plonegooglesitemaps/tests/testUpgrade.py

@@ -6 +6 @@
 from Products.CMFPlone.utils import _createObjectByType
 from Products.GenericSetup.upgrade import _upgrade_registry
+from quintagroup.plonegooglesitemaps import config
+from quintagroup.plonegooglesitemaps import upgrades as gsm_upgrades
 from quintagroup.canonicalpath.interfaces import ICanonicalPath
 from quintagroup.canonicalpath.interfaces import ICanonicalLink
@@ -43 +45 @@
         if step is not None:
             step.doStep(self.setup)
-        # canonical_path column must be added to portal_catalog
+        # Canonical_path column must be added to portal_catalog
         self.assertEqual("canonical_path" in catalog._catalog.names, True)
 
@@ -68 +70 @@
         self.assertEqual(brain.canonical_link, migrated_link)
 
+    def testUpgradeCallOnReinstall(self):
+        # Get upgrade steps
+        upgrades = _upgrade_registry.getUpgradeStepsForProfile(self.profile)
+        upgrades = dict([(u.sortkey, u) for u in upgrades.values()])
+        try:
+            # Replace original handlers with patched ones for test calls
+            called = []
+            upgrades[1].handler = lambda st:called.append("1.0 to 1.1")
+            upgrades[2].handler = lambda st:called.append("1.1 to 1.2")
+            # Run reinstallation
+            self.portal.portal_quickinstaller.reinstallProducts(products=config.PROJECTNAME)
+            # Test upgrades call
+            self.assertEqual("1.0 to 1.1" in called, True)
+            self.assertEqual("1.1 to 1.2" in called, True)
+        finally:
+            # Restore original upgrade handlers
+            upgrades[1].handler = gsm_upgrades.upgrade_1_0_to_1_1
+            upgrades[2].handler = gsm_upgrades.upgrade_1_1_to_1_2
+
+        
 
 def test_suite():
@@ -77 +99 @@
 if __name__ == '__main__':
     unittest.main(defaultTest='test_suite')
-#    framework()