public JdbcExpiringCodeStore(DataSource dataSource, TimeService timeService) { setDataSource(dataSource); setTimeService(timeService); }
@Before public void initExpiringCodeStoreTests() throws Exception { expiringCodeStore = (ExpiringCodeStore) expiringCodeStoreClass.newInstance(); if (expiringCodeStore instanceof InMemoryExpiringCodeStore) { ((InMemoryExpiringCodeStore) expiringCodeStore).setTimeService(timeService); } else { // confirm that everything is clean prior to test. TestUtils.deleteFrom(jdbcTemplate.getDataSource(), JdbcExpiringCodeStore.tableName); ((JdbcExpiringCodeStore) expiringCodeStore).setDataSource(jdbcTemplate.getDataSource()); ((JdbcExpiringCodeStore) expiringCodeStore).setTimeService(timeService); } }