2015-07-27 12:45:23 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
package org.gcube.accounting.datamodel.usagerecords;
|
|
|
|
|
|
|
|
import java.util.HashSet;
|
|
|
|
import java.util.Set;
|
|
|
|
|
2015-12-18 17:09:05 +01:00
|
|
|
import org.gcube.accounting.datamodel.UsageRecord;
|
2016-01-28 18:29:11 +01:00
|
|
|
import org.gcube.accounting.datamodel.basetypes.AbstractStorageUsageRecord;
|
2015-07-27 12:45:23 +02:00
|
|
|
import org.gcube.accounting.datamodel.basetypes.TestUsageRecord;
|
2016-02-09 18:55:16 +01:00
|
|
|
import org.gcube.accounting.testutility.TestUtility;
|
2016-02-09 18:46:23 +01:00
|
|
|
import org.gcube.common.authorization.library.provider.SecurityTokenProvider;
|
2015-12-18 17:09:05 +01:00
|
|
|
import org.gcube.documentstore.exception.InvalidValueException;
|
|
|
|
import org.gcube.documentstore.records.Record;
|
2015-10-20 14:40:30 +02:00
|
|
|
import org.junit.After;
|
2015-07-27 12:45:23 +02:00
|
|
|
import org.junit.Assert;
|
2015-10-20 14:40:30 +02:00
|
|
|
import org.junit.Before;
|
2015-07-27 12:45:23 +02:00
|
|
|
import org.junit.Test;
|
2015-10-09 14:52:42 +02:00
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
2015-07-27 12:45:23 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* @author Luca Frosini (ISTI - CNR) http://www.lucafrosini.com/
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
public class StorageUsageRecordTest {
|
2015-10-09 14:52:42 +02:00
|
|
|
|
|
|
|
private static Logger logger = LoggerFactory.getLogger(StorageUsageRecordTest.class);
|
|
|
|
|
2015-10-20 14:40:30 +02:00
|
|
|
@Before
|
2016-02-09 18:46:23 +01:00
|
|
|
public void before() throws Exception{
|
2016-02-09 18:55:16 +01:00
|
|
|
SecurityTokenProvider.instance.set(TestUtility.TOKEN);
|
2015-10-20 14:40:30 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@After
|
2016-02-09 18:55:16 +01:00
|
|
|
public void after() throws Exception{
|
2016-02-09 18:46:23 +01:00
|
|
|
SecurityTokenProvider.instance.reset();
|
2015-10-20 14:40:30 +02:00
|
|
|
}
|
|
|
|
|
2015-07-27 12:45:23 +02:00
|
|
|
public static Set<String> getExpectedRequiredFields(){
|
|
|
|
Set<String> expectedRequiredFields = new HashSet<String>();
|
2015-12-18 17:09:05 +01:00
|
|
|
expectedRequiredFields.add(Record.ID);
|
|
|
|
expectedRequiredFields.add(UsageRecord.CONSUMER_ID);
|
|
|
|
expectedRequiredFields.add(UsageRecord.CREATION_TIME);
|
|
|
|
expectedRequiredFields.add(UsageRecord.RECORD_TYPE);
|
|
|
|
expectedRequiredFields.add(UsageRecord.SCOPE);
|
|
|
|
expectedRequiredFields.add(UsageRecord.OPERATION_RESULT);
|
2016-01-28 18:29:11 +01:00
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.RESOURCE_OWNER);
|
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.RESOURCE_SCOPE);
|
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.RESOURCE_URI);
|
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.PROVIDER_URI);
|
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.OPERATION_TYPE);
|
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.DATA_TYPE);
|
|
|
|
expectedRequiredFields.add(AbstractStorageUsageRecord.DATA_VOLUME);
|
|
|
|
|
2015-07-27 12:45:23 +02:00
|
|
|
return expectedRequiredFields;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Test(expected=InvalidValueException.class)
|
|
|
|
public void scopeNotSetValidationError() throws InvalidValueException {
|
2016-02-09 18:46:23 +01:00
|
|
|
SecurityTokenProvider.instance.reset();
|
2016-01-28 18:29:11 +01:00
|
|
|
StorageUsageRecord usageRecord = TestUsageRecord.createTestStorageUsageRecordAutomaticScope();
|
|
|
|
usageRecord.validate();
|
|
|
|
logger.debug("{}", usageRecord);
|
2015-07-27 12:45:23 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testRequiredFields() throws InvalidValueException{
|
2016-01-28 18:29:11 +01:00
|
|
|
StorageUsageRecord usageRecord = TestUsageRecord.createTestStorageUsageRecordAutomaticScope();
|
2015-10-20 14:40:30 +02:00
|
|
|
|
2015-07-27 12:45:23 +02:00
|
|
|
Set<String> expectedRequiredFields = getExpectedRequiredFields();
|
2015-12-18 17:09:05 +01:00
|
|
|
logger.debug("Expected Required Fields : {}", expectedRequiredFields);
|
|
|
|
|
2016-01-28 18:29:11 +01:00
|
|
|
Set<String> gotRequiredFields = usageRecord.getRequiredFields();
|
2015-12-18 17:09:05 +01:00
|
|
|
logger.debug("Got Required Fields : {}", gotRequiredFields);
|
|
|
|
|
2015-07-27 12:45:23 +02:00
|
|
|
Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields));
|
|
|
|
Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields));
|
|
|
|
|
2016-01-28 18:29:11 +01:00
|
|
|
usageRecord.validate();
|
|
|
|
logger.debug("{}", usageRecord);
|
2015-07-27 12:45:23 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testRequiredFieldsWithScopeSetExplicitely() throws InvalidValueException{
|
2016-02-09 18:46:23 +01:00
|
|
|
SecurityTokenProvider.instance.reset();
|
2016-01-28 18:29:11 +01:00
|
|
|
StorageUsageRecord usageRecord = TestUsageRecord.createTestStorageUsageRecordExplicitScope();
|
|
|
|
Assert.assertTrue(usageRecord.getScope()==null);
|
|
|
|
usageRecord.setScope(TestUsageRecord.TEST_SCOPE);
|
2015-07-27 12:45:23 +02:00
|
|
|
|
2015-10-20 14:40:30 +02:00
|
|
|
Set<String> expectedRequiredFields = getExpectedRequiredFields();
|
2016-01-28 18:29:11 +01:00
|
|
|
Set<String> gotRequiredFields = usageRecord.getRequiredFields();
|
2015-07-27 12:45:23 +02:00
|
|
|
|
|
|
|
Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields));
|
|
|
|
Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields));
|
|
|
|
|
2016-01-28 18:29:11 +01:00
|
|
|
usageRecord.validate();
|
|
|
|
logger.debug("{}", usageRecord);
|
2015-10-09 14:52:42 +02:00
|
|
|
}
|
2016-01-28 18:29:11 +01:00
|
|
|
|
2015-07-27 12:45:23 +02:00
|
|
|
}
|