| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.rocketmq.eventbridge.domain.service; |
| |
| import com.google.common.collect.Lists; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.UUID; |
| import org.apache.rocketmq.eventbridge.domain.common.enums.AuthorizationTypeEnum; |
| import org.apache.rocketmq.eventbridge.domain.common.enums.NetworkTypeEnum; |
| import org.apache.rocketmq.eventbridge.domain.model.PaginationResult; |
| import org.apache.rocketmq.eventbridge.domain.model.connection.ConnectionDTO; |
| import org.apache.rocketmq.eventbridge.domain.model.connection.ConnectionService; |
| import org.apache.rocketmq.eventbridge.domain.model.connection.parameter.AuthParameters; |
| import org.apache.rocketmq.eventbridge.domain.model.connection.parameter.BasicAuthParameters; |
| import org.apache.rocketmq.eventbridge.domain.model.connection.parameter.NetworkParameters; |
| import org.apache.rocketmq.eventbridge.domain.model.quota.QuotaService; |
| import org.apache.rocketmq.eventbridge.domain.repository.ApiDestinationRepository; |
| import org.apache.rocketmq.eventbridge.domain.repository.ConnectionRepository; |
| import org.apache.rocketmq.eventbridge.domain.rpc.NetworkServiceAPI; |
| import org.apache.rocketmq.eventbridge.domain.rpc.SecretManagerAPI; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Ignore; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.mockito.InjectMocks; |
| import org.mockito.Mock; |
| import org.mockito.Mockito; |
| import org.mockito.junit.MockitoJUnitRunner; |
| |
| import static org.mockito.ArgumentMatchers.any; |
| import static org.mockito.ArgumentMatchers.anyInt; |
| import static org.mockito.ArgumentMatchers.anyString; |
| |
| @RunWith(MockitoJUnitRunner.class) |
| @Ignore |
| public class ConnectionServiceTest { |
| |
| @InjectMocks |
| private ConnectionService connectionService; |
| @Mock |
| private ConnectionRepository connectionRepository; |
| @Mock |
| private SecretManagerAPI secretManagerAPI; |
| @Mock |
| private NetworkServiceAPI networkServiceAPI; |
| @Mock |
| private ApiDestinationRepository apiDestinationRepository; |
| @Mock |
| private QuotaService quotaService; |
| |
| @Before |
| public void testBefore() throws Exception { |
| Mockito.when(connectionRepository.createConnection(any())).thenReturn(Boolean.TRUE); |
| Mockito.when(connectionRepository.listConnections(anyString(), anyString(), anyString(), anyInt())).thenReturn(new ArrayList<>()); |
| Mockito.when(connectionRepository.getConnectionCount(any(), any())).thenReturn(8); |
| ConnectionDTO connectionDTO = new ConnectionDTO(); |
| connectionDTO.setConnectionName(UUID.randomUUID().toString()); |
| NetworkParameters networkParameters = new NetworkParameters(); |
| networkParameters.setNetworkType(NetworkTypeEnum.PUBLIC_NETWORK.getNetworkType()); |
| connectionDTO.setNetworkParameters(networkParameters); |
| List<ConnectionDTO> connectionDTOS = Lists.newArrayList(); |
| connectionDTOS.add(connectionDTO); |
| Mockito.when(connectionRepository.getConnection(any(), any())).thenReturn(connectionDTOS); |
| Mockito.when(quotaService.getTotalQuota(any(), any())).thenReturn(10); |
| } |
| |
| @Test |
| public void testCreateConnection() throws Exception { |
| Mockito.when(connectionRepository.getConnection(any(), any())).thenReturn(null); |
| ConnectionDTO connectionDTO = new ConnectionDTO(); |
| connectionDTO.setConnectionName(UUID.randomUUID().toString()); |
| connectionDTO.setDescription(UUID.randomUUID().toString()); |
| NetworkParameters networkParameters = new NetworkParameters(); |
| networkParameters.setNetworkType(NetworkTypeEnum.PUBLIC_NETWORK.getNetworkType()); |
| networkParameters.setSecurityGroupId(UUID.randomUUID().toString()); |
| networkParameters.setVpcId(UUID.randomUUID().toString()); |
| networkParameters.setVswitcheId(UUID.randomUUID().toString()); |
| connectionDTO.setNetworkParameters(networkParameters); |
| AuthParameters authParameters = new AuthParameters(); |
| BasicAuthParameters basicAuthParameters = new BasicAuthParameters(); |
| basicAuthParameters.setPassword(UUID.randomUUID().toString()); |
| basicAuthParameters.setUsername(UUID.randomUUID().toString()); |
| authParameters.setBasicAuthParameters(basicAuthParameters); |
| authParameters.setAuthorizationType(AuthorizationTypeEnum.BASIC_AUTH.getType()); |
| connectionDTO.setAuthParameters(authParameters); |
| final String connection = connectionService.createConnection(connectionDTO); |
| Assert.assertNotNull(connection); |
| } |
| |
| @Test |
| public void testDeleteConnection() throws Exception { |
| Mockito.when(connectionRepository.deleteConnection(any(), any())).thenReturn(Boolean.TRUE); |
| Mockito.when(apiDestinationRepository.queryApiDestinationByConnectionName(any(), any())).thenReturn(null); |
| connectionService.deleteConnection(UUID.randomUUID().toString(), UUID.randomUUID().toString()); |
| } |
| |
| @Test |
| public void testUpdateConnection() { |
| ConnectionDTO connectionDTO = new ConnectionDTO(); |
| connectionDTO.setConnectionName(UUID.randomUUID().toString()); |
| NetworkParameters networkParameters = new NetworkParameters(); |
| networkParameters.setNetworkType(NetworkTypeEnum.PUBLIC_NETWORK.getNetworkType()); |
| networkParameters.setSecurityGroupId(UUID.randomUUID().toString()); |
| networkParameters.setVpcId(UUID.randomUUID().toString()); |
| networkParameters.setVswitcheId(UUID.randomUUID().toString()); |
| connectionDTO.setNetworkParameters(networkParameters); |
| AuthParameters authParameters = new AuthParameters(); |
| BasicAuthParameters basicAuthParameters = new BasicAuthParameters(); |
| basicAuthParameters.setPassword(UUID.randomUUID().toString()); |
| basicAuthParameters.setUsername(UUID.randomUUID().toString()); |
| authParameters.setBasicAuthParameters(basicAuthParameters); |
| authParameters.setAuthorizationType(AuthorizationTypeEnum.BASIC_AUTH.getType()); |
| connectionDTO.setAuthParameters(authParameters); |
| Mockito.when(connectionRepository.updateConnection(any())).thenReturn(Boolean.TRUE); |
| Mockito.when(connectionRepository.getConnectionByNameAccountId(anyString(), anyString())).thenReturn(connectionDTO); |
| connectionService.updateConnection(connectionDTO, UUID.randomUUID().toString()); |
| } |
| |
| @Test |
| public void testGetConnection() { |
| final List<ConnectionDTO> connectionDTO = connectionService.getConnection(UUID.randomUUID().toString(), UUID.randomUUID().toString()); |
| Assert.assertNotNull(connectionDTO); |
| } |
| |
| @Test |
| public void testListConnections() { |
| final PaginationResult<List<ConnectionDTO>> listPaginationResult = connectionService.listConnections(UUID.randomUUID().toString(), UUID.randomUUID().toString(), "0", 10); |
| Assert.assertNotNull(listPaginationResult.getData()); |
| } |
| } |