2015-06-26 12:40:24 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
*/
|
2015-12-18 17:09:05 +01:00
|
|
|
package org.gcube.accounting.datamodel.aggregation;
|
2015-06-26 12:40:24 +02:00
|
|
|
|
2016-02-29 15:30:48 +01:00
|
|
|
import java.util.Set;
|
|
|
|
|
2015-07-01 12:50:46 +02:00
|
|
|
import org.gcube.accounting.datamodel.usagerecords.StorageUsageRecord;
|
2016-02-29 15:30:48 +01:00
|
|
|
import org.gcube.accounting.datamodel.usagerecords.StorageUsageRecordTest;
|
2015-12-18 17:09:05 +01:00
|
|
|
import org.gcube.documentstore.exception.InvalidValueException;
|
|
|
|
import org.gcube.documentstore.exception.NotAggregatableRecordsExceptions;
|
2016-02-17 14:17:13 +01:00
|
|
|
import org.gcube.testutility.ScopedTest;
|
|
|
|
import org.gcube.testutility.TestUsageRecord;
|
2015-06-26 12:40:24 +02:00
|
|
|
import org.junit.Assert;
|
|
|
|
import org.junit.Test;
|
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
|
|
|
|
/**
|
2016-10-12 14:15:28 +02:00
|
|
|
* @author Luca Frosini (ISTI - CNR)
|
2015-06-26 12:40:24 +02:00
|
|
|
*
|
|
|
|
*/
|
2016-02-12 17:08:49 +01:00
|
|
|
public class AggregatedStorageUsageRecordTest extends ScopedTest {
|
2015-06-26 12:40:24 +02:00
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
private static Logger logger = LoggerFactory.getLogger(AggregatedStorageUsageRecordTest.class);
|
2015-06-26 12:40:24 +02:00
|
|
|
|
2016-02-29 15:30:48 +01:00
|
|
|
@Test
|
|
|
|
public void testRequiredFields() throws InvalidValueException {
|
|
|
|
StorageUsageRecord storageUsageRecord = TestUsageRecord.createTestStorageUsageRecord();
|
|
|
|
Assert.assertTrue(storageUsageRecord.getScope()==null);
|
|
|
|
storageUsageRecord.setScope(TestUsageRecord.TEST_SCOPE);
|
|
|
|
|
|
|
|
AggregatedStorageUsageRecord aggregatedStorageUsageRecord = new AggregatedStorageUsageRecord(storageUsageRecord);
|
|
|
|
|
|
|
|
Set<String> expectedRequiredFields = StorageUsageRecordTest.getExpectedRequiredFields();
|
|
|
|
expectedRequiredFields.addAll(AggregatedUsageRecordTest.getExpectedRequiredFields());
|
|
|
|
|
|
|
|
logger.debug("Expected Required Fields : {}", expectedRequiredFields);
|
|
|
|
|
|
|
|
Set<String> gotRequiredFields = aggregatedStorageUsageRecord.getRequiredFields();
|
|
|
|
|
|
|
|
logger.debug("Got Required Fields : {}", gotRequiredFields);
|
|
|
|
|
|
|
|
Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields));
|
|
|
|
Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields));
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2015-06-26 12:40:24 +02:00
|
|
|
@Test
|
|
|
|
public void secondAsNotAggregated() throws InvalidValueException, NotAggregatableRecordsExceptions {
|
2016-02-10 15:14:31 +01:00
|
|
|
StorageUsageRecord storageUsageRecord = TestUsageRecord.createTestStorageUsageRecord();
|
|
|
|
Assert.assertTrue(storageUsageRecord.getScope()==null);
|
|
|
|
storageUsageRecord.setScope(TestUsageRecord.TEST_SCOPE);
|
|
|
|
|
2015-07-01 16:40:49 +02:00
|
|
|
storageUsageRecord.setResourceProperty(TestUsageRecord.TEST_PROPERTY_NAME, TestUsageRecord.TEST_PROPERTY_VALUE);
|
2015-06-26 12:40:24 +02:00
|
|
|
storageUsageRecord.validate();
|
|
|
|
logger.debug("StorageUsageRecord : {}", storageUsageRecord);
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
AggregatedStorageUsageRecord aggregated = new AggregatedStorageUsageRecord(storageUsageRecord);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("StorageUsageRecord Converted to Aggregated: {}", aggregated);
|
|
|
|
aggregated.validate();
|
|
|
|
|
2016-02-10 15:14:31 +01:00
|
|
|
StorageUsageRecord storageUsageRecord2 = TestUsageRecord.createTestStorageUsageRecord();
|
2016-02-12 12:20:04 +01:00
|
|
|
storageUsageRecord2.setScope(TestUsageRecord.TEST_SCOPE);
|
2015-06-26 12:40:24 +02:00
|
|
|
storageUsageRecord2.validate();
|
|
|
|
logger.debug("StorageUsageRecord 2 : {}", storageUsageRecord2);
|
|
|
|
|
|
|
|
long firstDataVolume = storageUsageRecord.getDataVolume();
|
|
|
|
long secondDataVolume = storageUsageRecord2.getDataVolume();
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
aggregated.aggregate(storageUsageRecord2);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("Resulting Aggregated ServiceUsageRecord: {}", aggregated);
|
|
|
|
aggregated.validate();
|
|
|
|
|
2015-06-26 15:52:17 +02:00
|
|
|
Assert.assertTrue(aggregated.getDataVolume() == (firstDataVolume + secondDataVolume));
|
2015-06-26 12:40:24 +02:00
|
|
|
Assert.assertTrue(aggregated.getOperationCount() == 2);
|
|
|
|
|
2015-07-01 16:40:49 +02:00
|
|
|
Assert.assertFalse(aggregated.getResourceProperties().containsKey(TestUsageRecord.TEST_PROPERTY_NAME));
|
2015-06-26 12:40:24 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void secondAsAggregated() throws InvalidValueException, NotAggregatableRecordsExceptions {
|
2016-02-10 15:14:31 +01:00
|
|
|
StorageUsageRecord storageUsageRecord = TestUsageRecord.createTestStorageUsageRecord();
|
|
|
|
Assert.assertTrue(storageUsageRecord.getScope()==null);
|
|
|
|
storageUsageRecord.setScope(TestUsageRecord.TEST_SCOPE);
|
2015-06-26 12:40:24 +02:00
|
|
|
storageUsageRecord.validate();
|
|
|
|
logger.debug("StorageUsageRecord : {}", storageUsageRecord);
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
AggregatedStorageUsageRecord aggregated = new AggregatedStorageUsageRecord(storageUsageRecord);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("StorageUsageRecord Converted to Aggregated: {}", aggregated);
|
|
|
|
aggregated.validate();
|
|
|
|
|
2016-02-10 15:14:31 +01:00
|
|
|
StorageUsageRecord storageUsageRecord2 = TestUsageRecord.createTestStorageUsageRecord();
|
|
|
|
Assert.assertTrue(storageUsageRecord2.getScope()==null);
|
|
|
|
storageUsageRecord2.setScope(TestUsageRecord.TEST_SCOPE);
|
2015-07-01 16:40:49 +02:00
|
|
|
storageUsageRecord2.setResourceProperty(TestUsageRecord.TEST_PROPERTY_NAME, TestUsageRecord.TEST_PROPERTY_VALUE);
|
2015-06-26 12:40:24 +02:00
|
|
|
storageUsageRecord2.validate();
|
|
|
|
logger.debug("StorageUsageRecord 2 : {}", storageUsageRecord2);
|
2015-12-18 17:09:05 +01:00
|
|
|
AggregatedStorageUsageRecord converted = new AggregatedStorageUsageRecord(storageUsageRecord2);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("StorageUsageRecord Converted to Aggregated: {}", converted);
|
|
|
|
converted.validate();
|
|
|
|
|
|
|
|
long firstDataVolume = aggregated.getDataVolume();
|
|
|
|
long secondDataVolume = converted.getDataVolume();
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
aggregated.aggregate(converted);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("Resulting Aggregated StorageUsageRecord: {}", aggregated);
|
|
|
|
aggregated.validate();
|
|
|
|
|
2015-06-26 15:52:17 +02:00
|
|
|
Assert.assertTrue(aggregated.getDataVolume() == (firstDataVolume + secondDataVolume));
|
2015-06-26 12:40:24 +02:00
|
|
|
Assert.assertTrue(aggregated.getOperationCount() == 2);
|
2015-07-01 16:40:49 +02:00
|
|
|
Assert.assertFalse(aggregated.getResourceProperties().containsKey(TestUsageRecord.TEST_PROPERTY_NAME));
|
2015-06-26 12:40:24 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void aggregationStressTest() throws InvalidValueException, NotAggregatableRecordsExceptions {
|
2016-02-10 15:14:31 +01:00
|
|
|
StorageUsageRecord storageUsageRecord = TestUsageRecord.createTestStorageUsageRecord();
|
|
|
|
Assert.assertTrue(storageUsageRecord.getScope()==null);
|
|
|
|
storageUsageRecord.setScope(TestUsageRecord.TEST_SCOPE);
|
2015-07-01 16:40:49 +02:00
|
|
|
storageUsageRecord.setResourceProperty(TestUsageRecord.TEST_PROPERTY_NAME, TestUsageRecord.TEST_PROPERTY_VALUE);
|
2015-06-26 12:40:24 +02:00
|
|
|
storageUsageRecord.validate();
|
|
|
|
logger.debug("StorageUsageRecord : {}", storageUsageRecord);
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
AggregatedStorageUsageRecord aggregated = new AggregatedStorageUsageRecord(storageUsageRecord);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("StorageUsageRecord Converted to Aggregated: {}", aggregated);
|
|
|
|
aggregated.validate();
|
|
|
|
|
|
|
|
for(int i=2; i<1002; i++){
|
|
|
|
|
2016-02-10 15:14:31 +01:00
|
|
|
StorageUsageRecord sur = TestUsageRecord.createTestStorageUsageRecord();
|
2016-02-12 12:20:04 +01:00
|
|
|
sur.setScope(TestUsageRecord.TEST_SCOPE);
|
2015-07-01 16:40:49 +02:00
|
|
|
sur.setResourceProperty(TestUsageRecord.TEST_PROPERTY_NAME, TestUsageRecord.TEST_PROPERTY_VALUE);
|
2015-06-26 12:40:24 +02:00
|
|
|
sur.validate();
|
|
|
|
logger.debug("Cycle StorageUsageRecord {}: {}", i, sur);
|
|
|
|
|
|
|
|
long oldDataVolume = aggregated.getDataVolume();
|
|
|
|
long newDataVolume = sur.getDataVolume();
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
aggregated.aggregate(sur);
|
2015-06-26 12:40:24 +02:00
|
|
|
logger.debug("Resulting Aggregated StorageUsageRecord : {}", aggregated);
|
|
|
|
aggregated.validate();
|
|
|
|
|
2015-06-26 15:52:17 +02:00
|
|
|
Assert.assertTrue(aggregated.getDataVolume() == (oldDataVolume + newDataVolume));
|
2015-06-26 12:40:24 +02:00
|
|
|
Assert.assertTrue(aggregated.getOperationCount() == i);
|
2015-07-01 16:40:49 +02:00
|
|
|
Assert.assertFalse(aggregated.getResourceProperties().containsKey(TestUsageRecord.TEST_PROPERTY_NAME));
|
2015-06-26 12:40:24 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
logger.debug("Resulting Aggregated StorageUsageRecord: {}", aggregated);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|