KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > scriptella > jdbc > LobsTest


1 /*
2  * Copyright 2006-2007 The Scriptella Project Team.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package scriptella.jdbc;
17
18 import scriptella.AbstractTestCase;
19 import scriptella.util.IOUtils;
20
21 import java.io.ByteArrayInputStream JavaDoc;
22 import java.io.File JavaDoc;
23 import java.io.IOException JavaDoc;
24 import java.io.InputStream JavaDoc;
25 import java.io.OutputStream JavaDoc;
26 import java.io.StringReader JavaDoc;
27 import java.net.URL JavaDoc;
28 import java.sql.SQLException JavaDoc;
29 import java.util.Arrays JavaDoc;
30
31 /**
32  * Tests for {@link Lobs}.
33  *
34  * @author Fyodor Kupolov
35  * @version 1.0
36  */
37 public class LobsTest extends AbstractTestCase {
38     private static final byte[] bsob=new byte[1024*10]; //small
39 private static final byte[] blob=new byte[1024*110]; //large
40 private static final String JavaDoc clob;
41     private static final String JavaDoc csob;
42     static {
43         //Initializing blob/clob and small objects
44
45         for (int i=0;i<blob.length;i++) {
46             blob[i]= (byte) (0x30+i%10);
47         }
48         clob=new String JavaDoc(blob);
49         for (int i=0;i<bsob.length;i++) {
50             bsob[i]= (byte) (0x30+i%10);
51         }
52         csob=new String JavaDoc(bsob);
53
54         //reading test properties containing a set of drivers
55
56     }
57
58     public void testBlob() throws IOException JavaDoc {
59         //Test the BLOB 110K
60 Lobs.ReadonlyBlob b = (Lobs.ReadonlyBlob) Lobs.newBlob(new ByteArrayInputStream JavaDoc(blob));
61         assertNull(b.tmpFile);
62         assertEquals(blob.length, b.length());
63         File JavaDoc tmpFile = b.tmpFile;
64         assertNotNull(tmpFile);
65         assertTrue(tmpFile.exists()); //temp file should exist
66 assertEquals(blob.length, tmpFile.length()); //and contain the same number of bytes as blob
67 assertTrue(Arrays.equals(blob, IOUtils.toByteArray(b.getBinaryStream())));
68         //obtain a new stream second time to check if fresh stream is created for each call
69 assertTrue(Arrays.equals(blob, IOUtils.toByteArray(b.getBinaryStream())));
70
71         b.close();
72         assertFalse(tmpFile.exists()); //temp file should be deleted on close
73 //Test the small object 10K
74 b = (Lobs.ReadonlyBlob) Lobs.newBlob(new ByteArrayInputStream JavaDoc(bsob));
75         assertNull(b.tmpFile);
76         assertEquals(bsob.length, b.length());
77         assertNull(b.tmpFile);
78         assertTrue(Arrays.equals(bsob, IOUtils.toByteArray(b.getBinaryStream())));
79         b.close(); //Just to make sure closing works (smoke test)
80 //Now test the constructor with length
81 b = (Lobs.ReadonlyBlob) Lobs.newBlob(new ByteArrayInputStream JavaDoc(blob), blob.length);
82         assertNull(b.tmpFile);
83         assertEquals(blob.length, b.length());
84         assertNull(b.tmpFile); //length is known no need to read file
85 assertTrue(Arrays.equals(blob, IOUtils.toByteArray(b.getBinaryStream())));
86         tmpFile = b.tmpFile;
87         assertNotNull(tmpFile); //temp file should be created
88 b.close();
89
90     }
91
92     public void testUrlBlob() throws IOException JavaDoc, SQLException JavaDoc {
93         testURLHandler = new TestURLHandler() {
94             public InputStream JavaDoc getInputStream(final URL JavaDoc u) {
95                 return new ByteArrayInputStream JavaDoc(bsob);
96             }
97
98             public OutputStream JavaDoc getOutputStream(final URL JavaDoc u) {
99                 throw new UnsupportedOperationException JavaDoc();
100             }
101
102             public int getContentLength(final URL JavaDoc u) {
103                 return bsob.length;
104             }
105         };
106         Lobs.UrlBlob b= (Lobs.UrlBlob) Lobs.newBlob(new URL JavaDoc("tst://file"));
107         assertTrue(Arrays.equals(bsob, IOUtils.toByteArray(b.getBinaryStream())));
108         //Check again to receive a new stream
109 assertTrue(Arrays.equals(bsob, IOUtils.toByteArray(b.getBinaryStream())));
110
111         assertEquals(bsob.length, b.length());
112         b.close();
113         testURLHandler = new TestURLHandler() {
114             public InputStream JavaDoc getInputStream(final URL JavaDoc u) {
115                 return new ByteArrayInputStream JavaDoc(bsob);
116             }
117
118             public OutputStream JavaDoc getOutputStream(final URL JavaDoc u) {
119                 throw new UnsupportedOperationException JavaDoc();
120             }
121
122             public int getContentLength(final URL JavaDoc u) {
123                 return -1;
124             }
125         };
126         b = (Lobs.UrlBlob) Lobs.newBlob(new URL JavaDoc("tst://file"));
127         assertTrue(Arrays.equals(bsob, IOUtils.toByteArray(b.getBinaryStream())));
128         //Check again to receive a new stream
129 assertTrue(Arrays.equals(bsob, IOUtils.toByteArray(b.getBinaryStream())));
130         assertEquals(bsob.length, b.length());
131         b.close();
132         assertNull(b.tmpFile);
133     }
134
135     public void testClob() throws IOException JavaDoc {
136         //Test the CLOB 110K
137 Lobs.ReadonlyClob c = (Lobs.ReadonlyClob) Lobs.newClob(new StringReader JavaDoc(clob));
138         assertNull(c.tmpFile);
139         assertEquals(clob.length(), c.length());
140         assertNotNull(c.tmpFile);
141         File JavaDoc tmp = c.tmpFile;
142         assertEquals(clob, IOUtils.toString(c.getCharacterStream()));
143         //obtain a new stream second time to check if fresh stream is created for each call
144 assertEquals(clob, IOUtils.toString(c.getCharacterStream()));
145         c.close();
146         assertFalse(tmp.exists()); //temp file should be deleted
147 //Test the CSOB (small object) 10K
148 c = (Lobs.ReadonlyClob) Lobs.newClob(new StringReader JavaDoc(csob));
149         assertNull(c.tmpFile);
150         assertEquals(csob.length(), c.length());
151         assertNull(c.tmpFile);
152         assertEquals(csob, IOUtils.toString(c.getCharacterStream()));
153         c.close();
154         //Test the CLOB with lengtj
155 c = (Lobs.ReadonlyClob) Lobs.newClob(new StringReader JavaDoc(clob), clob.length());
156         assertNull(c.tmpFile);
157         assertEquals(clob.length(), c.length());
158         assertNull(c.tmpFile); //tmp file should not created
159 c.close();
160     }
161 }
Popular Tags