blob: f3f3d154aa5e82941f640a02f7e0882f8e431f58 [file] [log] [blame]
/*
* 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.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);
}
}