/** * */ 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; import org.gcube.accounting.datamodel.basetypes.TestUsageRecord; import org.gcube.accounting.exception.InvalidValueException; import org.gcube.common.scope.api.ScopeProvider; import org.junit.Assert; import org.junit.Test; /** * @author Luca Frosini (ISTI - CNR) http://www.lucafrosini.com/ * */ public class JobUsageRecordTest { public static Set getExpectedRequiredFields(){ Set expectedRequiredFields = new HashSet(); 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); expectedRequiredFields.add(org.gcube.accounting.datamodel.basetypes.AbstractServiceUsageRecord.CALLER_HOST); return expectedRequiredFields; } @Test(expected=InvalidValueException.class) public void scopeNotSetValidationError() throws InvalidValueException { ScopeProvider.instance.reset(); ServiceUsageRecord serviceUsageRecord = TestUsageRecord.createTestServiceUsageRecordAutomaticScope(); serviceUsageRecord.validate(); } @Test public void testRequiredFields() throws InvalidValueException{ ScopeProvider.instance.set(TestUsageRecord.TEST_SCOPE); ServiceUsageRecord serviceUsageRecord = TestUsageRecord.createTestServiceUsageRecordAutomaticScope(); Set expectedRequiredFields = getExpectedRequiredFields(); Set gotRequiredFields = serviceUsageRecord.getRequiredFields(); Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields)); Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields)); serviceUsageRecord.validate(); } @Test public void testRequiredFieldsWithScopeSetExplicitely() throws InvalidValueException{ ServiceUsageRecord serviceUsageRecord = TestUsageRecord.createTestServiceUsageRecordAutomaticScope(); serviceUsageRecord.setScope(TestUsageRecord.TEST_SCOPE); Set expectedRequiredFields = getExpectedRequiredFields(); Set gotRequiredFields = serviceUsageRecord.getRequiredFields(); Assert.assertTrue(expectedRequiredFields.containsAll(gotRequiredFields)); Assert.assertTrue(gotRequiredFields.containsAll(expectedRequiredFields)); serviceUsageRecord.validate(); } }