aboutsummaryrefslogtreecommitdiffstats
path: root/src/test/java/junitparams/IteratorMethodTest.java
blob: 219643f8b29bf40aa75dbdd957fd545704bc5094 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package junitparams;

import static org.junit.Assert.assertEquals;

import java.util.*;

import org.junit.*;
import org.junit.runner.*;

@RunWith(JUnitParamsRunner.class)
public class IteratorMethodTest {
	private static boolean firstCalled = false;
	
    @Test
    @Parameters
    public void shouldHandleIteratorsWithObject(String a) {
    	assertEquals(1, a.length());
    	if(a.equals("a")) firstCalled = true;
    	else assertEquals(firstCalled, true);
    }

    public Iterator<Object> parametersForShouldHandleIteratorsWithObject() {
        ArrayList<Object> params = new ArrayList<Object>();
        params.add("a");
        params.add("b");
        return params.iterator();
    }
	
    @Test
    @Parameters
    public void shouldHandleIteratorsWithObjectArray(String a) {
    	assertEquals(1, a.length());
    	if(a.equals("a")) firstCalled = true;
    	else assertEquals(firstCalled, true);
    }

    public Iterator<Object[]> parametersForShouldHandleIteratorsWithObjectArray() {
        ArrayList<Object[]> params = new ArrayList<Object[]>();
        params.add(new Object[]{"a"});
        params.add(new Object[]{"b"});
        return params.iterator();
    }
    
}