|
15 | 15 | */
|
16 | 16 | package org.quartz.impl.jdbcjobstore;
|
17 | 17 |
|
| 18 | +import static org.junit.Assert.assertThat; |
18 | 19 | import static org.mockito.Matchers.any;
|
19 | 20 | import static org.mockito.Matchers.anyString;
|
20 | 21 | import static org.mockito.Mockito.mock;
|
|
27 | 28 | import java.sql.PreparedStatement;
|
28 | 29 | import java.sql.ResultSet;
|
29 | 30 | import java.sql.SQLException;
|
| 31 | +import java.util.List; |
30 | 32 |
|
| 33 | +import org.hamcrest.collection.IsIterableWithSize; |
31 | 34 | import org.quartz.JobPersistenceException;
|
32 | 35 | import org.quartz.TriggerKey;
|
33 | 36 | import org.quartz.spi.OperableTrigger;
|
@@ -135,6 +138,26 @@ public void testSelectSimpleTriggerWithDeleteBeforeSelectExtendedProps() throws
|
135 | 138 | verify(persistenceDelegate).loadExtendedTriggerProperties(any(Connection.class), any(TriggerKey.class));
|
136 | 139 | }
|
137 | 140 |
|
| 141 | + public void testSelectTriggerToAcquireHonorsMaxCount() throws SQLException { |
| 142 | + |
| 143 | + StdJDBCDelegate jdbcDelegate = new StdJDBCDelegate(); |
| 144 | + |
| 145 | + Connection conn = mock(Connection.class); |
| 146 | + PreparedStatement preparedStatement = mock(PreparedStatement.class); |
| 147 | + ResultSet resultSet = mock(ResultSet.class); |
| 148 | + |
| 149 | + when(conn.prepareStatement(anyString())).thenReturn(preparedStatement); |
| 150 | + |
| 151 | + when(preparedStatement.executeQuery()).thenReturn(resultSet); |
| 152 | + |
| 153 | + when(resultSet.next()).thenReturn(true); |
| 154 | + when(resultSet.getString(anyString())).thenReturn("test"); |
| 155 | + |
| 156 | + List<TriggerKey> triggerKeys = jdbcDelegate.selectTriggerToAcquire(conn, Long.MAX_VALUE, Long.MIN_VALUE, 10); |
| 157 | + |
| 158 | + assertThat(triggerKeys, IsIterableWithSize.<TriggerKey>iterableWithSize(10)); |
| 159 | + } |
| 160 | + |
138 | 161 | static class TestStdJDBCDelegate extends StdJDBCDelegate {
|
139 | 162 |
|
140 | 163 | private final TriggerPersistenceDelegate testDelegate;
|
|
0 commit comments