package com.myapp;

import junit.framework.TestCase;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;

import java.util.Arrays;
import java.util.Collection;


@RunWith(Parameterized.class)
public class PrimeNumberCheckerTest extends TestCase {

    private int inputNum;
    private Boolean result;
    private PrimeNumberChecker primeNumberChecker;


    @Before
    public void initialize() {
        primeNumberChecker = new PrimeNumberChecker();

    }



    public PrimeNumberCheckerTest(int inputNum, boolean result) {
        this.inputNum = inputNum;
        this.result = result;
    }



    @Parameterized.Parameters
    public static Collection primeNumbers() {
//        return Arrays.asList(1,2,3,4,5);
        return Arrays.asList(new Object[][] {
                { 2, true },
                { 6, false },
                { 19, true },
                { 22, false },
                { 23, true }
        });
    }


    @Test
    public void testPrimeNumberChecker() {
        Assert.assertEquals(result, primeNumberChecker.validate(inputNum));
    }


}