| // 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 |
| // 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 com.cloud.usage; |
| |
| import java.io.IOException; |
| |
| import org.mockito.Mockito; |
| import org.springframework.context.annotation.Bean; |
| import org.springframework.context.annotation.ComponentScan; |
| import org.springframework.context.annotation.ComponentScan.Filter; |
| import org.springframework.context.annotation.Configuration; |
| import org.springframework.context.annotation.FilterType; |
| import org.springframework.core.type.classreading.MetadataReader; |
| import org.springframework.core.type.classreading.MetadataReaderFactory; |
| import org.springframework.core.type.filter.TypeFilter; |
| |
| import org.apache.cloudstack.framework.config.dao.ConfigurationDaoImpl; |
| import org.apache.cloudstack.test.utils.SpringUtils; |
| |
| import com.cloud.alert.AlertManager; |
| import com.cloud.event.dao.UsageEventDao; |
| import com.cloud.usage.UsageManagerTestConfiguration.Library; |
| import com.cloud.usage.dao.UsageDaoImpl; |
| import com.cloud.usage.dao.UsageIPAddressDaoImpl; |
| import com.cloud.usage.dao.UsageJobDaoImpl; |
| import com.cloud.usage.dao.UsageLoadBalancerPolicyDaoImpl; |
| import com.cloud.usage.dao.UsageNetworkDaoImpl; |
| import com.cloud.usage.dao.UsageNetworkOfferingDaoImpl; |
| import com.cloud.usage.dao.UsagePortForwardingRuleDaoImpl; |
| import com.cloud.usage.dao.UsageSecurityGroupDaoImpl; |
| import com.cloud.usage.dao.UsageStorageDaoImpl; |
| import com.cloud.usage.dao.UsageVMInstanceDaoImpl; |
| import com.cloud.usage.dao.UsageVPNUserDaoImpl; |
| import com.cloud.usage.dao.UsageVmDiskDaoImpl; |
| import com.cloud.usage.dao.UsageVolumeDaoImpl; |
| import com.cloud.usage.parser.IPAddressUsageParser; |
| import com.cloud.usage.parser.LoadBalancerUsageParser; |
| import com.cloud.usage.parser.NetworkOfferingUsageParser; |
| import com.cloud.usage.parser.NetworkUsageParser; |
| import com.cloud.usage.parser.PortForwardingUsageParser; |
| import com.cloud.usage.parser.SecurityGroupUsageParser; |
| import com.cloud.usage.parser.StorageUsageParser; |
| import com.cloud.usage.parser.VMInstanceUsageParser; |
| import com.cloud.usage.parser.VPNUserUsageParser; |
| import com.cloud.usage.parser.VmDiskUsageParser; |
| import com.cloud.usage.parser.VolumeUsageParser; |
| import com.cloud.user.dao.AccountDaoImpl; |
| import com.cloud.user.dao.UserStatisticsDaoImpl; |
| |
| @Configuration |
| @ComponentScan(basePackageClasses = {AccountDaoImpl.class, UsageDaoImpl.class, UsageJobDaoImpl.class, UsageVMInstanceDaoImpl.class, UsageIPAddressDaoImpl.class, |
| UsageNetworkDaoImpl.class, UsageVolumeDaoImpl.class, UsageStorageDaoImpl.class, UsageLoadBalancerPolicyDaoImpl.class, |
| UsagePortForwardingRuleDaoImpl.class, UsageNetworkOfferingDaoImpl.class, UsageVPNUserDaoImpl.class, UsageVmDiskDaoImpl.class, |
| UsageSecurityGroupDaoImpl.class, ConfigurationDaoImpl.class, UsageManagerImpl.class, VMInstanceUsageParser.class, IPAddressUsageParser.class, |
| LoadBalancerUsageParser.class, NetworkOfferingUsageParser.class, NetworkUsageParser.class, PortForwardingUsageParser.class, |
| SecurityGroupUsageParser.class, StorageUsageParser.class, VmDiskUsageParser.class, VolumeUsageParser.class, VPNUserUsageParser.class, |
| UserStatisticsDaoImpl.class}, |
| includeFilters = {@Filter(value = Library.class, type = FilterType.CUSTOM)}, |
| useDefaultFilters = false) |
| public class UsageManagerTestConfiguration { |
| |
| @Bean |
| public AlertManager alertManager() { |
| return Mockito.mock(AlertManager.class); |
| } |
| |
| @Bean |
| public UsageEventDao usageEventDao() { |
| return Mockito.mock(UsageEventDao.class); |
| } |
| |
| public static class Library implements TypeFilter { |
| |
| @Override |
| public boolean match(MetadataReader mdr, MetadataReaderFactory arg1) throws IOException { |
| mdr.getClassMetadata().getClassName(); |
| ComponentScan cs = UsageManagerTestConfiguration.class.getAnnotation(ComponentScan.class); |
| return SpringUtils.includedInBasePackageClasses(mdr.getClassMetadata().getClassName(), cs); |
| } |
| |
| } |
| } |