package apple;

import org.testng.Assert;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

public class stringProblemTest {

@DataProvider(name= "inputPositiveTesting")

public Object[][] generateDataSetPositive() {

return new Object [][]{

new Object[] {"cantalope",'a',"cntlope","iterate"},

new Object[] {"cantalope",'a',"cntlope","oneLine"}

};

}

@DataProvider(name= "inputNegativeTesting")

public Object[][] generateDataSetNegative() {

return new Object [][]{

new Object[] {"cantalope",'a',"ctlope","iterate"},

new Object[] {"cantalope",'a',"ctlope","oneLine"}

};

}

@Test(dataProvider = "inputPositiveTesting")

public void testUserMapping(String word,char charRemove, String result, String method){

Assert.assertEquals(stringExample(word,charRemove, method), result);

}

@Test(dataProvider = "inputNegativeTesting")

public void testUserMappingNeg(String word,char charRemove, String result, String method){

Assert.assertNotEquals(stringExample(word,charRemove, method), result);

}

public String stringExample(String word, char charRemove, String method) {

String result="";

if(method == "iterate"){

char[] charList = word.toCharArray();

for(char c:charList){

if(c != charRemove){

result = result + c;

}

}

} else if(method == "oneLine"){

result = word.replaceAll(Character.toString(charRemove), "");

}

return result;

}

}