1 21 package net.sf.hajdbc.sql; 22 23 import java.io.ByteArrayInputStream ; 24 import java.io.File ; 25 import java.io.FileOutputStream ; 26 import java.io.FileWriter ; 27 import java.io.IOException ; 28 import java.io.InputStream ; 29 import java.io.OutputStream ; 30 import java.io.Reader ; 31 import java.io.StringReader ; 32 import java.io.Writer ; 33 import java.sql.SQLException ; 34 import java.util.Arrays ; 35 36 import org.testng.annotations.AfterClass; 37 import org.testng.annotations.Test; 38 39 45 @Test 46 public class TestFileSupportImpl 47 { 48 private FileSupport fileSupport = new FileSupportImpl(); 49 50 @AfterClass 51 public void tearDown() throws Exception 52 { 53 this.fileSupport.close(); 54 } 55 56 59 public void testCreateFileInputStream() 60 { 61 InputStream inputStream = new ByteArrayInputStream (new byte[0]); 62 63 try 64 { 65 File file = this.fileSupport.createFile(inputStream); 66 67 assert file != null; 68 assert file.exists(); 69 assert file.getName().startsWith("ha-jdbc-"); 70 assert file.getName().endsWith(".lob"); 71 } 72 catch (SQLException e) 73 { 74 assert false : e; 75 } 76 } 77 78 81 public void testCreateFileReader() 82 { 83 Reader reader = new StringReader (""); 84 85 try 86 { 87 File file = this.fileSupport.createFile(reader); 88 89 assert file != null; 90 assert file.exists(); 91 assert file.getName().startsWith("ha-jdbc-"); 92 assert file.getName().endsWith(".lob"); 93 } 94 catch (SQLException e) 95 { 96 assert false : e; 97 } 98 } 99 100 103 public void testGetReader() 104 { 105 try 106 { 107 File file = File.createTempFile("test", ".test"); 108 Writer writer = new FileWriter (file); 109 writer.write("test"); 110 writer.flush(); 111 writer.close(); 112 113 Reader reader = this.fileSupport.getReader(file); 114 115 char[] buffer = new char[4]; 116 117 assert reader != null; 118 assert reader.read(buffer) == 4; 119 assert new String (buffer).equals("test"); 120 assert reader.read(buffer) < 0; 121 } 122 catch (IOException e) 123 { 124 assert false : e; 125 } 126 catch (SQLException e) 127 { 128 assert false : e; 129 } 130 } 131 132 135 public void testGetInputStream() 136 { 137 try 138 { 139 File file = File.createTempFile("test", ".test"); 140 OutputStream outputStream = new FileOutputStream (file); 141 outputStream.write(new byte[] { 1, 2, 3, 4 }); 142 outputStream.flush(); 143 outputStream.close(); 144 145 InputStream inputStream = this.fileSupport.getInputStream(file); 146 147 byte[] buffer = new byte[4]; 148 149 assert inputStream != null; 150 assert inputStream.read(buffer) == 4; 151 assert Arrays.equals(new byte[] { 1, 2, 3, 4 }, buffer); 152 assert inputStream.read(buffer) < 0; 153 } 154 catch (IOException e) 155 { 156 assert false : e; 157 } 158 catch (SQLException e) 159 { 160 assert false : e; 161 } 162 } 163 164 167 public void testClose() 168 { 169 Reader reader = new StringReader (""); 170 171 try 172 { 173 File file = this.fileSupport.createFile(reader); 174 175 assert file.exists(); 176 177 this.fileSupport.close(); 178 179 assert !file.exists(); 180 } 181 catch (SQLException e) 182 { 183 assert false : e; 184 } 185 } 186 } 187 | Popular Tags |