#190 Add recovery e2e tests.

tests/recovery/setup_test.go

@@ -0,0 +1,63 @@
+package recovery
+
+import (
+	"context"
+	"github.com/openHPI/poseidon/internal/api"
+	"github.com/openHPI/poseidon/pkg/dto"
+	"github.com/openHPI/poseidon/tests"
+	"github.com/openHPI/poseidon/tests/e2e"
+	"github.com/openHPI/poseidon/tests/helpers"
+	"net/http"
+	"os"
+	"os/exec"
+	"time"
+)
+
+func (s *E2ERecoveryTestSuite) SetupTest() {
+	<-time.After(InactivityTimeout * time.Second)
+	// We do not want runner from the previous tests
+
+	var err error
+	s.runnerID, err = e2e.ProvideRunner(&dto.RunnerRequest{
+		ExecutionEnvironmentID: tests.DefaultEnvironmentIDAsInteger,
+		InactivityTimeout:      InactivityTimeout,
+	})
+	if err != nil {
+		log.WithError(err).Fatal("Could not provide runner")
+	}
+
+	<-time.After(tests.ShortTimeout)
+	s.poseidonCancel()
+	<-time.After(tests.ShortTimeout)
+	ctx, cancelPoseidon := context.WithCancel(context.Background())
+	s.poseidonCancel = cancelPoseidon
+	startPoseidon(ctx, cancelPoseidon)
+	waitForPoseidon()
+}
+
+func TearDown() {
+	path := helpers.BuildURL(api.BasePath, api.EnvironmentsPath, tests.DefaultEnvironmentIDAsString)
+	_, err := helpers.HTTPDelete(path, nil)
+	if err != nil {
+		log.WithError(err).Fatal("Could not remove default environment")
+	}
+}
+
+func startPoseidon(ctx context.Context, cancelPoseidon context.CancelFunc) {
+	poseidon := exec.CommandContext(ctx, *poseidonBinary) //nolint:gosec // We accept that another binary can be executed.
+	poseidon.Stdout = os.Stdout
+	poseidon.Stderr = os.Stderr
+	if err := poseidon.Start(); err != nil {
+		cancelPoseidon()
+		log.WithError(err).Fatal("Failed to start Poseidon")
+	}
+}
+
+func waitForPoseidon() {
+	done := false
+	for !done {
+		<-time.After(time.Second)
+		resp, err := http.Get(helpers.BuildURL(api.BasePath, api.HealthPath))
+		done = err == nil && resp.StatusCode == http.StatusNoContent
+	}
+}