Implemented the functions of file source connector.

apache · Jan 19, 2024 · 3729fdb · 3729fdb
1 parent c8a45f6
commit 3729fdb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/eventmesh-connectors/eventmesh-connector-file/build.gradle b/eventmesh-connectors/eventmesh-connector-file/build.gradle
@@ -17,8 +17,8 @@
 
 dependencies {
     api project(":eventmesh-openconnect:eventmesh-openconnect-java")
-    implementation 'org.junit.jupiter:junit-jupiter:5.8.1'
-    implementation 'org.mockito:mockito-junit-jupiter:5.3.1'
+    testImplementation('org.junit.jupiter:junit-jupiter')
+    testImplementation('org.mockito:mockito-junit-jupiter')
     compileOnly 'org.projectlombok:lombok'
     annotationProcessor 'org.projectlombok:lombok'
 }
diff --git a/...ce/connector/FileSourceConnectorTest.java → ...nnector/file/FileSourceConnectorTest.java b/...ce/connector/FileSourceConnectorTest.java → ...nnector/file/FileSourceConnectorTest.java
@@ -15,13 +15,14 @@
  * limitations under the License.
  */
 
-package org.apache.eventmesh.connector.file.source.connector;
+package org.apache.eventmesh.connector.file;
 
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
 import org.apache.eventmesh.connector.file.source.config.FileSourceConfig;
 import org.apache.eventmesh.connector.file.source.config.SourceConnectorConfig;
+import org.apache.eventmesh.connector.file.source.connector.FileSourceConnector;
 import org.apache.eventmesh.openconnect.offsetmgmt.api.data.ConnectRecord;
 
 import java.nio.charset.StandardCharsets;
@@ -35,23 +36,22 @@
 import org.mockito.Mock;
 
 class FileSourceConnectorTest {
+
     private FileSourceConnector fileSourceConnector;
     @Mock
     private FileSourceConfig fileSourceConfig;
 
     @Test
     void testFileSourceConnector() throws Exception {
-        String directoryPath = "d/f/";
+        String directoryPath = "d/g/";
         Path directory = Paths.get(directoryPath);
         Files.createDirectories(directory);
         Path newFilePath = directory.resolve("foo.txt");
-        if (!Files.exists(newFilePath)) {
-            Files.createFile(newFilePath);
-        }
+        Files.createFile(newFilePath);
         fileSourceConfig = mock(FileSourceConfig.class);
         SourceConnectorConfig connectorConfig = mock(SourceConnectorConfig.class);
         when(fileSourceConfig.getConnectorConfig()).thenReturn(connectorConfig);
-        when(fileSourceConfig.getConnectorConfig().getFilePath()).thenReturn("d/f/foo.txt");
+        when(fileSourceConfig.getConnectorConfig().getFilePath()).thenReturn("d/g/foo.txt");
         String filePath = fileSourceConfig.getConnectorConfig().getFilePath();
         Path mockPath = Paths.get(filePath);
         String content = "line1\nline2\nline3";
@@ -62,6 +62,7 @@ void testFileSourceConnector() throws Exception {
         fileSourceConnector.start();
         List<ConnectRecord> connectRecords = fileSourceConnector.poll();
         fileSourceConnector.stop();
+        Files.delete(newFilePath);
         Assertions.assertEquals(content, connectRecords.get(0).getData().toString());
     }
 }