From 94b16a169b950fec6ca7c426d05cd6156c0fdd18 Mon Sep 17 00:00:00 2001 From: Sam Brannen Date: Fri, 28 Aug 2009 09:23:02 +0000 Subject: [PATCH] Added assertPersonCount() functionality for increased robustness. git-svn-id: https://src.springframework.org/svn/spring-framework/trunk@1784 50f2f4bb-b051-0410-bef5-90022cba6387 --- .../junit4/orm/HibernateSessionFlushingTests.java | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/org.springframework.test/src/test/java/org/springframework/test/context/junit4/orm/HibernateSessionFlushingTests.java b/org.springframework.test/src/test/java/org/springframework/test/context/junit4/orm/HibernateSessionFlushingTests.java index 93cf7dd83a7..d12eb1d46a9 100644 --- a/org.springframework.test/src/test/java/org/springframework/test/context/junit4/orm/HibernateSessionFlushingTests.java +++ b/org.springframework.test/src/test/java/org/springframework/test/context/junit4/orm/HibernateSessionFlushingTests.java @@ -52,6 +52,14 @@ public class HibernateSessionFlushingTests extends AbstractTransactionalJUnit4Sp private SessionFactory sessionFactory; + protected int countRowsInPersonTable() { + return countRowsInTable("person"); + } + + protected void assertPersonCount(int expectedCount) { + assertEquals("Verifying number of rows in the 'person' table.", expectedCount, countRowsInPersonTable()); + } + @Before public void setUp() { assertInTransaction(true); @@ -72,7 +80,9 @@ public class HibernateSessionFlushingTests extends AbstractTransactionalJUnit4Sp public void saveJuergenWithDriversLicense() { DriversLicense driversLicense = new DriversLicense(2L, 2222L); Person juergen = new Person(JUERGEN, driversLicense); + int numRows = countRowsInPersonTable(); personService.save(juergen); + assertPersonCount(numRows + 1); assertNotNull("Should be able to save and retrieve Juergen", personService.findByName(JUERGEN)); assertNotNull("Juergen's ID should have been set", juergen.getId()); }