From 2c0488da0b13f8430cb6847b5eb5569e54a1f7bb Mon Sep 17 00:00:00 2001
From: Kristoffer Dalby <kristoffer@tailscale.com>
Date: Mon, 24 Oct 2022 16:40:49 +0200
Subject: [PATCH] Add Execute helper for controlserver

Signed-off-by: Kristoffer Dalby <kristoffer@tailscale.com>
---
 integration/control.go   |  1 +
 integration/hsic/hsic.go | 23 +++++++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/integration/control.go b/integration/control.go
index 58a4661f..33a687c0 100644
--- a/integration/control.go
+++ b/integration/control.go
@@ -6,6 +6,7 @@ import (
 
 type ControlServer interface {
 	Shutdown() error
+	Execute(command []string) (string, error)
 	GetHealthEndpoint() string
 	GetEndpoint() string
 	WaitForReady() error
diff --git a/integration/hsic/hsic.go b/integration/hsic/hsic.go
index a6372a5a..7db8233a 100644
--- a/integration/hsic/hsic.go
+++ b/integration/hsic/hsic.go
@@ -103,6 +103,29 @@ func (t *HeadscaleInContainer) Shutdown() error {
 	return t.pool.Purge(t.container)
 }
 
+func (t *HeadscaleInContainer) Execute(
+	command []string,
+) (string, error) {
+	log.Println("command", command)
+	log.Printf("running command for %s\n", t.hostname)
+	stdout, stderr, err := dockertestutil.ExecuteCommand(
+		t.container,
+		command,
+		[]string{},
+	)
+	if err != nil {
+		log.Printf("command stderr: %s\n", stderr)
+
+		return "", err
+	}
+
+	if stdout != "" {
+		log.Printf("command stdout: %s\n", stdout)
+	}
+
+	return stdout, nil
+}
+
 func (t *HeadscaleInContainer) GetIP() string {
 	return t.container.GetIPInNetwork(t.network)
 }