dumps the content of blobs, and reworks table formatting to add spaces around lines
[squelch.git] / src / test / java / net / jaekl / squelch / stmt / TabularTest.java
index 6b5462dac3f00320f1a4b0b4cbdeebc0344fffbb..b72add390ab0aefcc64d8c59e20a96c0d1f71ff2 100644 (file)
@@ -1,28 +1,37 @@
 package net.jaekl.squelch.stmt;
 
-import static org.junit.Assert.*;
+import static org.junit.Assert.assertEquals;
 
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.OutputStreamWriter;
+import java.io.PrintWriter;
+import java.nio.charset.StandardCharsets;
 import java.sql.SQLException;
 import java.sql.Types;
 
+import javax.sql.rowset.serial.SerialBlob;
+import javax.sql.rowset.serial.SerialException;
+
+import junit.framework.Assert;
+
 import net.jaekl.squelch.sql.Column;
 import net.jaekl.squelch.sql.Row;
 
 import org.junit.Test;
 
 public class TabularTest {
-       private static class TabularMock extends Tabular {
-               @Override
-               Column[] getCols() throws SQLException {
-                       return null;
-               }
-
-               @Override
-               Row getNext() {
-                       return null;
-               }
+       @Test
+       public void test_centrePad() {
+               Tabular tabular = new TabularMock();
+               
+               assertEquals("Vestibule", tabular.centrePad("Vestibule", 2));
+               assertEquals("  Fred  ", tabular.centrePad("Fred", 8));
+               assertEquals("NULL", tabular.centrePad(null, 0));
+               assertEquals("  NULL  ", tabular.centrePad(null, 8));
+               assertEquals("   Wilma    ", tabular.centrePad("Wilma", 12));
        }
-       
+
        @Test
        public void test_classForSqlType() {
                Object[][] data = { 
@@ -70,7 +79,53 @@ public class TabularTest {
                        assertEquals(expected, actual);
                }
        }
+       
+       @Test
+       public void test_printCsv_empTable() throws IOException, SQLException
+       {
+               TabularMock tabular = createEmpTable();
+               
+               try (
+                               ByteArrayOutputStream baos = new ByteArrayOutputStream();
+                               PrintWriter pw = new PrintWriter(new OutputStreamWriter(baos, StandardCharsets.UTF_8));
+                       )
+               {
+                       tabular.printCsv(pw);
+                       pw.close();
+                       baos.close();
+                       String actual = baos.toString();
+                       assertEquals(  "EmpId,FirstName,LastName\n"
+                                            + "12345,Fred,Flintstone\n"
+                                            + "7654321,Barney,Rubble\n",
+                                            actual);
+               }               
+       }
 
+       @Test
+       public void test_printTable_empTable() throws IOException, SQLException
+       {
+               TabularMock tabular = createEmpTable();
+               
+               try (
+                               ByteArrayOutputStream baos = new ByteArrayOutputStream();
+                               PrintWriter pw = new PrintWriter(new OutputStreamWriter(baos, StandardCharsets.UTF_8));
+                       )
+               {
+                       tabular.printTable(pw, "No rows returned.");
+                       pw.close();
+                       baos.close();
+                       String actual = baos.toString();
+                       assertEquals(  "+---------+-----------+------------+\n"
+                                            + "|  EmpId  | FirstName |  LastName  |\n"
+                                            + "+---------+-----------+------------+\n"
+                                            + "| 12345   | Fred      | Flintstone |\n"
+                                            + "| 7654321 | Barney    | Rubble     |\n"
+                                            + "+---------+-----------+------------+\n"
+                                            + "2 row(s) returned.\n",
+                                            actual);
+               }
+       }
+       
        @Test
        public void test_repChar() {
                Tabular tabular = new TabularMock();
@@ -80,15 +135,77 @@ public class TabularTest {
                assertEquals("###", tabular.repChar('#', 3));
                assertEquals("------", tabular.repChar('-', 6));
        }
-
+       
        @Test
-       public void test_centrePad() {
+       public void test_stringify() throws SerialException, SQLException {
                Tabular tabular = new TabularMock();
                
-               assertEquals("Vestibule", tabular.centrePad("Vestibule", 2));
-               assertEquals("  Fred  ", tabular.centrePad("Fred", 8));
-               assertEquals("NULL", tabular.centrePad(null, 0));
-               assertEquals("  NULL  ", tabular.centrePad(null, 8));
-               assertEquals("   Wilma    ", tabular.centrePad("Wilma", 12));
+               String[] data = {
+                       "This is the way the world ends, not with a bang but a whimper.",
+                       
+                       "To be, or not to be, that is the question:\n" +
+                       "Whether 'tis Nobler in the mind to suffer\n" +
+                       "The Slings and Arrows of outrageous Fortune,\n" +
+                       "Or to take Arms against a Sea of troubles,\n" +
+                       "And by opposing end them: to die, to sleep\n" +
+                       "No more; and by a sleep, to say we end\n" +
+                       "The Heart-ache, and the thousand Natural shocks\n" +
+                       "That Flesh is heir to? 'Tis a consummation\n" +
+                       "Devoutly to be wished. To die, to sleep,\n" +
+                       "To sleep, perchance to Dream; aye, there's the rub,\n" +
+                       "For in that sleep of death, what dreams may come,\n" +
+                       "When we have shuffled off this mortal coil,\n" +
+                       "Must give us pause. There's the respect\n" +
+                       "That makes Calamity of so long life:\n",
+                       
+                       "ἄνδρα μοι ἔννεπε, μοῦσα, πολύτροπον, ὃς μάλα πολλὰ\n" +
+                       "πλάγχθη, ἐπεὶ Τροίης ἱερὸν πτολίεθρον ἔπερσεν:\n" +
+                       "πολλῶν δ᾽ ἀνθρώπων ἴδεν ἄστεα καὶ νόον ἔγνω,\n" +
+                       "πολλὰ δ᾽ ὅ γ᾽ ἐν πόντῳ πάθεν ἄλγεα ὃν κατὰ θυμόν,\n" +
+                       "5ἀρνύμενος ἥν τε ψυχὴν καὶ νόστον ἑταίρων.\n" +
+                       "ἀλλ᾽ οὐδ᾽ ὣς ἑτάρους ἐρρύσατο, ἱέμενός περ:\n" +
+                       "αὐτῶν γὰρ σφετέρῃσιν ἀτασθαλίῃσιν ὄλοντο,\n" +
+                       "νήπιοι, οἳ κατὰ βοῦς Ὑπερίονος Ἠελίοιο\n" +
+                       "ἤσθιον: αὐτὰρ ὁ τοῖσιν ἀφείλετο νόστιμον ἦμαρ.\n" +
+                       "10τῶν ἁμόθεν γε, θεά, θύγατερ Διός, εἰπὲ καὶ ἡμῖν.\n"
+               };
+               
+               for (String datum : data) 
+               {
+                       byte[] content = datum.getBytes(StandardCharsets.UTF_8);
+                       SerialBlob sblob = new SerialBlob(content);
+                       
+                       String expected = datum;
+                       String actual = tabular.stringify(sblob);
+                       
+                       Assert.assertEquals(expected, actual);
+               }
+       }
+       
+       
+       private TabularMock createEmpTable()
+       {
+               TabularMock tabular = new TabularMock();
+               
+               Column[] cols = { 
+                               new Column("EmpId", Long.class, 10),
+                               new Column("FirstName", String.class, 14),
+                               new Column("LastName", String.class, 14)
+               };
+               tabular.mock_setCols(cols);
+               
+               Row row = new Row(cols.length);
+               row.setValue(1, Long.valueOf(12345));
+               row.setValue(2, "Fred");
+               row.setValue(3, "Flintstone");
+               tabular.mock_addRow(row);
+               
+               row = new Row(cols.length);
+               row.setValue(1, Long.valueOf(7654321));
+               row.setValue(2, "Barney");
+               row.setValue(3, "Rubble");
+               tabular.mock_addRow(row);
+               
+               return tabular;
        }
 }