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)