UndoServiceTest.java

package org.jtheque.undo;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.LoggerFactory;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import javax.annotation.Resource;
import javax.swing.undo.AbstractUndoableEdit;
import javax.swing.undo.CannotRedoException;
import javax.swing.undo.CannotUndoException;
import javax.swing.undo.UndoableEdit;

import java.util.concurrent.atomic.AtomicInteger;

import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.Logger;

import static org.junit.Assert.*;

/*
 * Copyright JTheque (Baptiste Wicht)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "jtheque-undo-test.xml")
public class UndoServiceTest {
    @Resource
    private UndoRedoService undoService;

    static {
        ((Logger) LoggerFactory.getLogger("root")).setLevel(Level.ERROR);
    }

    @Test
    public void initOK() {
        assertNotNull(undoService);
    }

    @Test
    @DirtiesContext
    public void addError() {
        final AtomicInteger undo = new AtomicInteger(0);
        final AtomicInteger redo = new AtomicInteger(0);

        UndoableEdit edit = new TestEdit(undo, redo);

        undoService.addEdit(edit);

        assertEquals(0, undo.intValue());
        assertEquals(0, redo.intValue());

        undoService.undo();

        assertEquals(1, undo.intValue());
        assertEquals(0, redo.intValue());

        undoService.redo();

        assertEquals(1, undo.intValue());
        assertEquals(1, redo.intValue());
    }

    @Test
    @DirtiesContext
    public void listenerCalled() {
        final AtomicInteger counter = new AtomicInteger(0);

        undoService.addStateListener(new StateListener(){
            @Override
            public void stateChanged(String undoName, boolean canUndo, String redoName, boolean canRedo) {
                counter.incrementAndGet();
            }
        });

        undoService.addEdit(new TestEdit(new AtomicInteger(0), new AtomicInteger(0)));

        assertEquals(1, counter.intValue());

        undoService.addEdit(new TestEdit(new AtomicInteger(0), new AtomicInteger(0)));

        assertEquals(2, counter.intValue());

        undoService.undo();

        assertEquals(3, counter.intValue());

        undoService.redo();

        assertEquals(4, counter.intValue());
    }

    @Test
    @DirtiesContext
    public void listenerRemoved() {
        final AtomicInteger counter = new AtomicInteger(0);

        StateListener listener = new StateListener() {
            @Override
            public void stateChanged(String undoName, boolean canUndo, String redoName, boolean canRedo) {
                counter.incrementAndGet();
            }
        };

        undoService.addStateListener(listener);

        undoService.addEdit(new TestEdit(new AtomicInteger(0), new AtomicInteger(0)));

        undoService.removeStateListener(listener);

        undoService.addEdit(new TestEdit(new AtomicInteger(0), new AtomicInteger(0)));

        assertEquals(1, counter.intValue());
    }

    private static class TestEdit extends AbstractUndoableEdit {
        private final AtomicInteger undo;
        private final AtomicInteger redo;

        private TestEdit(AtomicInteger undo, AtomicInteger redo) {
            this.undo = undo;
            this.redo = redo;
        }

        @Override
        public void undo() throws CannotUndoException {
            super.undo();

            undo.incrementAndGet();
        }

        @Override
        public void redo() throws CannotRedoException {
            super.redo();

            redo.incrementAndGet();
        }
    }
}