diff --git a/HIRS_AttestationCA/build.gradle b/HIRS_AttestationCA/build.gradle index 893997d9..6c927d93 100644 --- a/HIRS_AttestationCA/build.gradle +++ b/HIRS_AttestationCA/build.gradle @@ -29,6 +29,7 @@ dependencies { implementation 'org.springframework.boot:spring-boot-starter-data-jpa:3.0.1' implementation 'com.github.darrachequesne:spring-data-jpa-datatables:6.0.1' implementation 'org.springframework.retry:spring-retry:2.0.0' + implementation 'org.springframework:spring-test:6.0.11' implementation libs.springdatajpa implementation libs.bouncycastle diff --git a/HIRS_AttestationCA/src/test/java/hirs/attestationca/persist/SpringPersistenceTest.java b/HIRS_AttestationCA/src/test/java/hirs/attestationca/persist/SpringPersistenceTest.java new file mode 100644 index 00000000..c6ad3f33 --- /dev/null +++ b/HIRS_AttestationCA/src/test/java/hirs/attestationca/persist/SpringPersistenceTest.java @@ -0,0 +1,24 @@ +package hirs.attestationca.persist; + +import hirs.attestationca.persist.PersistenceConfiguration; +import org.hibernate.SessionFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.test.annotation.DirtiesContext; +import org.springframework.test.context.ContextConfiguration; +import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests; + +/** + * Base class that autowires a session factory for use of + * any tests that need a database connection. + */ +@ContextConfiguration(classes = PersistenceConfiguration.class) +@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS) +public class SpringPersistenceTest extends AbstractJUnit4SpringContextTests { + + /** + * Autowired session factory. + */ + @SuppressWarnings("checkstyle:visibilitymodifier") + @Autowired + protected SessionFactory sessionFactory; +} \ No newline at end of file