[CMF-checkins] CVS: CMF/CMFStaging/tests - testStaging.py:1.2

[Shane Hathaway]

Update of /cvs-repository/CMF/CMFStaging/tests
In directory cvs.zope.org:/tmp/cvs-serv26014/tests

Modified Files:
	testStaging.py 
Log Message:
Expanded tests and corrected the use of _delOb() with _setObject();
this code has to use _delObject().


=== CMF/CMFStaging/tests/testStaging.py 1.1.1.1 => 1.2 ===
 from OFS.Application import Application
 
-from Products.CMFStaging.StagingTool import StagingTool
+from Products.CMFStaging.StagingTool import StagingTool, StagingError
+from Products.ZopeVersionControl.Utility import VersionControlError
 
 
 class Tests(unittest.TestCase):
@@ -31,7 +32,7 @@
     def setUp(self):
         # Set up an application with a repository, 3 stages, the tools,
         # and a little content in the repository.
-        # Note that you don't actually need a CMF site. ;-)
+        # Note that you don't actually need a CMF site to use staging! ;-)
         app = Zope.app()
         self.app = app
         if hasattr(app, 'testroot'):
@@ -50,19 +51,25 @@
         dev_stage = self.stages.Development
         dev_stage.manage_addProduct['OFSP'].manage_addFolder('c1')
         dev_stage.manage_addProduct['OFSP'].manage_addFolder('c2')
+        dev_stage.manage_addProduct['OFSP'].manage_addFolder('nonv')
         self.dev_stage = dev_stage
+        self.review_stage = self.stages.Review
+        self.prod_stage = self.stages.Production
 
         repo = self.root.VersionRepository
         repo.applyVersionControl(dev_stage.c1)
         repo.applyVersionControl(dev_stage.c2)
 
+
     def tearDown(self):
         self.app._p_jar.close()
 
+
     def testStageable(self):
         st = self.root.portal_staging
         self.assert_(st.isStageable(self.dev_stage.c1))
 
+
     def testGetVersionIds(self):
         st = self.root.portal_staging
         versions = st.getVersionIds(self.dev_stage.c1)
@@ -70,27 +77,81 @@
         self.assert_(not versions['review'])
         self.assert_(not versions['prod'])
 
+        # Can't get version IDs of something not located in any stage
+        self.assertRaises(StagingError, st.getVersionIds, self.root)
+
+
     def testUpdateStages(self):
         st = self.root.portal_staging
+        self.assert_('c1' not in self.review_stage.objectIds())
 
         st.updateStages(self.dev_stage.c1, 'dev', ['review'])
         versions = st.getVersionIds(self.dev_stage.c1)
         self.assert_(versions['dev'])
         self.assert_(versions['review'])
         self.assert_(not versions['prod'])
+        self.assert_('c1' in self.review_stage.objectIds())
+        self.assert_('c1' not in self.prod_stage.objectIds())
 
         st.updateStages(self.dev_stage.c2, 'dev', ['review', 'prod'])
         versions = st.getVersionIds(self.dev_stage.c2)
         self.assert_(versions['dev'])
         self.assert_(versions['review'])
         self.assert_(versions['prod'])
-        
+
         st.updateStages(self.dev_stage.c1, 'dev', ['prod'])
         versions = st.getVersionIds(self.dev_stage.c1)
         self.assert_(versions['dev'])
         self.assert_(versions['review'])
         self.assert_(versions['prod'])
+        self.assert_('c1' in self.prod_stage.objectIds())
+
+
+    def testUpdateStagesExceptions(self):
+        st = self.root.portal_staging
+        # "nonv" is not under version control.
+        self.assertRaises(VersionControlError, st.updateStages,
+                          self.dev_stage.nonv, 'dev', ['review'])
+        # Put something in the way and make sure it doesn't get overwritten.
+        self.review_stage.manage_addProduct['OFSP'].manage_addFolder('c1')
+        self.assertRaises(StagingError, st.updateStages, self.dev_stage.c1,
+                          'dev', ['review'])
+        # Put the blocker under version control and verify it still doesn't
+        # get overwritten, since it is backed by a different version history.
+        repo = self.root.VersionRepository
+        repo.applyVersionControl(self.review_stage.c1)
+        self.assertRaises(StagingError, st.updateStages, self.dev_stage.c1,
+                          'dev', ['review'])
+        # Move the blocker out of the way and verify updates can occur again.
+        self.review_stage._delObject('c1')
+        st.updateStages(self.dev_stage.c1, 'dev', ['review'])
+        
+
+    def testRemoveStages(self):
+        st = self.root.portal_staging
+        self.assert_('c1' not in self.review_stage.objectIds())
+        st.updateStages(self.dev_stage.c1, 'dev', ['review'])
+        self.assert_('c1' in self.review_stage.objectIds())
+        st.removeStages(self.dev_stage.c1, ['review'])
+        self.assert_('c1' not in self.review_stage.objectIds())
         
+
+    def testCheckContainer(self):
+        st = self.root.portal_staging
+        self.dev_stage.c1.manage_addProduct['OFSP'].manage_addFolder(
+            'test')
+        self.assert_('c1' not in self.review_stage.objectIds())
+        # c1 is ok
+        st.checkContainer(self.dev_stage.c1, 'review')
+        # c1 doesn't exist on the review stage yet, so updates
+        # to c1.test can't work yet.
+        self.assertRaises(StagingError, st.checkContainer,
+                          self.dev_stage.c1.test, 'review')
+
+    def testGetURLForStage(self):
+        st = self.root.portal_staging
+        url = st.getURLForStage(self.dev_stage.c1, 'dev', 1)
+        self.assert_(url.find('/Stages/Development/c1') >= 0)
 
 
 def test_suite():