def afterSetUp(self):
super(TestMigrate, self).afterSetUp()
self.version = Version(9, 9, 9)
self.migration = Migration()
self.oldCurrentVersion = self.migration._currentVersion
self.migration._currentVersion = self.getTestVersion
def testDetermineSteps(self):
m = Migration(noopts=True)
m.allSteps = [
MyTestStep(1, 0, 0),
MyTestStep(1, 1, 0),
MyTestStep(1, 2, 0),
]
m.options.level = '1.1.0'
self.assertEquals(m.determineSteps(), m.allSteps[1:])
m.options.level = None
m.options.steps = ['MyTestStep_1.1.0']
self.assertEquals(m.determineSteps(), m.allSteps[1:2])
def testGetEarliestAppropriateStepVersion(self):
m = Migration(noopts=True)
self.assertEquals(
Version(1, 0, 70),
m.getEarliestAppropriateStepVersion(codeVers=Version(1, 1, 50)))
self.assertEquals(
Version(1, 1, 70),
m.getEarliestAppropriateStepVersion(codeVers=Version(1, 1, 70)))
self.assertEquals(
Version(1, 1, 70),
m.getEarliestAppropriateStepVersion(codeVers=Version(1, 1, 99)))
self.assertEquals(
Version(1, 1, 70),
m.getEarliestAppropriateStepVersion(codeVers=Version(1, 2, 0)))
m.allSteps.append(MyTestStep(98, 3, 71))
self.assertEquals(
Version(98, 3, 70),
m.getEarliestAppropriateStepVersion(codeVers=Version(99, 0, 1)))
