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;
import static java.lang.Thread.sleep;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyList;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.atLeast;
import static org.mockito.Mockito.doNothing;
......@@ -182,6 +183,16 @@ public class SubscriptionManagerImplTest {
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() {
Subscription hmac_subscription = new Subscription();
hmac_subscription.setName("hamc_test_subscription");
......
......@@ -9,6 +9,7 @@ import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.test.util.ReflectionTestUtils;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
......@@ -76,4 +77,14 @@ public class TopicSubscriptionsTest {
assertEquals(existingTopicSubscriptions.size(), 1);
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