add forgotten files

Author: rjernst

build-tools-internal/src/integTest/groovy/org/elasticsearch/gradle/internal/transport/UpdateTransportVersionsFuncTest.groovy

@@ -0,0 +1,98 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the "Elastic License
+ * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
+ * Public License v 1"; you may not use this file except in compliance with, at
+ * your election, the "Elastic License 2.0", the "GNU Affero General Public
+ * License v3.0 only", or the "Server Side Public License, v 1".
+ */
+
+package org.elasticsearch.gradle.internal.transport
+
+import org.gradle.testkit.runner.BuildResult
+import org.gradle.testkit.runner.TaskOutcome
+
+class UpdateTransportVersionsFuncTest extends AbstractTransportVersionFuncTest {
+    def runUpdateTask(String... additionalArgs) {
+        List<String> args = new ArrayList<>()
+        args.add(":myserver:updateTransportVersions")
+        args.addAll(additionalArgs);
+        return gradleRunner(args.toArray())
+    }
+
+    void assertUpdateSuccess(BuildResult result) {
+        assert result.task(":myserver:updateTransportVersions").outcome == TaskOutcome.SUCCESS
+    }
+
+    void assertUpdateFailure(BuildResult result, String expectedOutput) {
+        assert result.task(":myserver:updateTransportVersions").outcome == TaskOutcome.FAILED
+        assertOutputContains(result.output, expectedOutput)
+    }
+
+    void assertTransportVersionsCsv(String expectedContent) {
+        File csvFile = file("myserver/src/main/resources/org/elasticsearch/TransportVersions.csv")
+        assert csvFile.exists()
+        // Normalize both sides: strip leading/trailing whitespace from each line and the whole string
+        def actualLines = csvFile.text.readLines().collect { it.strip() }.findAll { it != "" }
+        def expectedLines = expectedContent.readLines().collect { it.strip() }.findAll { it != "" }
+        assert actualLines == expectedLines
+    }
+
+    def "updates csv with minor version"() {
+        when:
+        def result = runUpdateTask("--stack-version", "9.2.0").build()
+
+        then:
+        assertUpdateSuccess(result)
+        assertTransportVersionsCsv("""
+            9.0.0,8000000
+            9.1.0,8012001
+            9.2.0,8123000
+        """)
+    }
+
+    def "updates csv with patch version"() {
+        given:
+        execute("git checkout main")
+        unreferableTransportVersion("initial_9.1.1", "8012002")
+        transportVersionUpperBound("9.1", "initial_9.1.1", "8012002")
+        execute('git commit -a -m "update"')
+
+        when:
+        def result = runUpdateTask("--stack-version", "9.1.1").build()
+
+        then:
+        assertUpdateSuccess(result)
+        assertTransportVersionsCsv("""
+            9.0.0,8000000
+            9.1.0,8012001
+            9.1.1,8012002
+        """)
+    }
+
+    def "fails when upper bound does not exist"() {
+        when:
+        def result = runUpdateTask("--stack-version", "9.3.0").buildAndFail()
+
+        then:
+        assertUpdateFailure(result, "Missing upper bound 9.3 for stack version 9.3.0")
+    }
+
+    def "is idempotent when version already exists"() {
+        when:
+        // Run the task twice with the same version
+        def result1 = runUpdateTask("--stack-version", "9.2.0").build()
+        def result2 = runUpdateTask("--stack-version", "9.2.0").build()
+
+        then:
+        assertUpdateSuccess(result1)
+        assertUpdateSuccess(result2)
+        assertOutputContains(result2.output, "Version 9.2.0 already exists in TransportVersions.csv, skipping")
+        // Should only have one entry for 9.2.0
+        assertTransportVersionsCsv("""
+            9.0.0,8000000
+            9.1.0,8012001
+            9.2.0,8123000
+        """)
+    }
+}

build-tools-internal/src/main/java/org/elasticsearch/gradle/internal/transport/UpdateTransportVersionsTask.java

@@ -0,0 +1,76 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the "Elastic License
+ * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
+ * Public License v 1"; you may not use this file except in compliance with, at
+ * your election, the "Elastic License 2.0", the "GNU Affero General Public
+ * License v3.0 only", or the "Server Side Public License, v 1".
+ */
+
+package org.elasticsearch.gradle.internal.transport;
+
+import org.elasticsearch.gradle.Version;
+import org.gradle.api.DefaultTask;
+import org.gradle.api.file.RegularFileProperty;
+import org.gradle.api.provider.Property;
+import org.gradle.api.services.ServiceReference;
+import org.gradle.api.tasks.Input;
+import org.gradle.api.tasks.InputFile;
+import org.gradle.api.tasks.TaskAction;
+import org.gradle.api.tasks.options.Option;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.StandardOpenOption;
+
+public abstract class UpdateTransportVersionsTask extends DefaultTask {
+
+    @ServiceReference("transportVersionResources")
+    abstract Property<TransportVersionResourcesService> getResourceService();
+
+    @InputFile
+    public abstract RegularFileProperty getTransportVersionsFile();
+
+    @Input
+    @Option(option = "stack-version", description = "The Elasticsearch version to record in TransportVersions.csv")
+    public abstract Property<String> getStackVersion();
+
+    @TaskAction
+    public void run() throws IOException {
+        Version stackVersion = Version.fromString(getStackVersion().get());
+
+        // Check if this version is already in the CSV file (idempotency check)
+        if (isVersionAlreadyRecorded(stackVersion)) {
+            getLogger().lifecycle("Version {} already exists in TransportVersions.csv, skipping", stackVersion);
+            return;
+        }
+
+        String upperBoundName = getUpperBoundName(stackVersion);
+        TransportVersionResourcesService resources = getResourceService().get();
+        TransportVersionUpperBound upperBound = resources.getUpperBoundFromGitBase(upperBoundName);
+
+        if (upperBound == null) {
+            throw new RuntimeException("Missing upper bound " + upperBoundName + " for stack version " + stackVersion);
+        }
+
+        int transportVersionId = upperBound.definitionId().complete();
+        addTransportVersionRecord(stackVersion, transportVersionId);
+    }
+
+    private boolean isVersionAlreadyRecorded(Version stackVersion) throws IOException {
+        var csvFile = getTransportVersionsFile().getAsFile().get().toPath();
+
+        String versionPrefix = stackVersion.toString() + ",";
+        return Files.readAllLines(csvFile).stream()
+            .anyMatch(line -> line.trim().startsWith(versionPrefix));
+    }
+
+    private void addTransportVersionRecord(Version stackVersion, int transportVersionId) throws IOException {
+        String newEntry = stackVersion + "," + transportVersionId + "\n";
+        Files.writeString(getTransportVersionsFile().getAsFile().get().toPath(), newEntry, StandardOpenOption.APPEND);
+    }
+
+    private String getUpperBoundName(Version version) {
+        return version.getMajor() + "." + version.getMinor();
+    }
+}