1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.chukwa;
19
20
21 import java.io.IOException;
22 import junit.framework.TestCase;
23 import org.apache.hadoop.io.DataInputBuffer;
24 import org.apache.hadoop.io.DataOutputBuffer;
25
26 public class ChunkImplTest extends TestCase {
27 public void testVersion() {
28 ChunkBuilder cb = new ChunkBuilder();
29 cb.addRecord("foo".getBytes());
30 cb.addRecord("bar".getBytes());
31 cb.addRecord("baz".getBytes());
32 Chunk c = cb.getChunk();
33 DataOutputBuffer ob = new DataOutputBuffer(c.getSerializedSizeEstimate());
34 try {
35 c.write(ob);
36 DataInputBuffer ib = new DataInputBuffer();
37 ib.reset(ob.getData(), c.getSerializedSizeEstimate());
38 int version = ib.readInt();
39 ib.close();
40 assertEquals(version, ChunkImpl.PROTOCOL_VERSION);
41 } catch (IOException e) {
42 e.printStackTrace();
43 fail("Should nor raise any exception");
44 }
45 }
46
47 public void testTag() {
48 ChunkBuilder cb = new ChunkBuilder();
49 cb.addRecord("foo".getBytes());
50 cb.addRecord("bar".getBytes());
51 cb.addRecord("baz".getBytes());
52 Chunk c = cb.getChunk();
53 assertNull(c.getTag("foo"));
54 c.addTag("foo=\"bar\"");
55 assertEquals("bar", c.getTag("foo"));
56 }
57 }