def init_commands(self):
self.cli = Application(orator_application.get_name(),
orator_application.get_version())
self.cli.add(InstallCommand(self))
self.cli.add(MigrateCommand(self))
self.cli.add(MigrateMakeCommand(self))
self.cli.add(RollbackCommand(self))
self.cli.add(StatusCommand(self))
self.cli.add(ResetCommand(self))
def test_migration_database_can_be_set(self):
resolver = flexmock(DatabaseManager)
resolver.should_receive('connection').and_return(None)
migrator_mock = flexmock(Migrator)
migrator_mock.should_receive('set_connection').once().with_args('foo')
migrator_mock.should_receive('rollback').once().with_args(os.path.join(os.getcwd(), 'migrations'), False)
migrator_mock.should_receive('get_notes').and_return([])
command = flexmock(RollbackCommand())
command.should_receive('_get_config').and_return({})
self.run_command(command, [('--database', 'foo')], input_stream=self.get_input_stream('y\n'))
def test_basic_migrations_call_migrator_with_proper_arguments(self):
resolver = flexmock(DatabaseManager)
resolver.should_receive('connection').and_return(None)
migrator_mock = flexmock(Migrator)
migrator_mock.should_receive('set_connection').once().with_args(None)
migrator_mock.should_receive('rollback').once().with_args(os.path.join(os.getcwd(), 'migrations'), False)
migrator_mock.should_receive('get_notes').and_return([])
command = flexmock(RollbackCommand())
command.should_receive('_get_config').and_return({})
self.run_command(command, input_stream=self.get_input_stream('y\n'))
def test_migration_can_be_forced(self):
resolver = flexmock(DatabaseManager)
resolver.should_receive("connection").and_return(None)
migrator_mock = flexmock(Migrator)
migrator_mock.should_receive("set_connection").once().with_args(None)
migrator_mock.should_receive("rollback").once().with_args(
os.path.join(os.getcwd(), "migrations"), False)
migrator_mock.should_receive("get_notes").and_return([])
command = flexmock(RollbackCommand())
command.should_receive("_get_config").and_return({})
self.run_command(command, [("--force", True)])
def test_migration_can_be_forced(self):
resolver = flexmock(DatabaseManager)
resolver.should_receive('connection').and_return(None)
migrator_mock = flexmock(Migrator)
migrator_mock.should_receive('set_connection').once().with_args(None)
migrator_mock.should_receive('rollback').once().with_args(
os.path.join(os.getcwd(), 'migrations'), False)
migrator_mock.should_receive('get_notes').and_return([])
command = flexmock(RollbackCommand())
command.should_receive('_get_config').and_return({})
self.run_command(command, [('--force', True)])
def test_basic_migrations_call_migrator_with_proper_arguments(self):
resolver = flexmock(DatabaseManager)
resolver.should_receive("connection").and_return(None)
migrator_mock = flexmock(Migrator)
migrator_mock.should_receive("set_connection").once().with_args(None)
migrator_mock.should_receive("rollback").once().with_args(
os.path.join(os.getcwd(), "migrations"), False)
migrator_mock.should_receive("get_notes").and_return([])
command = flexmock(RollbackCommand())
command.should_receive("_get_config").and_return({})
command.should_receive("confirm").and_return(True)
self.run_command(command)