def testErrorsOnLastBrokenCodebase(self):
mock_db = test_util.MockDbClient(migration_id_seed=88)
mock_editor = test_util.MockEditor()
mock_client = test_util.MockClient(
lambda migration_strategy, revisions: mock_editor)
report = base.MoeReport()
project = test_util.EmptyMoeProjectConfig()
internal_creator = test_util.StaticCodebaseCreator(
{
'1001': 'simple_python',
'1002': None,
}, lambda: mock_client)
public_creator = test_util.StaticCodebaseCreator(
{'1': 'simple_python'})
config = actions.MigrationConfig(base.Migration.EXPORT,
internal_creator,
project.internal_repository_config,
public_creator,
project.public_repository_config,
None, project.export_strategy)
revisions = [base.Revision('1002', changelog='1002')]
action = actions.Migration(base.Revision('1001'), base.Revision('1'),
revisions, project, [], config, False, 1)
self.assertRaises(base.CodebaseCreationError,
action.Perform,
db=mock_db)
def testSingleImport(self):
mock_db = test_util.MockDbClient(migration_id_seed=88)
mock_editor = test_util.MockEditor()
mock_client = test_util.MockClient(
lambda migration_strategy, revisions: mock_editor)
project = test_util.EmptyMoeProjectConfig()
internal_creator = test_util.StaticCodebaseCreator(
{'1001': 'simple_python'}, lambda: mock_client)
public_creator = test_util.StaticCodebaseCreator({
'1': 'simple_python',
'2': 'simple_python2'
})
config = actions.MigrationConfig(base.Migration.IMPORT, public_creator,
project.public_repository_config,
internal_creator,
project.internal_repository_config,
None, project.import_strategy)
revisions = [base.Revision('2', changelog='log')]
action = actions.Migration(base.Revision('1'), base.Revision('1001'),
revisions, project, [], config, False, -1)
result = action.Perform(db=mock_db)
self.assertFalse(result)
self.assertTrue(mock_editor.ChangesMade())
expected = base.Migration(migration_id='88',
direction=base.Migration.IMPORT,
status=base.Migration.ACTIVE,
up_to_revision=base.Revision('2'),
changelog='log',
revisions=revisions)
self.assertEqual(mock_db.GetMigration('88').Dict(), expected.Dict())
def testDivideMultipleChangelogs(self):
mock_db = test_util.MockDbClient()
mock_editor = test_util.MockEditor()
mock_client = test_util.MockClient(
lambda migration_strategy, revisions: mock_editor)
project = test_util.EmptyMoeProjectConfig()
project.export_strategy.separate_revisions = True
internal_creator = test_util.StaticCodebaseCreator({
'1001':
'simple_python',
'1002':
'simple_python2',
'1003':
'simple_python',
'1004':
'simple_python2'
})
public_creator = test_util.StaticCodebaseCreator(
{'1': 'simple_python'}, lambda: mock_client)
config = actions.MigrationConfig(base.Migration.EXPORT,
internal_creator,
project.internal_repository_config,
public_creator,
project.public_repository_config,
None, project.export_strategy)
revisions = [
base.Revision('1002',
changelog='Change 1: a change',
single_scrubbed_log='a change'),
base.Revision('1003',
changelog='Change 2: another change',
single_scrubbed_log='another change'),
base.Revision('1004',
changelog='Change 3: something',
single_scrubbed_log='something')
]
action = actions.Migration(base.Revision('1001'), base.Revision('1'),
revisions, project, [], config, False, 1)
result = action.Perform(db=mock_db)
self.assertEqual(len(result.actions), 1)
self.assertEqual(
mock_db.GetMigration('1', abbreviated=False).changelog, 'a change')
result = result.actions[0].Perform(db=mock_db)
self.assertEqual(len(result.actions), 1)
self.assertEqual(
mock_db.GetMigration('2', abbreviated=False).changelog,
'another change')
result = result.actions[0].Perform(db=mock_db)
self.assertEqual(result, None)
self.assertEqual(
mock_db.GetMigration('3', abbreviated=False).changelog,
'something')
def testSingleExportCommit(self):
mock_db = test_util.MockDbClient(migration_id_seed=88)
mock_editor = test_util.MockEditor(commit_id='2')
mock_client = test_util.MockClient(
lambda migration_strategy, revisions: mock_editor)
report = base.MoeReport()
project = test_util.EmptyMoeProjectConfig()
project.public_repository_config = test_util.MockRepositoryConfig(
'', repository=test_util.MockRepository(''))
internal_creator = test_util.StaticCodebaseCreator({
'1001':
'simple_python',
'1002':
'simple_python2'
})
public_creator = test_util.StaticCodebaseCreator(
{'1': 'simple_python'}, lambda: mock_client)
config = actions.MigrationConfig(
base.Migration.EXPORT, internal_creator,
project.internal_repository_config, public_creator,
project.public_repository_config,
project.public_repository_config.MakeRepository()[0],
project.export_strategy)
revisions = [base.Revision('1002', changelog='log')]
action = actions.Migration(base.Revision('1001'), base.Revision('1'),
revisions, project, [], config, False, -1)
result = action.Perform(db=mock_db)
equivalence_check = result.actions[0]
self.assertEqual(equivalence_check.internal_revision, '1002')
self.assertEqual(equivalence_check.public_revision, '2')
self.assertFalse(result.actions[1:])
self.assertTrue(mock_editor.ChangesMade())
expected = base.Migration(
migration_id='88',
direction=base.Migration.EXPORT,
status=base.Migration.SUBMITTED,
up_to_revision=base.Revision('1002'),
revisions=revisions,
submitted_as=base.Revision('2'),
changelog='log',
)
self.assertEqual(mock_db.GetMigration('88').Dict(), expected.Dict())
def testSkipsBrokenCodebase(self):
mock_db = test_util.MockDbClient(migration_id_seed=88)
mock_editor = test_util.MockEditor()
mock_client = test_util.MockClient(
lambda migration_strategy, revisions: mock_editor)
project = test_util.EmptyMoeProjectConfig()
internal_creator = test_util.StaticCodebaseCreator(
{
'1001': 'simple_python',
'1002': None,
'1003': 'simple_python2',
}, lambda: mock_client)
public_creator = test_util.StaticCodebaseCreator(
{'1': 'simple_python'})
config = actions.MigrationConfig(base.Migration.EXPORT,
internal_creator,
project.internal_repository_config,
public_creator,
project.export_strategy, None,
project.public_repository_config)
revisions = [
base.Revision('1002', changelog='1002'),
base.Revision('1003', changelog='1003')
]
action = actions.Migration(base.Revision('1001'), base.Revision('1'),
revisions, project, [], config, False, 1)
result = action.Perform(db=mock_db)
migration = result.actions[0]
migrated_revisions = migration.revisions
# because we couldn't migrate 1002 on its own, we now want to migrate
# 1002 *and* 1003
self.assertEqual(len(migrated_revisions), 2)
self.assertEqual(migrated_revisions[0].rev_id, '1002')
self.assertEqual(migrated_revisions[1].rev_id, '1003')