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}