11package com .lemick .integration .spring ;
22
33
4- import com .lemick .api .AssertHibernateSQLStatementCount ;
4+ import com .lemick .api .AssertHibernateSQLCount ;
55import com .lemick .assertions .HibernateStatementAssertionResult ;
66import com .lemick .assertions .HibernateStatementAssertionResults ;
77import com .lemick .integration .hibernate .HibernateStatementCountInspector ;
@@ -30,7 +30,7 @@ public void beforeTestMethod(TestContext testContext) {
3030
3131 @ Override
3232 public void afterTestMethod (TestContext testContext ) {
33- AssertHibernateSQLStatementCount annotation = testContext .getTestMethod ().getAnnotation (AssertHibernateSQLStatementCount .class );
33+ AssertHibernateSQLCount annotation = testContext .getTestMethod ().getAnnotation (AssertHibernateSQLCount .class );
3434 if (annotation != null ) {
3535 flushExistingPersistenceContext (testContext , transactionAvailabilitySupplier );
3636 doStatementCountEvaluation (annotation );
@@ -56,7 +56,7 @@ private void flushExistingPersistenceContext(TestContext testContext, Supplier<B
5656 }
5757 }
5858
59- private void doStatementCountEvaluation (AssertHibernateSQLStatementCount annotation ) {
59+ private void doStatementCountEvaluation (AssertHibernateSQLCount annotation ) {
6060 HibernateStatementAssertionResults assertionResults = new HibernateStatementAssertionResults (List .of (
6161 new HibernateStatementAssertionResult (SELECT , statisticsSupplier .get ().getSelectStatements (), annotation .selects ()),
6262 new HibernateStatementAssertionResult (UPDATE , statisticsSupplier .get ().getUpdateStatements (), annotation .updates ()),
0 commit comments