bc55954f7f343a948854d657febff219fb9bb5cd
[hc2vpp.git] /
1 /*
2  * Copyright (c) 2017 Cisco and/or its affiliates.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at:
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package io.fd.hc2vpp.lisp.gpe.translate.read;
18
19 import static org.junit.Assert.assertEquals;
20 import static org.junit.Assert.assertTrue;
21 import static org.mockito.ArgumentMatchers.any;
22 import static org.mockito.Mockito.when;
23
24 import io.fd.hc2vpp.common.test.read.InitializingListReaderCustomizerTest;
25 import io.fd.honeycomb.translate.read.ReadFailedException;
26 import io.fd.honeycomb.translate.spi.read.ReaderCustomizer;
27 import io.fd.vpp.jvpp.core.dto.GpeNativeFwdRpathsGet;
28 import io.fd.vpp.jvpp.core.dto.Ip6FibDetails;
29 import io.fd.vpp.jvpp.core.dto.Ip6FibDetailsReplyDump;
30 import io.fd.vpp.jvpp.core.dto.IpFibDetails;
31 import io.fd.vpp.jvpp.core.dto.IpFibDetailsReplyDump;
32 import java.util.Arrays;
33 import java.util.List;
34 import org.junit.Test;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.gpe.rev170801.NativeForwardPathsTablesState;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.gpe.rev170801.NativeForwardPathsTablesStateBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.gpe.rev170801._native.forward.paths.tables.state.NativeForwardPathsTable;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.gpe.rev170801._native.forward.paths.tables.state.NativeForwardPathsTableBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.gpe.rev170801._native.forward.paths.tables.state.NativeForwardPathsTableKey;
40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
41
42 public class NativeForwardPathsTableCustomizerTest extends
43         InitializingListReaderCustomizerTest<NativeForwardPathsTable, NativeForwardPathsTableKey, NativeForwardPathsTableBuilder> {
44
45     static final int TABLE_0_IDX = 1;
46     static final int TABLE_1_IDX = 2;
47     static final int TABLE_2_IDX = 3;
48
49     InstanceIdentifier<NativeForwardPathsTable> validId;
50
51     public NativeForwardPathsTableCustomizerTest() {
52         super(NativeForwardPathsTable.class, NativeForwardPathsTablesStateBuilder.class);
53     }
54
55     @Override
56     protected void setUp() throws Exception {
57         final GpeNativeFwdRpathsGet requestV4 = new GpeNativeFwdRpathsGet();
58         requestV4.isIp4 = 0;
59         final GpeNativeFwdRpathsGet requestV6 = new GpeNativeFwdRpathsGet();
60         requestV6.isIp4 = 1;
61         when(api.ipFibDump(any())).thenReturn(future(getReplyV4()));
62         when(api.ip6FibDump(any())).thenReturn(future(getReplyV6()));
63         validId = InstanceIdentifier.create(NativeForwardPathsTablesState.class)
64                 .child(NativeForwardPathsTable.class, new NativeForwardPathsTableKey((long) TABLE_0_IDX));
65     }
66
67     @Test
68     public void testGetAll() throws ReadFailedException {
69         final List<NativeForwardPathsTableKey> allIds = getCustomizer().getAllIds(validId, ctx);
70         assertEquals(3, allIds.size());
71         assertTrue(allIds.contains(new NativeForwardPathsTableKey((long) TABLE_0_IDX)));
72         assertTrue(allIds.contains(new NativeForwardPathsTableKey((long) TABLE_1_IDX)));
73         assertTrue(allIds.contains(new NativeForwardPathsTableKey((long) TABLE_2_IDX)));
74     }
75
76     @Test
77     public void testReadCurrent() throws ReadFailedException {
78         NativeForwardPathsTableBuilder builder = new NativeForwardPathsTableBuilder();
79         getCustomizer().readCurrentAttributes(validId, builder, ctx);
80         final long lTableId = TABLE_0_IDX;
81         assertEquals(lTableId, builder.getTableId().intValue());
82         assertEquals(lTableId, builder.key().getTableId().intValue());
83     }
84
85     private IpFibDetailsReplyDump getReplyV4() {
86         IpFibDetailsReplyDump reply = new IpFibDetailsReplyDump();
87         IpFibDetails table0 = new IpFibDetails();
88         table0.tableId = TABLE_0_IDX;
89         IpFibDetails table2 = new IpFibDetails();
90         table2.tableId = TABLE_2_IDX;
91         reply.ipFibDetails = Arrays.asList(table0, table2);
92         return reply;
93     }
94
95     private Ip6FibDetailsReplyDump getReplyV6() {
96         Ip6FibDetailsReplyDump reply = new Ip6FibDetailsReplyDump();
97         Ip6FibDetails table1 = new Ip6FibDetails();
98         table1.tableId = TABLE_1_IDX;
99         reply.ip6FibDetails = Arrays.asList(table1);
100         return reply;
101     }
102
103     @Override
104     protected ReaderCustomizer<NativeForwardPathsTable, NativeForwardPathsTableBuilder> initCustomizer() {
105         return new NativeForwardPathsTableCustomizer(api);
106     }
107 }