ArpTest.java
3.84 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
* Copyright 2015-present Open Networking Laboratory
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.onlab.packet;
import org.apache.commons.lang3.StringUtils;
import org.junit.Before;
import org.junit.Test;
import java.nio.ByteBuffer;
import java.util.Arrays;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
/**
* Unit tests for the ARP class.
*/
public class ArpTest {
private Deserializer<ARP> deserializer = ARP.deserializer();
private final byte hwAddressLength = 6;
private final byte protoAddressLength = 4;
private MacAddress srcMac = MacAddress.valueOf(1);
private MacAddress targetMac = MacAddress.valueOf(2);
private Ip4Address srcIp = Ip4Address.valueOf(1);
private Ip4Address targetIp = Ip4Address.valueOf(2);
private byte[] byteHeader;
@Before
public void setUp() {
ByteBuffer bb = ByteBuffer.allocate(ARP.INITIAL_HEADER_LENGTH +
2 * hwAddressLength + 2 * protoAddressLength);
bb.putShort(ARP.HW_TYPE_ETHERNET);
bb.putShort(ARP.PROTO_TYPE_IP);
bb.put(hwAddressLength);
bb.put(protoAddressLength);
bb.putShort(ARP.OP_REPLY);
bb.put(srcMac.toBytes());
bb.put(srcIp.toOctets());
bb.put(targetMac.toBytes());
bb.put(targetIp.toOctets());
byteHeader = bb.array();
}
@Test
public void testDeserializeBadInput() throws Exception {
PacketTestUtils.testDeserializeBadInput(deserializer);
}
@Test
public void testDeserializeTruncated() throws Exception {
PacketTestUtils.testDeserializeTruncated(deserializer, byteHeader);
}
/**
* Tests deserialize and getters.
*/
@Test
public void testDeserialize() throws Exception {
ARP arp = deserializer.deserialize(byteHeader, 0, byteHeader.length);
assertEquals(ARP.HW_TYPE_ETHERNET, arp.getHardwareType());
assertEquals(ARP.PROTO_TYPE_IP, arp.getProtocolType());
assertEquals(hwAddressLength, arp.getHardwareAddressLength());
assertEquals(protoAddressLength, arp.getProtocolAddressLength());
assertEquals(ARP.OP_REPLY, arp.getOpCode());
assertTrue(Arrays.equals(srcMac.toBytes(), arp.getSenderHardwareAddress()));
assertTrue(Arrays.equals(srcIp.toOctets(), arp.getSenderProtocolAddress()));
assertTrue(Arrays.equals(targetMac.toBytes(), arp.getTargetHardwareAddress()));
assertTrue(Arrays.equals(targetIp.toOctets(), arp.getTargetProtocolAddress()));
}
/**
* Tests toString.
*/
@Test
public void testToStringArp() throws Exception {
ARP arp = deserializer.deserialize(byteHeader, 0, byteHeader.length);
String str = arp.toString();
assertTrue(StringUtils.contains(str, "hardwareAddressLength=" + hwAddressLength));
assertTrue(StringUtils.contains(str, "protocolAddressLength=" + protoAddressLength));
assertTrue(StringUtils.contains(str, "senderHardwareAddress=" + Arrays.toString(srcMac.toBytes())));
assertTrue(StringUtils.contains(str, "senderProtocolAddress=" + Arrays.toString(srcIp.toOctets())));
assertTrue(StringUtils.contains(str, "targetHardwareAddress=" + Arrays.toString(targetMac.toBytes())));
assertTrue(StringUtils.contains(str, "targetProtocolAddress=" + Arrays.toString(targetIp.toOctets())));
}
}