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
|
package org.onosproject.provider.lldp.impl;
import static org.junit.Assert.*;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import org.junit.Before;
import org.junit.Test;
import org.onosproject.TestApplicationId;
import org.onosproject.net.Device;
import org.onosproject.net.DeviceId;
import org.onosproject.net.config.ConfigApplyDelegate;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
public class SuppressionConfigTest {
private static final String APP_NAME = "SuppressionConfigTest";
private static final TestApplicationId APP_ID = new TestApplicationId(APP_NAME);
private static final DeviceId DEVICE_ID_1 = DeviceId.deviceId("of:1111000000000000");
private static final DeviceId DEVICE_ID_2 = DeviceId.deviceId("of:2222000000000000");
private static final Device.Type DEVICE_TYPE_1 = Device.Type.ROADM;
private static final Device.Type DEVICE_TYPE_2 = Device.Type.FIBER_SWITCH;
private static final String ANNOTATION_KEY_1 = "no_lldp";
private static final String ANNOTATION_VALUE_1 = "true";
private static final String ANNOTATION_KEY_2 = "sendLLDP";
private static final String ANNOTATION_VALUE_2 = "false";
private SuppressionConfig cfg;
@Before
public void setUp() throws Exception {
ConfigApplyDelegate delegate = config -> { };
ObjectMapper mapper = new ObjectMapper();
cfg = new SuppressionConfig();
cfg.init(APP_ID, LldpLinkProvider.CONFIG_KEY, JsonNodeFactory.instance.objectNode(), mapper, delegate);
}
@Test
public void testDeviceTypes() {
Set<Device.Type> inputTypes = new HashSet<Device.Type>() { {
add(DEVICE_TYPE_1);
add(DEVICE_TYPE_2);
} };
assertNotNull(cfg.deviceTypes(inputTypes));
Set<Device.Type> outputTypes = cfg.deviceTypes();
assertTrue(outputTypes.contains(DEVICE_TYPE_1));
assertTrue(outputTypes.contains(DEVICE_TYPE_2));
assertEquals(outputTypes.size(), 2);
}
@Test
public void testDeviceAnnotation() {
Map<String, String> inputMap = new HashMap<String, String>() { {
put(ANNOTATION_KEY_1, ANNOTATION_VALUE_1);
put(ANNOTATION_KEY_2, ANNOTATION_VALUE_2);
} };
assertNotNull(cfg.annotation(inputMap));
Map<String, String> outputMap = cfg.annotation();
assertEquals(outputMap.get(ANNOTATION_KEY_1), ANNOTATION_VALUE_1);
assertEquals(outputMap.get(ANNOTATION_KEY_2), ANNOTATION_VALUE_2);
assertEquals(outputMap.size(), 2);
}
}
|
