StackTest.java

package com.coding.basic.stack;

import java.util.EmptyStackException;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

/**
 * KStack测试
 *
 * Created by Korben on 18/02/2017.
 */
public class StackTest {
    private Stack<Integer> stack;
    private int initTestSize;

    @Before
    public void init() {
        stack = new Stack<>();
        initTestSize = 5;

        for (int i = 0; i < initTestSize; i++) {
            stack.push(i);
        }
    }

    @Test
    public void size() throws Exception {
        Assert.assertEquals(initTestSize, stack.size());
    }

    @Test
    public void push() throws Exception {
        stack.push(9);
    }

    @Test(expected = EmptyStackException.class)
    public void pop() throws Exception {
        for (int i = 0; i < initTestSize; i++) {
            Integer value = stack.pop();
            Assert.assertEquals(value.intValue(), initTestSize - 1 - i);
        }

        stack.pop();
    }

    @Test(expected = EmptyStackException.class)
    public void peek() throws Exception {
        Assert.assertEquals(initTestSize - 1, stack.peek().intValue());
        for (int i = 0; i < initTestSize; i++) {
            stack.pop();
        }
        stack.peek();
    }

    @Test
    public void empty() throws Exception {
        Assert.assertFalse(stack.isEmpty());
        for (int i = 0; i < initTestSize; i++) {
            stack.pop();
        }
        Assert.assertTrue(stack.isEmpty());
    }

    @Test
    public void search() throws Exception {
        for (int i = 0; i < initTestSize; i++) {
            Assert.assertEquals(i, stack.search(i));
        }
    }
}