def upgrade(self):
from Products.BTreeFolder2.BTreeFolder2 import (
BTreeFolder2Base as BTreeFolder)
from plone.app.folder.migration import BTreeMigrationView
self.folder_class = BTreeFolder
self.folder_migrator = BTreeMigrationView(self.context, None)
super(CMFEditionsFolderMigrator, self).upgrade()
def testBTreeMigration(self):
# create (unmigrated) btree folder
folder = self.makeUnmigratedFolder(self.portal, 'test', title='Foo')
view = BTreeMigrationView(self.portal, self.app.REQUEST)
self.failUnless(view.migrate(folder))
folder = self.portal.test # get the object again...
self.failUnless(isSaneBTreeFolder(folder))
self.assertEqual(folder.getId(), 'test')
self.assertEqual(folder.Title(), 'Foo')
# a second migration should be skipped
self.failIf(view.migrate(folder))
