1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.directory.server.dns.io.encoder;
21
22
23 import java.io.IOException;
24 import java.net.UnknownHostException;
25 import java.util.Map;
26
27 import junit.framework.TestCase;
28
29 import org.apache.directory.server.dns.messages.RecordClass;
30 import org.apache.directory.server.dns.messages.RecordType;
31 import org.apache.directory.server.dns.messages.ResourceRecord;
32 import org.apache.directory.server.dns.messages.ResourceRecordImpl;
33 import org.apache.mina.common.ByteBuffer;
34
35
36
37
38
39
40
41
42
43
44 public abstract class AbstractResourceRecordEncoderTest extends TestCase
45 {
46 ByteBuffer expectedData;
47 String domainName = "herse.apache.org";
48 String[] domainNameParts = domainName.split( "\\." );
49 int timeToLive = 3400;
50 ResourceRecord record;
51
52
53 public void setUp() throws UnknownHostException
54 {
55 setUpResourceData();
56 record = new ResourceRecordImpl( domainName, RecordType.A, RecordClass.IN, timeToLive, getAttributes() );
57
58 expectedData = ByteBuffer.allocate( 128 );
59 expectedData.put( ( byte ) 18 );
60 expectedData.put( ( byte ) domainNameParts[0].length() );
61 expectedData.put( domainNameParts[0].getBytes() );
62 expectedData.put( ( byte ) domainNameParts[1].length() );
63 expectedData.put( domainNameParts[1].getBytes() );
64 expectedData.put( ( byte ) domainNameParts[2].length() );
65 expectedData.put( domainNameParts[2].getBytes() );
66 expectedData.put( ( byte ) 0x00 );
67 expectedData.putShort( RecordType.A.convert() );
68 expectedData.putShort( RecordClass.IN.convert() );
69 expectedData.putInt( timeToLive );
70 putExpectedResourceData( expectedData );
71 }
72
73
74 public void testEncode() throws IOException
75 {
76 ByteBuffer outBuffer = ByteBuffer.allocate( 128 );
77 getEncoder().put( outBuffer, record );
78 assertEquals( expectedData, outBuffer );
79 }
80
81
82
83
84
85
86
87 protected void setUpResourceData()
88 {
89 }
90
91
92
93
94
95 protected abstract ResourceRecordEncoder getEncoder();
96
97
98
99
100
101
102 protected abstract Map getAttributes();
103
104
105
106
107
108
109
110
111 protected abstract void putExpectedResourceData( ByteBuffer expectedData );
112 }