2015-07-02 11:00:10 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
package org.gcube.accounting.datamodel.usagerecords;
|
|
|
|
|
|
|
|
import java.util.HashSet;
|
|
|
|
import java.util.Set;
|
|
|
|
|
|
|
|
import org.gcube.accounting.datamodel.BasicUsageRecord;
|
|
|
|
import org.gcube.accounting.datamodel.BasicUsageRecordUtility;
|
2015-07-02 12:01:59 +02:00
|
|
|
import org.gcube.accounting.datamodel.basetypes.TestUsageRecord;
|
2015-07-20 17:02:43 +02:00
|
|
|
import org.gcube.accounting.exception.InvalidValueException;
|
|
|
|
import org.gcube.common.scope.api.ScopeProvider;
|
2015-07-02 11:00:10 +02:00
|
|
|
import org.junit.Assert;
|
|
|
|
import org.junit.Test;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @author Luca Frosini (ISTI - CNR) http://www.lucafrosini.com/
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
public class ServiceUsageRecordTest {
|
|
|
|
|
|
|
|
public static Set<String> getExpectedRequiredFields(){
|
|
|
|
Set<String> expectedRequiredFields = new HashSet<String>();
|
|
|
|
expectedRequiredFields.add(BasicUsageRecordUtility.ID);
|
|
|
|
expectedRequiredFields.add(BasicUsageRecord.CONSUMER_ID);
|
|
|
|
expectedRequiredFields.add(BasicUsageRecord.CREATION_TIME);
|
|
|
|
expectedRequiredFields.add(BasicUsageRecordUtility.USAGE_RECORD_TYPE);
|
|
|
|
expectedRequiredFields.add(BasicUsageRecord.SCOPE);
|
|
|
|
expectedRequiredFields.add(BasicUsageRecord.OPERATION_RESULT);
|
2015-07-29 14:41:28 +02:00
|
|
|
expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.CALLER_HOST);
|
|
|
|
expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.HOST);
|
|
|
|
expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.SERVICE_CLASS);
|
|
|
|
expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.SERVICE_NAME);
|
|
|
|
expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.CALLED_METHOD);
|
|
|
|
expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.DURATION);
|
2015-07-02 11:00:10 +02:00
|
|
|
return expectedRequiredFields;
|
|
|
|
}
|
|
|
|
|
2015-07-20 17:02:43 +02:00
|
|
|
@Test(expected=InvalidValueException.class)
|
|
|
|
public void scopeNotSetValidationError() throws InvalidValueException {
|
|
|
|
ScopeProvider.instance.reset();
|
|
|
|
ServiceUsageRecord serviceUsageRecord = TestUsageRecord.createTestServiceUsageRecordAutomaticScope();
|
|
|
|
serviceUsageRecord.validate();
|
|
|
|
}
|
|
|
|
|
2015-07-02 11:00:10 +02:00
|
|
|
@Test
|
2015-07-20 17:02:43 +02:00
|
|
|
public void testRequiredFields() throws InvalidValueException{
|
|
|
|
ScopeProvider.instance.set(TestUsageRecord.TEST_SCOPE);
|
|
|
|
ServiceUsageRecord serviceUsageRecord = TestUsageRecord.createTestServiceUsageRecordAutomaticScope();
|
2015-07-02 11:00:10 +02:00
|
|
|
Set<String> expectedRequiredFields = getExpectedRequiredFields();
|
|
|
|
Set<String> gotRequiredFields = serviceUsageRecord.getRequiredFields();
|
|
|
|
|
|
|
|
Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields));
|
|
|
|
Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields));
|
|
|
|
|
2015-07-20 17:02:43 +02:00
|
|
|
serviceUsageRecord.validate();
|
2015-07-02 11:00:10 +02:00
|
|
|
}
|
|
|
|
|
2015-07-20 17:02:43 +02:00
|
|
|
@Test
|
|
|
|
public void testRequiredFieldsWithScopeSetExplicitely() throws InvalidValueException{
|
|
|
|
ServiceUsageRecord serviceUsageRecord = TestUsageRecord.createTestServiceUsageRecordAutomaticScope();
|
|
|
|
serviceUsageRecord.setScope(TestUsageRecord.TEST_SCOPE);
|
|
|
|
Set<String> expectedRequiredFields = getExpectedRequiredFields();
|
|
|
|
|
|
|
|
Set<String> gotRequiredFields = serviceUsageRecord.getRequiredFields();
|
|
|
|
|
|
|
|
Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields));
|
|
|
|
Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields));
|
|
|
|
|
|
|
|
serviceUsageRecord.validate();
|
|
|
|
}
|
2015-07-02 11:00:10 +02:00
|
|
|
}
|