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.inject.AbstractModule;
15 import java.util.concurrent.Future;
16 import org.mockito.Mockito;
17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
18 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
19 import org.opendaylight.controller.md.sal.binding.test.DataBrokerTestModule;
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.netvirt.aclservice.stats.TestOdlDirectStatisticsService;
25 import org.opendaylight.netvirt.aclservice.tests.infra.SynchronousEachOperationNewWriteTransaction;
26 import org.opendaylight.netvirt.aclservice.utils.AclClusterUtil;
27 import org.opendaylight.netvirt.aclservice.utils.AclConstants;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.direct.statistics.rev160511.OpendaylightDirectStatisticsService;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdInput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdOutput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.AllocateIdOutputBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.CreateIdPoolInput;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.DeleteIdPoolInput;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.ReleaseIdInput;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.config.rev160806.AclserviceConfig;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.config.rev160806.AclserviceConfig.SecurityGroupMode;
38 import org.opendaylight.yangtools.yang.common.RpcResult;
39 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
40 import org.slf4j.Logger;
41 import org.slf4j.LoggerFactory;
44 * Test Dependency Injection (DI) Wiring (currently through Guice).
46 * @author Michael Vorburger
48 public class AclServiceTestModule extends AbstractModule {
49 private static final Logger LOG = LoggerFactory.getLogger(AclServiceTestModule.class);
51 final SecurityGroupMode securityGroupMode;
53 public AclServiceTestModule(SecurityGroupMode securityGroupMode) {
55 this.securityGroupMode = securityGroupMode;
59 protected void configure() {
60 bind(DataBroker.class).toInstance(DataBrokerTestModule.dataBroker());
61 bind(AclserviceConfig.class).toInstance(aclServiceConfig());
63 bind(AclClusterUtil.class).toInstance(() -> true);
65 TestIMdsalApiManager singleton = TestIMdsalApiManager.newInstance();
66 bind(IMdsalApiManager.class).toInstance(singleton);
67 bind(TestIMdsalApiManager.class).toInstance(singleton);
69 bind(WriteTransaction.class).to(SynchronousEachOperationNewWriteTransaction.class);
71 bind(IdManagerService.class).toInstance(Mockito.mock(TestIdManagerService.class, realOrException()));
72 bind(OpendaylightDirectStatisticsService.class)
73 .toInstance(Mockito.mock(TestOdlDirectStatisticsService.class, realOrException()));
75 bind(JobCoordinatorEventsWaiter.class).to(TestableJobCoordinatorEventsWaiter.class);
78 private AclserviceConfig aclServiceConfig() {
79 AclserviceConfig aclServiceConfig = mock(AclserviceConfig.class);
80 Mockito.when(aclServiceConfig.getSecurityGroupMode()).thenReturn(securityGroupMode);
81 return aclServiceConfig;
84 private abstract static class TestIdManagerService implements IdManagerService {
87 public Future<RpcResult<Void>> createIdPool(CreateIdPoolInput input) {
88 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
92 public Future<RpcResult<AllocateIdOutput>> allocateId(AllocateIdInput input) {
93 String key = input.getIdKey();
94 long id = IdHelper.getId(key) == null ? AclConstants.PROTO_MATCH_PRIORITY
95 : IdHelper.getId(key);
96 AllocateIdOutputBuilder output = new AllocateIdOutputBuilder();
97 output.setIdValue(id);
99 LOG.info("ID allocated for key {}: {}", key, id);
101 RpcResultBuilder<AllocateIdOutput> allocateIdRpcBuilder = RpcResultBuilder.success();
102 allocateIdRpcBuilder.withResult(output.build());
103 return Futures.immediateFuture(allocateIdRpcBuilder.build());
107 public Future<RpcResult<Void>> releaseId(ReleaseIdInput input) {
108 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());
112 public Future<RpcResult<java.lang.Void>> deleteIdPool(DeleteIdPoolInput poolName) {
113 return Futures.immediateFuture(RpcResultBuilder.<Void>success().build());