Allow to define which ports should be exposed

Author: mrodm

internal/agentdeployer/_static/docker-agent-base.yml.tmpl

@@ -20,6 +20,12 @@ services:
     {{ end }}
     cap_drop:
       - ALL
+    {{ if .exposePorts }}
+    ports:
+    {{- range .exposePorts }}
+      - "{{ . }}"
+    {{- end }}
+    {{ end }}
     environment:
       - FLEET_ENROLL=1
       - FLEET_URL=https://fleet-server:8220

internal/agentdeployer/agent.go

@@ -256,6 +256,7 @@ func (d *DockerComposeAgentDeployer) installDockerfile(agentInfo AgentInfo) (str
 		"capabilities": agentInfo.Agent.LinuxCapabilities,
 		"runtime":      agentInfo.Agent.Runtime,
 		"pidMode":      agentInfo.Agent.PidMode,
+		"exposePorts":  agentInfo.Agent.ExposePorts,
 	})
 	if err != nil {
 		return "", fmt.Errorf("failed to create contents of the docker-compose file %q: %w", customAgentDockerfile, err)

internal/agentdeployer/info.go

@@ -82,6 +82,8 @@ type AgentInfo struct {
 		Runtime string
 		// LinuxCapabilities is a list of the capabilities needed to run the Elastic Agent process
 		LinuxCapabilities []string
+		// ExposePorts is a list of ports to make them available to communicate to the Elastic Agent process
+		ExposePorts []string
 	}
 
 	// CustomProperties store additional data used to boot up the service, e.g. AWS credentials.

internal/testrunner/runners/system/runner.go

@@ -338,6 +338,7 @@ func (r *runner) createAgentInfo(policy *kibana.Policy, config *testConfig, agen
 	info.Agent.LinuxCapabilities = config.Agent.LinuxCapabilities
 	info.Agent.Runtime = config.Agent.Runtime
 	info.Agent.PidMode = config.Agent.PidMode
+	info.Agent.ExposePorts = config.Agent.ExposePorts
 
 	// If user is defined in the configuration file, it has preference
 	// and it should not be overwritten by the value in the manifest

internal/testrunner/runners/system/test_config.go

@@ -58,6 +58,7 @@ type testConfig struct {
 		PidMode           string   `config:"pid_mode"`
 		LinuxCapabilities []string `config:"linux_capabilities"`
 		Runtime           string   `config:"runtime"`
+		ExposePorts       []string `config:"expose_ports"`
 	} `config:"agent"`
 }