2 * Copyright (c) 2016 Red Hat, 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
8 package org.opendaylight.netvirt.aclservice.tests;
10 import static org.mockito.Mockito.mock;
11 import static org.opendaylight.yangtools.testutils.mockito.MoreAnswers.realOrException;
13 import com.google.common.util.concurrent.Futures;
14 import com.google.common.util.concurrent.ListenableFuture;
15 import com.google.common.util.concurrent.ListeningExecutorService;
16 import com.google.common.util.concurrent.MoreExecutors;
17 import com.google.inject.AbstractModule;
18 import java.util.concurrent.Executors;
19 import org.mockito.Mockito;
20 import org.opendaylight.genius.datastoreutils.testutils.JobCoordinatorEventsWaiter;
21 import org.opendaylight.genius.datastoreutils.testutils.TestableJobCoordinatorEventsWaiter;
22 import org.opendaylight.genius.mdsalutil.interfaces.IMdsalApiManager;
23 import org.opendaylight.genius.mdsalutil.interfaces.testutils.TestIMdsalApiManager;
24 import org.opendaylight.mdsal.binding.api.DataBroker;
25 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractBaseDataBrokerTest;
26 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractDataBrokerTestCustomizer;
27 import org.opendaylight.netvirt.aclservice.AclInterfaceCacheImpl;
28 import org.opendaylight.netvirt.aclservice.api.AclInterfaceCache;
29 import org.opendaylight.netvirt.aclservice.stats.TestOdlDirectStatisticsService;
30 import org.opendaylight.netvirt.aclservice.utils.AclClusterUtil;
31 import org.opendaylight.netvirt.aclservice.utils.AclConstants;
32 import org.opendaylight.netvirt.aclservice.utils.AclDataUtil;
33 import org.opendaylight.serviceutils.srm.ServiceRecoveryRegistry;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.OpendaylightDirectStatisticsService;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdInput;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdOutput;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdOutputBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.CreateIdPoolInput;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.CreateIdPoolOutput;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.CreateIdPoolOutputBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.DeleteIdPoolInput;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.DeleteIdPoolOutput;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.DeleteIdPoolOutputBuilder;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.ReleaseIdInput;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.ReleaseIdOutput;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.ReleaseIdOutputBuilder;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.config.rev160806.AclserviceConfig;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.config.rev160806.AclserviceConfig.SecurityGroupMode;
50 import org.opendaylight.yangtools.yang.common.RpcResult;
51 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
52 import org.slf4j.Logger;
53 import org.slf4j.LoggerFactory;
56 * Test Dependency Injection (DI) Wiring (currently through Guice).
58 * @author Michael Vorburger
60 public class AclServiceTestModule extends AbstractModule {
61 private static final Logger LOG = LoggerFactory.getLogger(AclServiceTestModule.class);
63 final SecurityGroupMode securityGroupMode;
65 public AclServiceTestModule(SecurityGroupMode securityGroupMode) {
66 this.securityGroupMode = securityGroupMode;
69 @SuppressWarnings("checkstyle:IllegalCatch")
71 protected void configure() {
72 AbstractBaseDataBrokerTest test = new AbstractBaseDataBrokerTest() {
74 protected AbstractDataBrokerTestCustomizer createDataBrokerTestCustomizer() {
75 return new AbstractDataBrokerTestCustomizer() {
77 public ListeningExecutorService getCommitCoordinatorExecutor() {
78 return MoreExecutors.listeningDecorator(Executors.newCachedThreadPool());
85 } catch (Exception e) {
86 LOG.trace("Exception", e.getMessage());
88 bind(DataBroker.class).toInstance(test.getDataBroker());
89 bind(AclserviceConfig.class).toInstance(aclServiceConfig());
90 bind(AclDataUtil.class).toInstance(aclDataUtil());
91 bind(AclClusterUtil.class).toInstance(() -> true);
93 TestIMdsalApiManager singleton = TestIMdsalApiManager.newInstance();
94 bind(IMdsalApiManager.class).toInstance(singleton);
95 bind(TestIMdsalApiManager.class).toInstance(singleton);
97 bind(IdManagerService.class).toInstance(Mockito.mock(TestIdManagerService.class, realOrException()));
98 bind(OpendaylightDirectStatisticsService.class)
99 .toInstance(Mockito.mock(TestOdlDirectStatisticsService.class, realOrException()));
101 bind(JobCoordinatorEventsWaiter.class).to(TestableJobCoordinatorEventsWaiter.class);
103 bind(AclInterfaceCache.class).toInstance(new AclInterfaceCacheImpl());
104 bind(ServiceRecoveryRegistry.class).toInstance(mock(ServiceRecoveryRegistry.class));
107 private AclserviceConfig aclServiceConfig() {
108 AclserviceConfig aclServiceConfig = mock(AclserviceConfig.class);
109 Mockito.when(aclServiceConfig.getSecurityGroupMode()).thenReturn(securityGroupMode);
110 return aclServiceConfig;
113 private AclDataUtil aclDataUtil() {
114 AclDataUtil aclDataUtil = new AclDataUtil();
115 aclDataUtil.addAclTag("85cc3048-abc3-43cc-89b3-377341426ac5", 2);
116 aclDataUtil.addAclTag("85cc3048-abc3-43cc-89b3-377341426ac8", 4);
120 private abstract static class TestIdManagerService implements IdManagerService {
123 public ListenableFuture<RpcResult<CreateIdPoolOutput>> createIdPool(CreateIdPoolInput input) {
124 return RpcResultBuilder.success(new CreateIdPoolOutputBuilder().build()).buildFuture();
128 public ListenableFuture<RpcResult<AllocateIdOutput>> allocateId(AllocateIdInput input) {
129 String key = input.getIdKey();
130 long id = IdHelper.getId(key) == null ? AclConstants.PROTO_MATCH_PRIORITY
131 : IdHelper.getId(key);
132 AllocateIdOutputBuilder output = new AllocateIdOutputBuilder();
133 output.setIdValue(id);
135 LOG.info("ID allocated for key {}: {}", key, id);
137 RpcResultBuilder<AllocateIdOutput> allocateIdRpcBuilder = RpcResultBuilder.success();
138 allocateIdRpcBuilder.withResult(output.build());
139 return Futures.immediateFuture(allocateIdRpcBuilder.build());
143 public ListenableFuture<RpcResult<ReleaseIdOutput>> releaseId(ReleaseIdInput input) {
144 return RpcResultBuilder.success(new ReleaseIdOutputBuilder().build()).buildFuture();
148 public ListenableFuture<RpcResult<DeleteIdPoolOutput>> deleteIdPool(DeleteIdPoolInput poolName) {
149 return RpcResultBuilder.success(new DeleteIdPoolOutputBuilder().build()).buildFuture();