|
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