78 lines
2.7 KiB
Java
78 lines
2.7 KiB
Java
/**
|
|
*
|
|
*/
|
|
package org.gcube.documentstore.persistence;
|
|
|
|
import org.gcube.accounting.datamodel.UsageRecord;
|
|
import org.gcube.common.authorization.client.exceptions.ObjectNotFound;
|
|
import org.junit.Assert;
|
|
import org.junit.Test;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
/**
|
|
* @author Luca Frosini (ISTI - CNR)
|
|
*/
|
|
public class PersistencePostgreSQLTest extends ContextTest {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(PersistencePostgreSQLTest.class);
|
|
|
|
@Test
|
|
public void persistenceIsPostgreSQL() throws ObjectNotFound, Exception {
|
|
logger.debug("Going to check if the Persistence is CouchBase");
|
|
PersistenceBackendFactory.setFallbackLocation(null);
|
|
String context = ContextTest.getCurrentContextFullName();
|
|
FallbackPersistenceBackend fallbackPersistenceBackend = PersistenceBackendFactory.createFallback(context);
|
|
|
|
PersistenceBackend persistenceBackend = PersistenceBackendFactory.rediscoverPersistenceBackend(fallbackPersistenceBackend, context);
|
|
Assert.assertTrue(persistenceBackend instanceof PersistencePostgreSQL);
|
|
}
|
|
|
|
@Test
|
|
public void testInsertRecords() throws ObjectNotFound, Exception {
|
|
PersistenceBackendFactory.setFallbackLocation(null);
|
|
String context = ContextTest.getCurrentContextFullName();
|
|
FallbackPersistenceBackend fallbackPersistenceBackend = PersistenceBackendFactory.createFallback(context);
|
|
PersistenceBackend persistenceBackend = PersistenceBackendFactory.rediscoverPersistenceBackend(fallbackPersistenceBackend, context);
|
|
Assert.assertTrue(persistenceBackend instanceof PersistencePostgreSQL);
|
|
PersistencePostgreSQL persistencePostgreSQL = (PersistencePostgreSQL) persistenceBackend;
|
|
for(int i=0; i<1000000; i++) {
|
|
if(i%50000 == 0) {
|
|
persistencePostgreSQL.commitAndClose();
|
|
}
|
|
|
|
/*
|
|
AggregatedServiceUsageRecord usageRecord = getTestServiceUsageRecord();
|
|
/*
|
|
logger.debug("operationCount : {} - min : {} duration : {} - max : {}",
|
|
usageRecord.getOperationCount(),
|
|
usageRecord.getMinInvocationTime(),
|
|
usageRecord.getDuration(),
|
|
usageRecord.getMaxInvocationTime());
|
|
* /
|
|
persistencePostgreSQL.insert(usageRecord);
|
|
*/
|
|
|
|
/*
|
|
UsageRecord usageRecord = getTestAggregatedJobUsageRecord();
|
|
persistencePostgreSQL.insert(usageRecord);
|
|
|
|
usageRecord = getTestAggregatedPortletUsageRecord();
|
|
persistencePostgreSQL.insert(usageRecord);
|
|
|
|
usageRecord = getTestAggregatedServiceUsageRecord();
|
|
persistencePostgreSQL.insert(usageRecord);
|
|
|
|
usageRecord = getTestAggregatedStorageStatusRecord();
|
|
persistencePostgreSQL.insert(usageRecord);
|
|
*/
|
|
|
|
UsageRecord usageRecord = TestUsageRecord.getTestAggregatedStorageUsageRecord();
|
|
persistencePostgreSQL.insert(usageRecord);
|
|
}
|
|
persistencePostgreSQL.commitAndClose();
|
|
}
|
|
|
|
|
|
}
|