Browse Source

Handle custom JMS acknowledgment modes as client acknowledge (#30619)

This commit updates JmsAccessor to handle custom JMS acknowledgment
modes as client acknowledge, which is useful when working with JMS
providers that provide non-standard variations of CLIENT_ACKNOWLEDGE,
such as AWS SQS and its UNORDERED_ACKNOWLEDGE mode.
pull/30651/head
Vedran Pavic 3 years ago committed by GitHub
parent
commit
3d61d9e0d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 7
      spring-jms/src/main/java/org/springframework/jms/support/JmsAccessor.java
  2. 22
      spring-jms/src/test/java/org/springframework/jms/support/JmsAccessorTests.java

7
spring-jms/src/main/java/org/springframework/jms/support/JmsAccessor.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2022 the original author or authors. * Copyright 2002-2023 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -223,7 +223,10 @@ public abstract class JmsAccessor implements InitializingBean {
* @see jakarta.jms.Session#CLIENT_ACKNOWLEDGE * @see jakarta.jms.Session#CLIENT_ACKNOWLEDGE
*/ */
protected boolean isClientAcknowledge(Session session) throws JMSException { protected boolean isClientAcknowledge(Session session) throws JMSException {
return (session.getAcknowledgeMode() == Session.CLIENT_ACKNOWLEDGE); int mode = session.getAcknowledgeMode();
return (mode != Session.SESSION_TRANSACTED &&
mode != Session.AUTO_ACKNOWLEDGE &&
mode != Session.DUPS_OK_ACKNOWLEDGE);
} }
} }

22
spring-jms/src/test/java/org/springframework/jms/support/JmsAccessorTests.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2019 the original author or authors. * Copyright 2002-2023 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -21,24 +21,27 @@ import org.junit.jupiter.api.Test;
import static org.assertj.core.api.Assertions.assertThat; import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException; import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
/** /**
* Unit tests for the {@link JmsAccessor} class. * Unit tests for the {@link JmsAccessor} class.
* *
* @author Rick Evans * @author Rick Evans
* @author Chris Beams * @author Chris Beams
* @author Vedran Pavic
*/ */
public class JmsAccessorTests { class JmsAccessorTests {
@Test @Test
public void testChokesIfConnectionFactoryIsNotSupplied() throws Exception { void testChokesIfConnectionFactoryIsNotSupplied() {
JmsAccessor accessor = new StubJmsAccessor(); JmsAccessor accessor = new StubJmsAccessor();
assertThatIllegalArgumentException().isThrownBy( assertThatIllegalArgumentException().isThrownBy(
accessor::afterPropertiesSet); accessor::afterPropertiesSet);
} }
@Test @Test
public void testSessionTransactedModeReallyDoesDefaultToFalse() throws Exception { void testSessionTransactedModeReallyDoesDefaultToFalse() {
JmsAccessor accessor = new StubJmsAccessor(); JmsAccessor accessor = new StubJmsAccessor();
assertThat(accessor.isSessionTransacted()).as("The [sessionTransacted] property of JmsAccessor must default to " + assertThat(accessor.isSessionTransacted()).as("The [sessionTransacted] property of JmsAccessor must default to " +
"false. Change this test (and the attendant Javadoc) if you have " + "false. Change this test (and the attendant Javadoc) if you have " +
@ -46,7 +49,7 @@ public class JmsAccessorTests {
} }
@Test @Test
public void testAcknowledgeModeReallyDoesDefaultToAutoAcknowledge() throws Exception { void testAcknowledgeModeReallyDoesDefaultToAutoAcknowledge() {
JmsAccessor accessor = new StubJmsAccessor(); JmsAccessor accessor = new StubJmsAccessor();
assertThat(accessor.getSessionAcknowledgeMode()).as("The [sessionAcknowledgeMode] property of JmsAccessor must default to " + assertThat(accessor.getSessionAcknowledgeMode()).as("The [sessionAcknowledgeMode] property of JmsAccessor must default to " +
"[Session.AUTO_ACKNOWLEDGE]. Change this test (and the attendant " + "[Session.AUTO_ACKNOWLEDGE]. Change this test (and the attendant " +
@ -54,11 +57,18 @@ public class JmsAccessorTests {
} }
@Test @Test
public void testSetAcknowledgeModeNameChokesIfBadAckModeIsSupplied() throws Exception { void testSetAcknowledgeModeNameChokesIfBadAckModeIsSupplied() {
assertThatIllegalArgumentException().isThrownBy(() -> assertThatIllegalArgumentException().isThrownBy(() ->
new StubJmsAccessor().setSessionAcknowledgeModeName("Tally ho chaps!")); new StubJmsAccessor().setSessionAcknowledgeModeName("Tally ho chaps!"));
} }
@Test
void testCustomAcknowledgeModeIsConsideredClientAcknowledge() throws Exception {
Session session = mock(Session.class);
given(session.getAcknowledgeMode()).willReturn(100);
JmsAccessor accessor = new StubJmsAccessor();
assertThat(accessor.isClientAcknowledge(session)).isTrue();
}
/** /**
* Crummy, stub, do-nothing subclass of the JmsAccessor class for use in testing. * Crummy, stub, do-nothing subclass of the JmsAccessor class for use in testing.

Loading…
Cancel
Save