mdedetrich commented on code in PR #12728: URL: https://github.com/apache/kafka/pull/12728#discussion_r1383701619
########## connect/runtime/src/test/java/org/apache/kafka/connect/runtime/standalone/StandaloneHerderTest.java: ########## @@ -127,94 +134,86 @@ private enum SourceSink { noneConnectorClientConfigOverridePolicy = new SampleConnectorClientConfigOverridePolicy(); @Before - public void setup() { - worker = PowerMock.createMock(Worker.class); - String[] methodNames = new String[]{"connectorType", "buildRestartPlan", "recordRestarting"}; - herder = PowerMock.createPartialMock(StandaloneHerder.class, methodNames, - worker, WORKER_ID, KAFKA_CLUSTER_ID, statusBackingStore, new MemoryConfigBackingStore(transformer), noneConnectorClientConfigOverridePolicy, new MockTime()); + public void setup() throws ExecutionException, InterruptedException { + worker = mock(Worker.class); + herder = mock(StandaloneHerder.class, withSettings() + .useConstructor(worker, WORKER_ID, KAFKA_CLUSTER_ID, statusBackingStore, new MemoryConfigBackingStore(transformer), noneConnectorClientConfigOverridePolicy, new MockTime()) + .defaultAnswer(CALLS_REAL_METHODS)); createCallback = new FutureCallback<>(); - plugins = PowerMock.createMock(Plugins.class); - pluginLoader = PowerMock.createMock(PluginClassLoader.class); - loaderSwap = PowerMock.createMock(LoaderSwap.class); - PowerMock.mockStatic(WorkerConnector.class); - Capture<Map<String, String>> configCapture = Capture.newInstance(); - EasyMock.expect(transformer.transform(eq(CONNECTOR_NAME), EasyMock.capture(configCapture))).andAnswer(configCapture::getValue).anyTimes(); + plugins = mock(Plugins.class); + pluginLoader = mock(PluginClassLoader.class); + loaderSwap = mock(LoaderSwap.class); + final ArgumentCaptor<Map<String, String>> configCapture = ArgumentCaptor.forClass(Map.class); + when(transformer.transform(eq(CONNECTOR_NAME), configCapture.capture())).thenAnswer(invocation -> configCapture.getValue()); + } + + @After + public void tearDown() { + verifyNoMoreInteractions(worker, statusBackingStore); } @Test public void testCreateSourceConnector() throws Exception { - connector = PowerMock.createMock(BogusSourceConnector.class); + connector = mock(BogusSourceConnector.class); expectAdd(SourceSink.SOURCE); Map<String, String> config = connectorConfig(SourceSink.SOURCE); - Connector connectorMock = PowerMock.createMock(SourceConnector.class); + Connector connectorMock = mock(SourceConnector.class); expectConfigValidation(connectorMock, true, config); - PowerMock.replayAll(); - herder.putConnectorConfig(CONNECTOR_NAME, config, false, createCallback); Herder.Created<ConnectorInfo> connectorInfo = createCallback.get(1000L, TimeUnit.SECONDS); assertEquals(createdInfo(SourceSink.SOURCE), connectorInfo.result()); - - PowerMock.verifyAll(); } @Test public void testCreateConnectorFailedValidation() { // Basic validation should be performed and return an error, but should still evaluate the connector's config - connector = PowerMock.createMock(BogusSourceConnector.class); + connector = mock(BogusSourceConnector.class); Map<String, String> config = connectorConfig(SourceSink.SOURCE); config.remove(ConnectorConfig.NAME_CONFIG); - Connector connectorMock = PowerMock.createMock(SourceConnector.class); - EasyMock.expect(worker.configTransformer()).andReturn(transformer).times(2); - final Capture<Map<String, String>> configCapture = EasyMock.newCapture(); - EasyMock.expect(transformer.transform(EasyMock.capture(configCapture))).andAnswer(configCapture::getValue); - EasyMock.expect(worker.getPlugins()).andReturn(plugins).times(4); - EasyMock.expect(plugins.newConnector(EasyMock.anyString())).andReturn(connectorMock); - EasyMock.expect(plugins.connectorLoader(EasyMock.anyString())).andReturn(pluginLoader); - EasyMock.expect(plugins.withClassLoader(pluginLoader)).andReturn(loaderSwap); + Connector connectorMock = mock(SourceConnector.class); + when(worker.configTransformer()).thenReturn(transformer); + final ArgumentCaptor<Map<String, String>> configCapture = ArgumentCaptor.forClass(Map.class); + when(transformer.transform(configCapture.capture())).thenAnswer(invocation -> configCapture.getValue()); + when(worker.getPlugins()).thenReturn(plugins); + when(plugins.newConnector(anyString())).thenReturn(connectorMock); + when(plugins.connectorLoader(anyString())).thenReturn(pluginLoader); + when(plugins.withClassLoader(pluginLoader)).thenReturn(loaderSwap); - EasyMock.expect(connectorMock.config()).andStubReturn(new ConfigDef()); + when(connectorMock.config()).thenReturn(new ConfigDef()); ConfigValue validatedValue = new ConfigValue("foo.bar"); - EasyMock.expect(connectorMock.validate(config)).andReturn(new Config(singletonList(validatedValue))); + when(connectorMock.validate(config)).thenReturn(new Config(singletonList(validatedValue))); loaderSwap.close(); - EasyMock.expectLastCall(); - - PowerMock.replayAll(); herder.putConnectorConfig(CONNECTOR_NAME, config, false, createCallback); ExecutionException exception = assertThrows(ExecutionException.class, () -> createCallback.get(1000L, TimeUnit.SECONDS)); if (BadRequestException.class != exception.getCause().getClass()) { throw new AssertionError(exception.getCause()); } - PowerMock.verifyAll(); } @Test public void testCreateConnectorAlreadyExists() throws Throwable { - connector = PowerMock.createMock(BogusSourceConnector.class); + connector = mock(BogusSourceConnector.class); // First addition should succeed expectAdd(SourceSink.SOURCE); Map<String, String> config = connectorConfig(SourceSink.SOURCE); - Connector connectorMock = PowerMock.createMock(SourceConnector.class); + Connector connectorMock = mock(SourceConnector.class); expectConfigValidation(connectorMock, true, config, config); - EasyMock.expect(worker.configTransformer()).andReturn(transformer).times(2); - final Capture<Map<String, String>> configCapture = EasyMock.newCapture(); - EasyMock.expect(transformer.transform(EasyMock.capture(configCapture))).andAnswer(configCapture::getValue); - EasyMock.expect(worker.getPlugins()).andReturn(plugins).times(3); - EasyMock.expect(plugins.connectorLoader(EasyMock.anyString())).andReturn(pluginLoader); - EasyMock.expect(plugins.withClassLoader(pluginLoader)).andReturn(loaderSwap); + when(worker.configTransformer()).thenReturn(transformer); + final ArgumentCaptor<Map<String, String>> configCapture = ArgumentCaptor.forClass(Map.class); + when(transformer.transform(configCapture.capture())).thenAnswer(invocation -> configCapture.getValue()); + when(worker.getPlugins()).thenReturn(plugins); + when(plugins.connectorLoader(anyString())).thenReturn(pluginLoader); + when(plugins.withClassLoader(pluginLoader)).thenReturn(loaderSwap); // No new connector is created Review Comment: So I just moved the comment to where I think it should be, if its good then resolve the conversation -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org