001package bradleyross.library.helpers.tests; 002import java.io.File; 003import java.io.FileNotFoundException; 004import java.io.PrintWriter; 005import java.sql.SQLException; 006import java.sql.ResultSet; 007import bradleyross.library.helpers.SpreadsheetHelpers; 008import bradleyross.local.databases.DatabaseConnection; 009import bradleyross.library.database.DatabaseProperties; 010/** 011 * Test driver for bradleyross.library.helpers.SpreadsheetHelpers. 012 * @author Bradley Ross 013 * @see bradleyross.library.helpers.SpreadsheetHelpers 014 * 015 */ 016public class TestSpreadsheetHelpers 017{ 018 /** 019 * Main driver 020 * @param args First argument is output file destination 021 */ 022 public static void main(String[] args) 023 { 024 File outputFile = null; 025 PrintWriter out = null; 026 if (args.length == 0) 027 { 028 outputFile = new File ("spreadsheet.xml"); 029 } 030 else 031 { 032 outputFile = new File(args[0]); 033 } 034 035 System.out.println("Starting program"); 036 DatabaseProperties data = new DatabaseConnection("sample"); 037 System.out.println(data.showAttributes()); 038 SpreadsheetHelpers helper = new SpreadsheetHelpers(); 039 try 040 { 041 data.connect(); 042 } 043 catch (SQLException e) 044 { 045 System.out.println(e.getClass().getName() + e.getMessage()); 046 } 047 try 048 { 049 String sqlCode = "SELECT * FROM " + data.prefixTableName("STATES"); 050 ResultSet rs = data.executeQuery(sqlCode); 051 out = new PrintWriter(outputFile); 052 out.println(helper.startWorkbook()); 053 out.println(helper.newWorksheet("first", rs)); 054 out.println(helper.endWorkbook()); 055 if (out.checkError()) 056 { 057 System.out.println("Error occurred while writing file"); 058 } 059 else 060 { 061 out.close(); 062 } 063 } 064 catch (FileNotFoundException e) 065 { 066 System.out.println(e.getClass().getName() + " " + e.getMessage()); 067 } 068 catch (SQLException e) 069 { 070 System.out.println(e.getClass().getName() + " " + e.getMessage()); 071 } 072 try 073 { 074 data.close(); 075 } 076 catch (SQLException e) 077 { 078 System.out.println(e.getClass().getName() + " " + e.getMessage()); 079 } 080 } 081}