|
23 | 23 | * things around page boundaries |
24 | 24 | */ |
25 | 25 | public class TestWriteBoundaries extends TestCase { |
26 | | - public static byte[] getBytes(int len) { |
27 | | - byte[] b = new byte[len]; |
28 | | - for(int i=0; i<len; i++) { |
29 | | - b[i] = IOUtils.fromInt( i % 256 ); |
30 | | - } |
31 | | - return b; |
32 | | - } |
33 | | - |
34 | | - public void testMix() throws Exception { |
35 | | - ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
36 | | - OggFile ogg = new OggFile(baos); |
37 | | - OggPacketWriter w = ogg.getPacketWriter(0x123456); |
38 | | - |
39 | | - OggPacket p; |
40 | | - |
41 | | - p = new OggPacket(new byte[] { |
42 | | - 0, 1, 2, 3, 4, 5 |
43 | | - }); |
44 | | - w.bufferPacket(p, false); |
45 | | - p = new OggPacket(new byte[] { |
46 | | - 10, 11, 12, 13, 14, 15 |
47 | | - }); |
48 | | - w.bufferPacket(p, true); |
49 | | - |
50 | | - p = new OggPacket(new byte[] { |
51 | | - 0, 1, 2, 3, 4, 5 |
52 | | - }); |
53 | | - w.bufferPacket(p, false); |
54 | | - |
55 | | - p = new OggPacket(getBytes(131072)); |
56 | | - w.bufferPacket(p, false); |
57 | | - |
58 | | - p = new OggPacket(new byte[] { |
59 | | - 0, 1, 2, 3, 4, 5 |
60 | | - }); |
61 | | - w.bufferPacket(p, false); |
62 | | - w.close(); |
63 | | - |
64 | | - |
65 | | - ogg = new OggFile(new ByteArrayInputStream(baos.toByteArray())); |
66 | | - OggPacketReader r = ogg.getPacketReader(); |
67 | | - |
68 | | - p = r.getNextPacket(); |
69 | | - assertEquals(0, p.getSequenceNumber()); |
70 | | - assertEquals(6, p.getData().length); |
71 | | - |
72 | | - p = r.getNextPacket(); |
73 | | - assertEquals(0, p.getSequenceNumber()); |
74 | | - assertEquals(6, p.getData().length); |
75 | | - |
76 | | - p = r.getNextPacket(); |
77 | | - assertEquals(1, p.getSequenceNumber()); |
78 | | - assertEquals(6, p.getData().length); |
79 | | - |
80 | | - p = r.getNextPacket(); |
81 | | - assertEquals(3, p.getSequenceNumber()); // over 3 pages |
82 | | - assertEquals(131072, p.getData().length); |
83 | | - |
84 | | - p = r.getNextPacket(); |
85 | | - assertEquals(3, p.getSequenceNumber()); |
86 | | - assertEquals(6, p.getData().length); |
87 | | - |
88 | | - p = r.getNextPacket(); |
89 | | - assertEquals(null, p); |
90 | | - } |
91 | | - |
92 | | - public void testWriteBigPackets() throws Exception { |
93 | | - ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
94 | | - OggFile ogg = new OggFile(baos); |
95 | | - OggPacketWriter w = ogg.getPacketWriter(0x123456); |
96 | | - |
97 | | - OggPacket p; |
98 | | - |
99 | | - // Small |
100 | | - p = new OggPacket(new byte[] { |
101 | | - 0, 1, 2, 3, 4, 5 |
102 | | - }); |
103 | | - w.bufferPacket(p, true); |
104 | | - |
105 | | - // Over 2-and-a-bit packets |
106 | | - p = new OggPacket(getBytes(131072)); |
107 | | - w.bufferPacket(p, true); |
108 | | - |
109 | | - // Over 2-and-a-bit packets, no flush |
110 | | - p = new OggPacket(getBytes(144077)); |
111 | | - w.bufferPacket(p, false); |
112 | | - // Ditto |
113 | | - p = new OggPacket(getBytes(161077)); |
114 | | - w.bufferPacket(p, false); |
115 | | - // Over 3-and-a-bit packets |
116 | | - p = new OggPacket(getBytes(231079)); |
117 | | - w.bufferPacket(p, false); |
118 | | - |
119 | | - // Small, flush |
120 | | - p = new OggPacket(new byte[] { |
121 | | - 0, 1, 2, 3, 4, 5 |
122 | | - }); |
123 | | - w.bufferPacket(p, true); |
124 | | - // Small, close |
125 | | - p = new OggPacket(new byte[] { |
126 | | - 0, 1, 2, 3, 4, 5 |
127 | | - }); |
128 | | - w.bufferPacket(p, false); |
129 | | - w.close(); |
130 | | - |
131 | | - |
132 | | - // Check |
133 | | - ogg = new OggFile(new ByteArrayInputStream(baos.toByteArray())); |
134 | | - OggPacketReader r = ogg.getPacketReader(); |
135 | | - |
136 | | - p = r.getNextPacket(); |
137 | | - assertEquals(0, p.getSequenceNumber()); |
138 | | - assertEquals(6, p.getData().length); |
139 | | - |
140 | | - p = r.getNextPacket(); |
141 | | - assertEquals(3, p.getSequenceNumber()); // 1,2,3 |
142 | | - assertEquals(131072, p.getData().length); |
143 | | - |
144 | | - p = r.getNextPacket(); |
145 | | - assertEquals(6, p.getSequenceNumber()); // 4,5,6 |
146 | | - assertEquals(144077, p.getData().length); |
147 | | - |
148 | | - p = r.getNextPacket(); |
149 | | - assertEquals(8, p.getSequenceNumber()); // 6,7,8 |
150 | | - assertEquals(161077, p.getData().length); |
151 | | - |
152 | | - p = r.getNextPacket(); |
153 | | - assertEquals(12, p.getSequenceNumber()); // 9,10,11,12 |
154 | | - assertEquals(231079, p.getData().length); |
155 | | - |
156 | | - p = r.getNextPacket(); |
157 | | - assertEquals(12, p.getSequenceNumber()); |
158 | | - assertEquals(6, p.getData().length); |
159 | | - |
160 | | - p = r.getNextPacket(); |
161 | | - assertEquals(13, p.getSequenceNumber()); |
162 | | - assertEquals(6, p.getData().length); |
163 | | - |
164 | | - p = r.getNextPacket(); |
165 | | - assertEquals(null, p); |
166 | | - } |
| 26 | + public static byte[] getBytes(int len) { |
| 27 | + byte[] b = new byte[len]; |
| 28 | + for(int i=0; i<len; i++) { |
| 29 | + b[i] = IOUtils.fromInt( i % 256 ); |
| 30 | + } |
| 31 | + return b; |
| 32 | + } |
| 33 | + |
| 34 | + public void testMix() throws Exception { |
| 35 | + ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
| 36 | + OggFile ogg = new OggFile(baos); |
| 37 | + OggPacketWriter w = ogg.getPacketWriter(0x123456); |
| 38 | + |
| 39 | + OggPacket p; |
| 40 | + |
| 41 | + p = new OggPacket(new byte[] { |
| 42 | + 0, 1, 2, 3, 4, 5 |
| 43 | + }); |
| 44 | + w.bufferPacket(p, false); |
| 45 | + p = new OggPacket(new byte[] { |
| 46 | + 10, 11, 12, 13, 14, 15 |
| 47 | + }); |
| 48 | + w.bufferPacket(p, true); |
| 49 | + |
| 50 | + p = new OggPacket(new byte[] { |
| 51 | + 0, 1, 2, 3, 4, 5 |
| 52 | + }); |
| 53 | + w.bufferPacket(p, false); |
| 54 | + |
| 55 | + p = new OggPacket(getBytes(131072)); |
| 56 | + w.bufferPacket(p, false); |
| 57 | + |
| 58 | + p = new OggPacket(new byte[] { |
| 59 | + 0, 1, 2, 3, 4, 5 |
| 60 | + }); |
| 61 | + w.bufferPacket(p, false); |
| 62 | + w.close(); |
| 63 | + |
| 64 | + |
| 65 | + ogg = new OggFile(new ByteArrayInputStream(baos.toByteArray())); |
| 66 | + OggPacketReader r = ogg.getPacketReader(); |
| 67 | + |
| 68 | + p = r.getNextPacket(); |
| 69 | + assertEquals(0, p.getSequenceNumber()); |
| 70 | + assertEquals(6, p.getData().length); |
| 71 | + |
| 72 | + p = r.getNextPacket(); |
| 73 | + assertEquals(0, p.getSequenceNumber()); |
| 74 | + assertEquals(6, p.getData().length); |
| 75 | + |
| 76 | + p = r.getNextPacket(); |
| 77 | + assertEquals(1, p.getSequenceNumber()); |
| 78 | + assertEquals(6, p.getData().length); |
| 79 | + |
| 80 | + p = r.getNextPacket(); |
| 81 | + assertEquals(3, p.getSequenceNumber()); // over 3 pages |
| 82 | + assertEquals(131072, p.getData().length); |
| 83 | + |
| 84 | + p = r.getNextPacket(); |
| 85 | + assertEquals(3, p.getSequenceNumber()); |
| 86 | + assertEquals(6, p.getData().length); |
| 87 | + |
| 88 | + p = r.getNextPacket(); |
| 89 | + assertEquals(null, p); |
| 90 | + } |
| 91 | + |
| 92 | + public void testWriteBigPackets() throws Exception { |
| 93 | + ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
| 94 | + OggFile ogg = new OggFile(baos); |
| 95 | + OggPacketWriter w = ogg.getPacketWriter(0x123456); |
| 96 | + |
| 97 | + OggPacket p; |
| 98 | + |
| 99 | + // Small |
| 100 | + p = new OggPacket(new byte[] { |
| 101 | + 0, 1, 2, 3, 4, 5 |
| 102 | + }); |
| 103 | + w.bufferPacket(p, true); |
| 104 | + |
| 105 | + // Over 2-and-a-bit packets |
| 106 | + p = new OggPacket(getBytes(131072)); |
| 107 | + w.bufferPacket(p, true); |
| 108 | + |
| 109 | + // Over 2-and-a-bit packets, no flush |
| 110 | + p = new OggPacket(getBytes(144077)); |
| 111 | + w.bufferPacket(p, false); |
| 112 | + // Ditto |
| 113 | + p = new OggPacket(getBytes(161077)); |
| 114 | + w.bufferPacket(p, false); |
| 115 | + // Over 3-and-a-bit packets |
| 116 | + p = new OggPacket(getBytes(231079)); |
| 117 | + w.bufferPacket(p, false); |
| 118 | + |
| 119 | + // Small, flush |
| 120 | + p = new OggPacket(new byte[] { |
| 121 | + 0, 1, 2, 3, 4, 5 |
| 122 | + }); |
| 123 | + w.bufferPacket(p, true); |
| 124 | + // Small, close |
| 125 | + p = new OggPacket(new byte[] { |
| 126 | + 0, 1, 2, 3, 4, 5 |
| 127 | + }); |
| 128 | + w.bufferPacket(p, false); |
| 129 | + w.close(); |
| 130 | + |
| 131 | + |
| 132 | + // Check |
| 133 | + ogg = new OggFile(new ByteArrayInputStream(baos.toByteArray())); |
| 134 | + OggPacketReader r = ogg.getPacketReader(); |
| 135 | + |
| 136 | + p = r.getNextPacket(); |
| 137 | + assertEquals(0, p.getSequenceNumber()); |
| 138 | + assertEquals(6, p.getData().length); |
| 139 | + |
| 140 | + p = r.getNextPacket(); |
| 141 | + assertEquals(3, p.getSequenceNumber()); // 1,2,3 |
| 142 | + assertEquals(131072, p.getData().length); |
| 143 | + |
| 144 | + p = r.getNextPacket(); |
| 145 | + assertEquals(6, p.getSequenceNumber()); // 4,5,6 |
| 146 | + assertEquals(144077, p.getData().length); |
| 147 | + |
| 148 | + p = r.getNextPacket(); |
| 149 | + assertEquals(8, p.getSequenceNumber()); // 6,7,8 |
| 150 | + assertEquals(161077, p.getData().length); |
| 151 | + |
| 152 | + p = r.getNextPacket(); |
| 153 | + assertEquals(12, p.getSequenceNumber()); // 9,10,11,12 |
| 154 | + assertEquals(231079, p.getData().length); |
| 155 | + |
| 156 | + p = r.getNextPacket(); |
| 157 | + assertEquals(12, p.getSequenceNumber()); |
| 158 | + assertEquals(6, p.getData().length); |
| 159 | + |
| 160 | + p = r.getNextPacket(); |
| 161 | + assertEquals(13, p.getSequenceNumber()); |
| 162 | + assertEquals(6, p.getData().length); |
| 163 | + |
| 164 | + p = r.getNextPacket(); |
| 165 | + assertEquals(null, p); |
| 166 | + } |
167 | 167 | } |
0 commit comments