Skip to content
Snippets Groups Projects
CSVParserTest.java 2.92 KiB
Newer Older
  • Learn to ignore specific revisions
  • HMoss's avatar
    HMoss committed
    package edu.ncsu.csc.itrust.unit;
    
    import java.util.ArrayList;
    import java.util.Scanner;
    
    import edu.ncsu.csc.itrust.CSVParser;
    import edu.ncsu.csc.itrust.exception.CSVFormatException;
    import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
    import junit.framework.TestCase;
    
    public class CSVParserTest extends TestCase {
    
    	public void testCSVParseHeader(){
    		Scanner CSVHeader=new Scanner("test1,test2,test3");
    		CSVParser test=null;
    		try{
    			test=new CSVParser(CSVHeader);
    		}catch(Exception e){
    			assertTrue(false);
    		}
    		ArrayList<String> header=test.getHeader();
    		assertEquals(3, header.size());
    		assertEquals("test1", header.get(0));
    		assertEquals("test2", header.get(1));
    		assertEquals("test3", header.get(2));
    		assertFalse(test.getErrors().hasErrors());
    	}
    	
    	public void testCSVParseData(){
    		Scanner CSVHeader=new Scanner("test1,test2,test3\ntesting,testing,123");
    		CSVParser test=null;
    		try{
    			test=new CSVParser(CSVHeader);
    		}catch(Exception e){
    			assertTrue(false);
    		}
    		ArrayList<ArrayList<String>> data=test.getData();
    		assertEquals(1, data.size());
    		assertEquals(3, data.get(0).size());
    		assertEquals("testing", data.get(0).get(0));
    		assertEquals("testing", data.get(0).get(1));
    		assertEquals("123", data.get(0).get(2));
    		assertFalse(test.getErrors().hasErrors());
    	}
    	@SuppressFBWarnings(value="DLS_DEAD_LOCAL_STORE")
    	public void testEmptyCSV(){
    		Scanner CSVHeader=new Scanner("");
    		@SuppressWarnings("unused")
    		CSVParser test=null;
    		try{
    			test=new CSVParser(CSVHeader);
    		}catch(CSVFormatException e){
    			//CSVFormatException is good here
    			return;
    		}catch(Exception e){
    			//Other exceptions are bad
    			assertTrue(false);
    		}
    	}
    	
    	public void testWrongNumberFields(){
    		Scanner CSVHeader=new Scanner("test1,test2,test3\ntesting,123");
    		CSVParser test=null;
    		try{
    			test=new CSVParser(CSVHeader);
    		}catch(Exception e){
    			assertTrue(false);
    		}
    		ArrayList<ArrayList<String>> data=test.getData();
    		assertEquals(0, data.size());
    		assertTrue(test.getErrors().hasErrors());
    		assertEquals("[Field number mismatch on line 2]", test.getErrors().toString());
    	}
    	
    	public void testCSVCommaInField(){
    		Scanner CSVHeader=new Scanner("\"test,1\",\"test,2\",\"test,3\"");
    		CSVParser test=null;
    		try{
    			test=new CSVParser(CSVHeader);
    		}catch(Exception e){
    			assertTrue(false);
    		}
    		ArrayList<String> header=test.getHeader();
    		assertEquals(3, header.size());
    		assertEquals("test,1", header.get(0));
    		assertEquals("test,2", header.get(1));
    		assertEquals("test,3", header.get(2));
    		assertFalse(test.getErrors().hasErrors());
    	}
    	
    	public void testCSVEndsWhileInQuote(){
    		Scanner CSVHeader=new Scanner("test1,test2,test3\ntesting,testing,\"123");
    		CSVParser test=null;
    		try{
    			test=new CSVParser(CSVHeader);
    		}catch(Exception e){
    			assertTrue(false);
    		}
    		ArrayList<ArrayList<String>> data=test.getData();
    		assertEquals(0, data.size());
    		assertEquals("[Line ended while inside quotes on line 2]", test.getErrors().toString());
    	}
    }