Skip to content
Snippets Groups Projects
Commit 277dcbf4 authored by Alok Joshi's avatar Alok Joshi
Browse files

unit tests

parent f3113184
No related branches found
No related tags found
1 merge request!364Close deleted subscriptions
Pipeline #175337 failed
...@@ -46,6 +46,7 @@ import java.util.concurrent.TimeUnit; ...@@ -46,6 +46,7 @@ import java.util.concurrent.TimeUnit;
import static java.lang.Thread.sleep; import static java.lang.Thread.sleep;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyList;
import static org.mockito.ArgumentMatchers.anyString; import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.atLeast; import static org.mockito.Mockito.atLeast;
import static org.mockito.Mockito.doNothing; import static org.mockito.Mockito.doNothing;
...@@ -182,6 +183,16 @@ public class SubscriptionManagerImplTest { ...@@ -182,6 +183,16 @@ public class SubscriptionManagerImplTest {
verify(subscriptionClient, times(1)).registerMessageHandler(any(), any(), any()); verify(subscriptionClient, times(1)).registerMessageHandler(any(), any(), any());
} }
@Test
public void shouldCloseSubscriptionClientWhenSubscriptionIsDeleted() throws ServiceBusException, InterruptedException {
when(topicSubscriptions.checkIfNewTopicSubscription(anyString(), anyString(), anyString())).thenReturn(false);
when(subscriptionClientFactory.getSubscriptionClient(dataPartition, sbTopic, sbSubscription)).thenReturn(subscriptionClient);
doNothing().when(topicSubscriptions).updateCurrentTopicSubscriptions(anyString(), anyString(), anyString());
when(topicSubscriptions.getDeletedTopicSubscriptions(anyString(), anyString(), anyList())).thenReturn(Collections.singletonList(getHmac_subscription().getNotificationId()));
subscriptionManager.subscribeNotificationsEvent();
verify(subscriptionClient, times(1)).close();
}
private static Subscription getHmac_subscription() { private static Subscription getHmac_subscription() {
Subscription hmac_subscription = new Subscription(); Subscription hmac_subscription = new Subscription();
hmac_subscription.setName("hamc_test_subscription"); hmac_subscription.setName("hamc_test_subscription");
......
...@@ -9,6 +9,7 @@ import org.mockito.junit.MockitoJUnitRunner; ...@@ -9,6 +9,7 @@ import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.test.util.ReflectionTestUtils; import org.springframework.test.util.ReflectionTestUtils;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap; import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
...@@ -76,4 +77,14 @@ public class TopicSubscriptionsTest { ...@@ -76,4 +77,14 @@ public class TopicSubscriptionsTest {
assertEquals(existingTopicSubscriptions.size(), 1); assertEquals(existingTopicSubscriptions.size(), 1);
assertEquals(currentTopicSubscriptions.size(), 0); assertEquals(currentTopicSubscriptions.size(), 0);
} }
@Test
public void deletedSubscriptionFoundTest() {
final String DELETED_SUBSCRIPTION = "test678asdeleted";
existingTopicSubscriptions.get(PARTITION).get(SBTOPICNAME).add(DELETED_SUBSCRIPTION);
List<String> deletedSubscriptions = topicSubscriptions.getDeletedTopicSubscriptions(PARTITION, SBTOPICNAME, Arrays.asList(EXISTINGSUBSCRIPTION));
assertEquals(1, deletedSubscriptions.size());
assertEquals(DELETED_SUBSCRIPTION, deletedSubscriptions.get(0));
}
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment