arcwritertest.html

用JAVA编写的,在做实验的时候留下来的,本来想删的,但是传上来,大家分享吧

<a name="126" href="#126">126</a>         assertTrue(<span class="string">"Doesn't exist: "</span> +
<a name="127" href="#127">127</a>                 arcWriter.getFile().getAbsolutePath(), 
<a name="128" href="#128">128</a>             arcWriter.getFile().exists());
<a name="129" href="#129">129</a>         <strong>return</strong> arcWriter.getFile();
<a name="130" href="#130">130</a>     }
<a name="131" href="#131">131</a> 
<a name="132" href="#132">132</a>     <strong>private</strong> <strong>void</strong> validate(File arcFile, <strong>int</strong> recordCount)
<a name="133" href="#133">133</a>     throws FileNotFoundException, IOException {
<a name="134" href="#134">134</a>         <a href="../../../../org/archive/io/arc/ARCReader.html">ARCReader</a> reader = ARCReaderFactory.get(arcFile);
<a name="135" href="#135">135</a>         assertNotNull(reader);
<a name="136" href="#136">136</a>         List metaDatas = <strong>null</strong>;
<a name="137" href="#137">137</a>         <strong>if</strong> (recordCount == -1) {
<a name="138" href="#138">138</a>             metaDatas = reader.validate();
<a name="139" href="#139">139</a>         } <strong>else</strong> {
<a name="140" href="#140">140</a>             metaDatas = reader.validate(recordCount);
<a name="141" href="#141">141</a>         }
<a name="142" href="#142">142</a>         reader.close();
<a name="143" href="#143">143</a>         <em class="comment">// Now, run through each of the records doing absolute get going from</em>
<a name="144" href="#144">144</a>         <em class="comment">// the end to start.  Reopen the arc so no context between this test</em>
<a name="145" href="#145">145</a>         <em class="comment">// and the previous.</em>
<a name="146" href="#146">146</a>         reader = ARCReaderFactory.get(arcFile);
<a name="147" href="#147">147</a>         <strong>for</strong> (<strong>int</strong> i = metaDatas.size() - 1; i >= 0; i--) {
<a name="148" href="#148">148</a>             <a href="../../../../org/archive/io/arc/ARCRecordMetaData.html">ARCRecordMetaData</a> meta = (ARCRecordMetaData)metaDatas.get(i);
<a name="149" href="#149">149</a>             <a href="../../../../org/archive/io/ArchiveRecord.html">ArchiveRecord</a> r = reader.get(meta.getOffset());
<a name="150" href="#150">150</a>             String mimeType = r.getHeader().getMimetype();
<a name="151" href="#151">151</a>             assertTrue(<span class="string">"Record is bogus"</span>,
<a name="152" href="#152">152</a>                 mimeType != <strong>null</strong> &amp;&amp; mimeType.length() > 0);
<a name="153" href="#153">153</a>         }
<a name="154" href="#154">154</a>         reader.close();
<a name="155" href="#155">155</a>         assertTrue(<span class="string">"Metadatas not equal"</span>, metaDatas.size() == recordCount);
<a name="156" href="#156">156</a>         <strong>for</strong> (Iterator i = metaDatas.iterator(); i.hasNext();) {
<a name="157" href="#157">157</a>                 <a href="../../../../org/archive/io/arc/ARCRecordMetaData.html">ARCRecordMetaData</a> r = (ARCRecordMetaData)i.next();
<a name="158" href="#158">158</a>                 assertTrue(<span class="string">"Record is empty"</span>, r.getLength() > 0);
<a name="159" href="#159">159</a>         }
<a name="160" href="#160">160</a>     }
<a name="161" href="#161">161</a> 
<a name="162" href="#162">162</a>     <strong>public</strong> <strong>void</strong> testCheckARCFileSize()
<a name="163" href="#163">163</a>     throws IOException {
<a name="164" href="#164">164</a>         runCheckARCFileSizeTest(<span class="string">"checkARCFileSize"</span>, false);
<a name="165" href="#165">165</a>     }
<a name="166" href="#166">166</a> 
<a name="167" href="#167">167</a>     <strong>public</strong> <strong>void</strong> testCheckARCFileSizeCompressed()
<a name="168" href="#168">168</a>     throws IOException {
<a name="169" href="#169">169</a>         runCheckARCFileSizeTest(<span class="string">"checkARCFileSize"</span>, <strong>true</strong>);
<a name="170" href="#170">170</a>     }
<a name="171" href="#171">171</a> 
<a name="172" href="#172">172</a>     <strong>public</strong> <strong>void</strong> testWriteRecord() throws IOException {
<a name="173" href="#173">173</a>         <strong>final</strong> <strong>int</strong> recordCount = 2;
<a name="174" href="#174">174</a>         File arcFile = writeRecords(<span class="string">"writeRecord"</span>, false,
<a name="175" href="#175">175</a>                 DEFAULT_MAX_ARC_FILE_SIZE, recordCount);
<a name="176" href="#176">176</a>         validate(arcFile, recordCount  + 1); <em class="comment">// Header record.</em>
<a name="177" href="#177">177</a>     }
<a name="178" href="#178">178</a>     
<a name="179" href="#179">179</a>     <strong>public</strong> <strong>void</strong> testRandomAccess() throws IOException {
<a name="180" href="#180">180</a>         <strong>final</strong> <strong>int</strong> recordCount = 3;
<a name="181" href="#181">181</a>         File arcFile = writeRecords(<span class="string">"writeRecord"</span>, <strong>true</strong>,
<a name="182" href="#182">182</a>             DEFAULT_MAX_ARC_FILE_SIZE, recordCount);
<a name="183" href="#183">183</a>         <a href="../../../../org/archive/io/arc/ARCReader.html">ARCReader</a> reader = ARCReaderFactory.get(arcFile);
<a name="184" href="#184">184</a>         <em class="comment">// Get to second record.  Get its offset for later use.</em>
<a name="185" href="#185">185</a>         <strong>boolean</strong> readFirst = false;
<a name="186" href="#186">186</a>         String url = <strong>null</strong>;
<a name="187" href="#187">187</a>         <strong>long</strong> offset = -1;
<a name="188" href="#188">188</a>         <strong>long</strong> totalRecords = 0;
<a name="189" href="#189">189</a>         <strong>boolean</strong> readSecond = false;
<a name="190" href="#190">190</a>         <strong>for</strong> (<strong>final</strong> Iterator i = reader.iterator(); i.hasNext(); totalRecords++) {
<a name="191" href="#191">191</a>             <a href="../../../../org/archive/io/arc/ARCRecord.html">ARCRecord</a> ar = (ARCRecord)i.next();
<a name="192" href="#192">192</a>             <strong>if</strong> (!readFirst) {
<a name="193" href="#193">193</a>                 readFirst = <strong>true</strong>;
<a name="194" href="#194">194</a>                 <strong>continue</strong>;
<a name="195" href="#195">195</a>             }
<a name="196" href="#196">196</a>             <strong>if</strong> (!readSecond) {
<a name="197" href="#197">197</a>                 url = ar.getMetaData().getUrl();
<a name="198" href="#198">198</a>                 offset = ar.getMetaData().getOffset();
<a name="199" href="#199">199</a>                 readSecond = <strong>true</strong>;
<a name="200" href="#200">200</a>             }
<a name="201" href="#201">201</a>         }
<a name="202" href="#202">202</a>         
<a name="203" href="#203">203</a>         reader = ARCReaderFactory.get(arcFile, offset);
<a name="204" href="#204">204</a>         <a href="../../../../org/archive/io/ArchiveRecord.html">ArchiveRecord</a> ar = reader.get();
<a name="205" href="#205">205</a>         assertEquals(ar.getHeader().getUrl(), url);
<a name="206" href="#206">206</a>         ar.close();
<a name="207" href="#207">207</a>         
<a name="208" href="#208">208</a>         <em class="comment">// Get reader again.  See how iterator works with offset</em>
<a name="209" href="#209">209</a>         reader = ARCReaderFactory.get(arcFile, offset);
<a name="210" href="#210">210</a>         <strong>int</strong> count = 0;
<a name="211" href="#211">211</a>         <strong>for</strong> (<strong>final</strong> Iterator i = reader.iterator(); i.hasNext(); i.next()) {
<a name="212" href="#212">212</a>             count++;
<a name="213" href="#213">213</a>         }
<a name="214" href="#214">214</a>         reader.close();
<a name="215" href="#215">215</a>         assertEquals(totalRecords - 1, count);
<a name="216" href="#216">216</a>     }
<a name="217" href="#217">217</a> 
<a name="218" href="#218">218</a>     <strong>public</strong> <strong>void</strong> testWriteRecordCompressed() throws IOException {
<a name="219" href="#219">219</a>         <strong>final</strong> <strong>int</strong> recordCount = 2;
<a name="220" href="#220">220</a>         File arcFile = writeRecords(<span class="string">"writeRecordCompressed"</span>, <strong>true</strong>,
<a name="221" href="#221">221</a>                 DEFAULT_MAX_ARC_FILE_SIZE, recordCount);
<a name="222" href="#222">222</a>         validate(arcFile, recordCount + 1 <em class="comment">/*<em class="comment">Header record*/</em>);</em>
<a name="223" href="#223">223</a>     }
<a name="224" href="#224">224</a>     
<a name="225" href="#225">225</a>     <strong>private</strong> <strong>void</strong> runCheckARCFileSizeTest(String baseName, <strong>boolean</strong> compress)
<a name="226" href="#226">226</a>     throws FileNotFoundException, IOException  {
<a name="227" href="#227">227</a>         writeRecords(baseName, compress, 1024, 15);
<a name="228" href="#228">228</a>         <em class="comment">// Now validate all files just created.</em>
<a name="229" href="#229">229</a>         File [] files = FileUtils.getFilesWithPrefix(getTmpDir(), PREFIX);
<a name="230" href="#230">230</a>         <strong>for</strong> (<strong>int</strong> i = 0; i &lt; files.length; i++) {
<a name="231" href="#231">231</a>             validate(files[i], -1);
<a name="232" href="#232">232</a>         }
<a name="233" href="#233">233</a>     }
<a name="234" href="#234">234</a>     
<a name="235" href="#235">235</a>     <strong>protected</strong> <a href="../../../../org/archive/io/arc/ARCWriter.html">ARCWriter</a> createARCWriter(String NAME, <strong>boolean</strong> compress) {
<a name="236" href="#236">236</a>         File [] files = {getTmpDir()};
<a name="237" href="#237">237</a>         <strong>return</strong> <strong>new</strong> <a href="../../../../org/archive/io/arc/ARCWriter.html">ARCWriter</a>(SERIAL_NO, Arrays.asList(files), NAME,
<a name="238" href="#238">238</a>             compress, DEFAULT_MAX_ARC_FILE_SIZE);
<a name="239" href="#239">239</a>     }
<a name="240" href="#240">240</a>     
<a name="241" href="#241">241</a>     <strong>protected</strong> <strong>static</strong> ByteArrayOutputStream getBaos(String str)
<a name="242" href="#242">242</a>     throws IOException {
<a name="243" href="#243">243</a>         ByteArrayOutputStream baos = <strong>new</strong> ByteArrayOutputStream();
<a name="244" href="#244">244</a>         baos.write(str.getBytes());
<a name="245" href="#245">245</a>         <strong>return</strong> baos;
<a name="246" href="#246">246</a>     }
<a name="247" href="#247">247</a>     
<a name="248" href="#248">248</a>     <strong>protected</strong> <strong>static</strong> <strong>void</strong> writeRecord(<a href="../../../../org/archive/io/arc/ARCWriter.html">ARCWriter</a> writer, String url,
<a name="249" href="#249">249</a>         String type, <strong>int</strong> len, ByteArrayOutputStream baos)
<a name="250" href="#250">250</a>     throws IOException {
<a name="251" href="#251">251</a>         writer.write(url, type, <span class="string">"192.168.1.1"</span>, (<strong>new</strong> Date()).getTime(), len,
<a name="252" href="#252">252</a>             baos);
<a name="253" href="#253">253</a>     }
<a name="254" href="#254">254</a>     
<a name="255" href="#255">255</a>     <strong>protected</strong> <strong>int</strong> iterateRecords(<a href="../../../../org/archive/io/arc/ARCReader.html">ARCReader</a> r)
<a name="256" href="#256">256</a>     throws IOException {
<a name="257" href="#257">257</a>         <strong>int</strong> count = 0;
<a name="258" href="#258">258</a>         <strong>for</strong> (Iterator i = r.iterator(); i.hasNext();) {
<a name="259" href="#259">259</a>             <a href="../../../../org/archive/io/arc/ARCRecord.html">ARCRecord</a> rec = (ARCRecord)i.next();