Refactor test server management and update configuration files for integration tests

Author: hasithaa

.gitignore

@@ -50,5 +50,9 @@ velocity.log*
 *Ballerina.lock
 Dependencies.toml
 
+# Auto-generated test configuration
+ballerina/tests/Config.toml
+integration-tests/tests/Config.toml
+
 # Java Logs
 java_pid*.hprof

ballerina/build.gradle

@@ -89,6 +89,9 @@ test.dependsOn ":${packageName}-compiler-plugin:build"
 publishToMavenLocal.dependsOn build
 publish.dependsOn build
 
+// Use the shared test server from root project
+test.dependsOn ":startSharedTestServer"
+
 def stripBallerinaExtensionVersion(String extVersion) {
     if (extVersion.matches(project.ext.timestampedVersionRegex)) {
         def splitVersion = extVersion.split('-');

ballerina/tests/Config.toml

@@ -1,6 +1,6 @@
-# Test configuration for unit tests
-# Note: Unit tests use lazy gRPC connection and don't require a running server
-# For integration tests, use the workflow-integration-tests module
+# Auto-generated test configuration
+# This file is generated by Gradle before running tests
+# DO NOT EDIT - changes will be overwritten
 
 [ballerina.workflow.workflowConfig]
 provider = "TEMPORAL"

build.gradle

@@ -91,6 +91,237 @@ task build {
     dependsOn(':workflow-compiler-plugin-tests:test')
 }
 
+// ============================================================================
+// Shared Temporal Test Server Configuration
+// Single server instance shared across module and integration tests
+// ============================================================================
+
+import java.net.InetSocketAddress
+import java.net.Socket
+
+def testServerProcess = null
+def testServerWorkDir = file("${project(':workflow-native-test').buildDir}")
+def testServerStarted = false
+def testServerPort = 7233
+def usingExternalServer = false
+
+// Helper function to check if a server is running on a given port
+def isServerRunning = { String host, int port ->
+    try {
+        def socket = new Socket()
+        socket.connect(new InetSocketAddress(host, port), 1000)
+        socket.close()
+        return true
+    } catch (Exception e) {
+        return false
+    }
+}
+
+// Helper function to stop the test server
+def stopTestServerIfRunning = {
+    if (!testServerStarted || usingExternalServer) {
+        return
+    }
+    
+    try {
+        def pidFile = new File(testServerWorkDir, 'test-server.pid')
+        if (pidFile.exists()) {
+            def shadowJarFile = project(':workflow-native-test').tasks.shadowJar.archiveFile.get().asFile
+            
+            logger.lifecycle("Stopping shared embedded Temporal test server...")
+            
+            def command = [
+                'java', '-jar', 
+                shadowJarFile.absolutePath,
+                'stop', 
+                testServerWorkDir.absolutePath
+            ]
+            
+            def processBuilder = new ProcessBuilder(command)
+            processBuilder.inheritIO()
+            processBuilder.directory(testServerWorkDir)
+            def stopProcess = processBuilder.start()
+            stopProcess.waitFor()
+            
+            logger.lifecycle("Shared embedded Temporal test server stopped")
+        }
+    } catch (Exception e) {
+        logger.warn("Error stopping test server: ${e.message}")
+    }
+    testServerStarted = false
+}
+
+// Register a build listener to stop the server after build completes
+gradle.buildFinished { result ->
+    if (testServerStarted) {
+        logger.lifecycle("Build finished - stopping shared test server...")
+        stopTestServerIfRunning()
+    }
+}
+
+// Start shared test server - used by both module and integration tests
+task startSharedTestServer {
+    dependsOn ":workflow-native-test:shadowJar"
+    
+    doLast {
+        // First, check if an external Temporal server is running (for debugging)
+        if (isServerRunning("localhost", testServerPort)) {
+            def externalTarget = "localhost:${testServerPort}"
+            logger.lifecycle("=" * 60)
+            logger.lifecycle("External Temporal server detected at ${externalTarget}")
+            logger.lifecycle("Using external server for debugging (skipping embedded server)")
+            logger.lifecycle("=" * 60)
+            
+            usingExternalServer = true
+            testServerStarted = true
+            
+            // Write Config.toml files for both test suites
+            writeTestConfig("${project(':workflow-ballerina').projectDir}/tests/Config.toml", externalTarget, true)
+            writeTestConfig("${project(':workflow-integration-tests').projectDir}/tests/Config.toml", externalTarget, true)
+            
+            logger.lifecycle("Test Config.toml files updated with external server URL: ${externalTarget}")
+            return
+        }
+        
+        // No external server found - start embedded test server
+        logger.lifecycle("No external Temporal server found on port ${testServerPort}")
+        logger.lifecycle("Starting shared embedded Temporal test server on port ${testServerPort}...")
+        
+        def shadowJarFile = project(':workflow-native-test').tasks.shadowJar.archiveFile.get().asFile
+        
+        // Ensure work directory exists
+        testServerWorkDir.mkdirs()
+        
+        def command = [
+            'java', '-jar', 
+            shadowJarFile.absolutePath,
+            'start', 
+            testServerWorkDir.absolutePath,
+            testServerPort.toString()
+        ]
+        
+        def processBuilder = new ProcessBuilder(command)
+        processBuilder.redirectOutput(new File(testServerWorkDir, 'test-server.log'))
+        processBuilder.redirectErrorStream(true)
+        processBuilder.directory(testServerWorkDir)
+        testServerProcess = processBuilder.start()
+        
+        // Wait for the server to start and write the target file with content
+        def targetFile = new File(testServerWorkDir, 'test-server.target')
+        def maxWait = 30 // seconds
+        def waited = 0
+        def target = ""
+        while (waited < maxWait) {
+            Thread.sleep(1000)
+            waited++
+            if (targetFile.exists()) {
+                target = targetFile.text.trim()
+                if (target.length() > 0) {
+                    logger.lifecycle("Target file found with content after ${waited}s")
+                    break
+                }
+            }
+            logger.lifecycle("Waiting for test server to start... (${waited}s)")
+        }
+        
+        if (target.length() > 0) {
+            testServerStarted = true
+            logger.lifecycle("Shared embedded Temporal test server started at: ${target}")
+            
+            // Write Config.toml files for both test suites
+            writeTestConfig("${project(':workflow-ballerina').projectDir}/tests/Config.toml", target, false)
+            writeTestConfig("${project(':workflow-integration-tests').projectDir}/tests/Config.toml", target, false)
+            
+            logger.lifecycle("Test Config.toml files updated with server URL: ${target}")
+        } else {
+            // Check if process is still running
+            if (!testServerProcess.isAlive()) {
+                def exitCode = testServerProcess.exitValue()
+                logger.error("Test server process exited with code: ${exitCode}")
+                def logFile = new File(testServerWorkDir, 'test-server.log')
+                if (logFile.exists()) {
+                    logger.error("Test server log:\n${logFile.text}")
+                }
+            }
+            throw new GradleException("Test server failed to start within ${maxWait} seconds")
+        }
+    }
+}
+
+// Helper method to write test Config.toml
+def writeTestConfig(String path, String serverUrl, boolean isExternal) {
+    def configFile = new File(path)
+    configFile.parentFile.mkdirs()
+    def comment = isExternal ? 
+        "# Using EXTERNAL Temporal server for debugging\n# Start your local Temporal server with: temporal server start-dev" :
+        "# This file is generated by Gradle before running tests\n# DO NOT EDIT - changes will be overwritten"
+    
+    configFile.text = """# Auto-generated test configuration
+${comment}
+
+[ballerina.workflow.workflowConfig]
+provider = "TEMPORAL"
+url = "${serverUrl}"
+namespace = "default"
+
+[ballerina.workflow.workflowConfig.params]
+taskQueue = "BALLERINA_WORKFLOW_TASK_QUEUE"
+maxConcurrentWorkflows = 100
+maxConcurrentActivities = 100
+"""
+}
+
+// Stop shared test server
+task stopSharedTestServer {
+    doLast {
+        stopTestServerIfRunning()
+    }
+}
+
+// ============================================================================
+// Coverage Merging Task
+// Merges coverage reports from module tests and integration tests
+// ============================================================================
+
+task mergeCoverageReports {
+    description = 'Merges Ballerina coverage reports from module and integration tests'
+    group = 'verification'
+    
+    dependsOn ':workflow-ballerina:test'
+    dependsOn ':workflow-integration-tests:test'
+    
+    doLast {
+        def moduleReport = file("${project(':workflow-ballerina').projectDir}/target/report/workflow/coverage-report.xml")
+        def integrationReport = file("${project(':workflow-integration-tests').projectDir}/target/report/workflow_tests/coverage-report.xml")
+        def mergedReportDir = file("${project.buildDir}/coverage")
+        def mergedReport = file("${mergedReportDir}/merged-coverage-report.xml")
+        
+        mergedReportDir.mkdirs()
+        
+        logger.lifecycle("Merging coverage reports...")
+        logger.lifecycle("  Module tests: ${moduleReport.exists() ? 'found' : 'not found'}")
+        logger.lifecycle("  Integration tests: ${integrationReport.exists() ? 'found' : 'not found'}")
+        
+        if (moduleReport.exists() && integrationReport.exists()) {
+            // Simple merge: Copy module report as base, then we can use external tools for proper merge
+            // For now, copy module report as the primary and log integration report location
+            mergedReport.text = moduleReport.text
+            
+            // Copy integration report alongside for reference
+            def integrationCopy = file("${mergedReportDir}/integration-coverage-report.xml")
+            integrationCopy.text = integrationReport.text
+            
+            logger.lifecycle("Coverage reports copied to: ${mergedReportDir}")
+            logger.lifecycle("  - merged-coverage-report.xml (from module tests)")
+            logger.lifecycle("  - integration-coverage-report.xml (from integration tests)")
+            logger.lifecycle("")
+            logger.lifecycle("To view combined coverage in IDEs, configure both report locations.")
+        } else {
+            logger.warn("Could not merge coverage reports - one or more reports missing")
+        }
+    }
+}
+
 release {
     buildTasks = ['build']
     failOnSnapshotDependencies = true