2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.groupbasedpolicy.gbp_ise_adapter.impl;
11 import com.google.common.collect.Lists;
12 import com.google.common.util.concurrent.CheckedFuture;
13 import com.google.common.util.concurrent.Futures;
14 import java.util.Collections;
15 import java.util.concurrent.TimeUnit;
16 import org.junit.Before;
17 import org.junit.Test;
18 import org.junit.runner.RunWith;
19 import org.mockito.InOrder;
20 import org.mockito.Mock;
21 import org.mockito.Mockito;
22 import org.mockito.runners.MockitoJUnitRunner;
23 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
24 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
26 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.SxpMapper;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgt;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgtBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgtKey;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
35 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
38 * Test for {@link SgtToEPTemplateGeneratorImpl}.
40 @RunWith(MockitoJUnitRunner.class)
41 public class SgtToEPTemplateGeneratorImplTest {
43 public static final TenantId TENANT_ID = new TenantId("tenant-unit-1");
44 public static final Sgt SGT_1 = new Sgt(1);
45 public static final String EPG_NAME_1 = "epg-unit-1";
46 public static final Sgt SGT_2 = new Sgt(2);
47 public static final String EPG_NAME_2 = "epg-unit-2";
50 private DataBroker dataBroker;
52 private WriteTransaction wTx;
54 private SgtToEPTemplateGeneratorImpl generator;
57 public void setUp() throws Exception {
58 Mockito.when(wTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
59 Mockito.when(dataBroker.newWriteOnlyTransaction()).thenReturn(wTx);
60 generator = new SgtToEPTemplateGeneratorImpl(dataBroker);
64 public void testProcessSgtInfo() throws Exception {
65 final SgtInfo sgtInfo1 = new SgtInfo(SGT_1, EPG_NAME_1);
66 final SgtInfo sgtInfo2 = new SgtInfo(SGT_2, EPG_NAME_2);
68 final CheckedFuture<Void, TransactionCommitFailedException> outcome =
69 generator.processSgtInfo(TENANT_ID, Lists.newArrayList(sgtInfo1, sgtInfo2));
71 outcome.get(10, TimeUnit.SECONDS);
73 final KeyedInstanceIdentifier<EndpointPolicyTemplateBySgt, EndpointPolicyTemplateBySgtKey> epTemplatePath1 =
74 InstanceIdentifier.create(SxpMapper.class)
75 .child(EndpointPolicyTemplateBySgt.class, new EndpointPolicyTemplateBySgtKey(SGT_1));
76 final EndpointPolicyTemplateBySgt epTemplate1 = new EndpointPolicyTemplateBySgtBuilder()
78 .setEndpointGroups(Collections.singletonList(new EndpointGroupId(EPG_NAME_1)))
82 final KeyedInstanceIdentifier<EndpointPolicyTemplateBySgt, EndpointPolicyTemplateBySgtKey> epTemplatePath2 =
83 InstanceIdentifier.create(SxpMapper.class)
84 .child(EndpointPolicyTemplateBySgt.class, new EndpointPolicyTemplateBySgtKey(SGT_2));
85 final EndpointPolicyTemplateBySgt epTemplate2 = new EndpointPolicyTemplateBySgtBuilder()
87 .setEndpointGroups(Collections.singletonList(new EndpointGroupId(EPG_NAME_2)))
92 final InOrder inOrder = Mockito.inOrder(wTx);
93 inOrder.verify(wTx, Mockito.calls(1)).put(LogicalDatastoreType.CONFIGURATION, epTemplatePath1, epTemplate1, true);
94 inOrder.verify(wTx, Mockito.calls(1)).put(LogicalDatastoreType.CONFIGURATION, epTemplatePath2, epTemplate2, false);
95 inOrder.verify(wTx).submit();
97 Mockito.verifyZeroInteractions(wTx);