diff options
Diffstat (limited to 'src/test/java/com/zaxxer/hikari/mocks/StubStatement.java')
-rw-r--r-- | src/test/java/com/zaxxer/hikari/mocks/StubStatement.java | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/test/java/com/zaxxer/hikari/mocks/StubStatement.java b/src/test/java/com/zaxxer/hikari/mocks/StubStatement.java index 73fec69..99e7b13 100644 --- a/src/test/java/com/zaxxer/hikari/mocks/StubStatement.java +++ b/src/test/java/com/zaxxer/hikari/mocks/StubStatement.java @@ -16,6 +16,8 @@ package com.zaxxer.hikari.mocks; +import static com.zaxxer.hikari.util.UtilityElf.quietlySleep; + import java.sql.Connection; import java.sql.ResultSet; import java.sql.SQLException; @@ -30,6 +32,8 @@ import java.sql.Statement; public class StubStatement implements Statement { public static volatile boolean oldDriver; + + private static volatile long simulatedQueryTime; private boolean closed; private Connection connection; @@ -37,6 +41,10 @@ public class StubStatement implements Statement this.connection = connection; } + public static void setSimulatedQueryTime(long time) { + simulatedQueryTime = time; + } + /** {@inheritDoc} */ @SuppressWarnings("unchecked") @Override @@ -168,6 +176,9 @@ public class StubStatement implements Statement public boolean execute(String sql) throws SQLException { checkClosed(); + if (simulatedQueryTime > 0) { + quietlySleep(simulatedQueryTime); + } return false; } |