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.sxp_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.concurrent.TimeUnit;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.junit.runner.RunWith;
18 import org.mockito.InOrder;
19 import org.mockito.Mock;
20 import org.mockito.Mockito;
21 import org.mockito.runners.MockitoJUnitRunner;
22 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
23 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
24 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
25 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
26 import org.opendaylight.groupbasedpolicy.util.IidFactory;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.Description;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.Name;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.policy.EndpointGroup;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.tenants.tenant.policy.EndpointGroupBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
37 * Test for {@link SgtToEpgGeneratorImpl}.
39 @RunWith(MockitoJUnitRunner.class)
40 public class SgtToEpgGeneratorImplTest {
42 public static final TenantId TENANT_ID = new TenantId("tenant-unit-1");
43 public static final Sgt SGT_1 = new Sgt(1);
44 public static final String SGT_1_UUID = "uuid-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 SGT_2_UUID = "uuid-2";
48 public static final String EPG_NAME_2 = "epg-unit-2";
51 private DataBroker dataBroker;
53 private WriteTransaction wTx;
55 private SgtToEpgGeneratorImpl generator;
58 public void setUp() throws Exception {
59 Mockito.when(wTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
60 Mockito.when(dataBroker.newWriteOnlyTransaction()).thenReturn(wTx);
61 generator = new SgtToEpgGeneratorImpl(dataBroker);
65 public void testProcessSgtInfo() throws Exception {
66 final SgtInfo sgtInfo1 = new SgtInfo(SGT_1, EPG_NAME_1, SGT_1_UUID);
67 final SgtInfo sgtInfo2 = new SgtInfo(SGT_2, EPG_NAME_2, SGT_2_UUID);
69 final CheckedFuture<Void, TransactionCommitFailedException> outcome =
70 generator.processSgtInfo(TENANT_ID, Lists.newArrayList(sgtInfo1, sgtInfo2));
72 outcome.get(10, TimeUnit.SECONDS);
74 final EndpointGroupId epgId1 = new EndpointGroupId(EPG_NAME_1);
75 final InstanceIdentifier<EndpointGroup> epgPath1 = IidFactory.endpointGroupIid(TENANT_ID, epgId1);
76 final EndpointGroup epg1 = createEpg(epgId1, SGT_1.getValue());
78 final EndpointGroupId epgId2 = new EndpointGroupId(EPG_NAME_2);
79 final InstanceIdentifier<EndpointGroup> epgPath2 = IidFactory.endpointGroupIid(TENANT_ID, epgId2);
80 final EndpointGroup epg2 = createEpg(epgId2, SGT_2.getValue());
82 final InOrder inOrder = Mockito.inOrder(wTx);
83 inOrder.verify(wTx, Mockito.calls(1)).put(LogicalDatastoreType.CONFIGURATION, epgPath1, epg1, true);
84 inOrder.verify(wTx, Mockito.calls(1)).put(LogicalDatastoreType.CONFIGURATION, epgPath2, epg2, false);
85 inOrder.verify(wTx).submit();
87 Mockito.verifyZeroInteractions(wTx);
90 private EndpointGroup createEpg(final EndpointGroupId epgId1, final int sgt) {
91 final String epgIdValue = epgId1.getValue();
92 return new EndpointGroupBuilder()
94 .setName(new Name(epgIdValue + "--"+sgt))
95 .setDescription(new Description("imported from ISE for sgt="+sgt))