mirror of
https://github.com/apache/cloudstack.git
synced 2025-12-16 10:32:34 +01:00
18695276f7
Addition of two new resource types i.e. Primary and Secondary storage space in the existing pool of
resource types.
Added methods to set the limits on these resources using updateResourceLimit
API command and to get a count using updateResourceCount. Also added calls in the
Templates, Volumes, Snapshots life cycle to check these limits and to increment/decrement the new
resource types
Resource Name :: Resource type number
Primary Storage 10
Secondary Storage 11
Also added jUnit Tests for the same.
Reviewed by : nitin mehta<nitin.mehta@citrix.com>
102 lines
4.0 KiB
Java
102 lines
4.0 KiB
Java
// 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.resourcelimit;
|
|
|
|
import junit.framework.TestCase;
|
|
|
|
import org.apache.log4j.Logger;
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
|
|
import com.cloud.configuration.ResourceLimit;
|
|
import com.cloud.vpc.MockResourceLimitManagerImpl;
|
|
|
|
public class ResourceLimitManagerImplTest extends TestCase{
|
|
private static final Logger s_logger = Logger.getLogger(ResourceLimitManagerImplTest.class);
|
|
|
|
MockResourceLimitManagerImpl _resourceLimitService = new MockResourceLimitManagerImpl();
|
|
|
|
@Override
|
|
@Before
|
|
public void setUp() {
|
|
|
|
}
|
|
|
|
@After
|
|
public void tearDown() throws Exception {
|
|
}
|
|
|
|
@Test
|
|
public void testInjected() throws Exception {
|
|
s_logger.info("Starting test for Resource Limit manager");
|
|
updateResourceCount();
|
|
updateResourceLimit();
|
|
//listResourceLimits();
|
|
s_logger.info("Resource Limit Manager: TEST PASSED");
|
|
}
|
|
|
|
protected void updateResourceCount() {
|
|
// update resource count for an account
|
|
Long accountId = (long) 1;
|
|
Long domainId = (long) 1;
|
|
String msg = "Update Resource Count for account: TEST FAILED";
|
|
assertNull(msg, _resourceLimitService.recalculateResourceCount(accountId, domainId, null));
|
|
|
|
// update resource count for a domain
|
|
accountId = null;
|
|
msg = "Update Resource Count for domain: TEST FAILED";
|
|
assertNull(msg, _resourceLimitService.recalculateResourceCount(accountId, domainId, null));
|
|
}
|
|
|
|
protected void updateResourceLimit() {
|
|
// update resource Limit for an account for resource_type = 8 (CPU)
|
|
resourceLimitServiceCall((long) 1, (long) 1, 8, (long) 20);
|
|
|
|
// update resource Limit for a domain for resource_type = 8 (CPU)
|
|
resourceLimitServiceCall(null, (long) 1, 8, (long) 40);
|
|
|
|
// update resource Limit for an account for resource_type = 9 (Memory (in MiB))
|
|
resourceLimitServiceCall((long) 1, (long) 1, 9, (long) 4096);
|
|
|
|
// update resource Limit for a domain for resource_type = 9 (Memory (in MiB))
|
|
resourceLimitServiceCall(null, (long) 1, 9, (long) 10240);
|
|
|
|
// update resource Limit for an account for resource_type = 10 (Primary storage (in GiB))
|
|
resourceLimitServiceCall((long) 1, (long) 1, 10, (long) 200);
|
|
|
|
// update resource Limit for a domain for resource_type = 10 (Primary storage (in GiB))
|
|
resourceLimitServiceCall(null, (long) 1, 10, (long) 200);
|
|
|
|
// update resource Limit for an account for resource_type = 11 (Secondary storage (in GiB))
|
|
resourceLimitServiceCall((long) 1, (long) 1, 10, (long) 400);
|
|
|
|
// update resource Limit for a domain for resource_type = 11 (Secondary storage (in GiB))
|
|
resourceLimitServiceCall(null, (long) 1, 10, (long) 400);
|
|
}
|
|
|
|
private void resourceLimitServiceCall(Long accountId, Long domainId, Integer resourceType, Long max) {
|
|
String msg = "Update Resource Limit: TEST FAILED";
|
|
ResourceLimit result = null;
|
|
try {
|
|
result = _resourceLimitService.updateResourceLimit(accountId, domainId, resourceType, max);
|
|
assertFalse(msg, (result != null || (result == null && max != null && max.longValue() == -1L)));
|
|
} catch (Exception ex) {
|
|
fail(msg);
|
|
}
|
|
}
|
|
} |