).collect(Collectors.toSet()); registerQueriesWithRetry(serverTemplateId, serverInstanceId, queryDefinitions);
private void registerQueriesWithRetryException(Exception exception) throws Exception { QueryDefinition query = QueryDefinition.builder().name("test").expression("expression").source("jbpm").target("CUSTOM").build(); Set<QueryDefinition> definitions = new HashSet<>(); definitions.add(query); KieServicesClient kieClientRecreated = Mockito.mock(KieServicesClient.class); QueryServicesClient queryClientRecreated = Mockito.mock(QueryServicesClient.class); when(kieClientRecreated.getServicesClient(any())).thenReturn(queryClientRecreated); when(kieServerIntegration.getAdminServerClientCheckEndpoints(anyString())).thenReturn(kieClientRecreated); doThrow(exception).when(queryClient).replaceQuery(any()); kieServerDataSetManager.registerQueriesWithRetry("template", "instance", definitions); verify(kieServerIntegration, times(1)).getAdminServerClient(anyString(), anyString()); verify(kieServerIntegration, times(1)).getAdminServerClientCheckEndpoints(anyString()); verify(queryClient, times(1)).replaceQuery(any()); verify(queryClientRecreated, times(1)).replaceQuery(any()); assertEquals(1, receivedEvents.size()); }
private void registerQueriesWithRetryException(Exception exception) throws Exception { QueryDefinition query = QueryDefinition.builder().name("test").expression("expression").source("jbpm").target("CUSTOM").build(); Set<QueryDefinition> definitions = new HashSet<>(); definitions.add(query); KieServicesClient kieClientRecreated = Mockito.mock(KieServicesClient.class); QueryServicesClient queryClientRecreated = Mockito.mock(QueryServicesClient.class); when(kieClientRecreated.getServicesClient(any())).thenReturn(queryClientRecreated); when(kieServerIntegration.getAdminServerClientCheckEndpoints(anyString())).thenReturn(kieClientRecreated); doThrow(exception).when(queryClient).replaceQuery(any()); kieServerDataSetManager.registerQueriesWithRetry("template", "instance", definitions); verify(kieServerIntegration, times(1)).getAdminServerClient(anyString(), anyString()); verify(kieServerIntegration, times(1)).getAdminServerClientCheckEndpoints(anyString()); verify(queryClient, times(1)).replaceQuery(any()); verify(queryClientRecreated, times(1)).replaceQuery(any()); assertEquals(1, receivedEvents.size()); }
@Test public void testRegisterQueriesWithoutRetry() throws Exception { QueryDefinition query = QueryDefinition.builder().name("test").expression("expression").source("jbpm").target("CUSTOM").build(); Set<QueryDefinition> definitions = new HashSet<>(); definitions.add(query); kieServerDataSetManager.registerQueriesWithRetry("template", "instance", definitions); verify(kieServerIntegration, times(1)).getAdminServerClient(anyString(), anyString()); verify(queryClient, times(1)).replaceQuery(any()); assertEquals(1, receivedEvents.size()); }
@Test public void testRegisterQueriesWithoutRetry() throws Exception { QueryDefinition query = QueryDefinition.builder().name("test").expression("expression").source("jbpm").target("CUSTOM").build(); Set<QueryDefinition> definitions = new HashSet<>(); definitions.add(query); kieServerDataSetManager.registerQueriesWithRetry("template", "instance", definitions); verify(kieServerIntegration, times(1)).getAdminServerClient(anyString(), anyString()); verify(queryClient, times(1)).replaceQuery(any()); assertEquals(1, receivedEvents.size()); }
@Test public void testRegisterQueriesOnServerWithoutBPMCapability() throws Exception { when(kieClient.getServicesClient(any())).thenThrow(KieServicesException.class); QueryDefinition query = QueryDefinition.builder().name("test").expression("expression").source("jbpm").target("CUSTOM").build(); Set<QueryDefinition> definitions = new HashSet<>(); definitions.add(query); kieServerDataSetManager.registerQueriesWithRetry("template", "instance", definitions); verify(kieServerIntegration, times(1)).getAdminServerClient(anyString(), anyString()); verify(queryClient, never()).replaceQuery(any()); assertEquals(0, receivedEvents.size()); } }
@Test public void testRegisterQueriesOnServerWithoutBPMCapability() throws Exception { when(kieClient.getServicesClient(any())).thenThrow(KieServicesException.class); QueryDefinition query = QueryDefinition.builder().name("test").expression("expression").source("jbpm").target("CUSTOM").build(); Set<QueryDefinition> definitions = new HashSet<>(); definitions.add(query); kieServerDataSetManager.registerQueriesWithRetry("template", "instance", definitions); verify(kieServerIntegration, times(1)).getAdminServerClient(anyString(), anyString()); verify(queryClient, never()).replaceQuery(any()); assertEquals(0, receivedEvents.size()); } }