7ce08bc788
os.Exit() doesn't execute defer-ed functions, OTOH we call panic() in case of setup errors, which does run defer-ed functions. So move the actual test setup and execution in a separate function which will execute all defer-ed functions and return the exit code from the test suite to TestMain.
53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
// Package client - integration_test contains functions to setup integration tests
|
|
// Copyright (C) 2020 by Red Hat, Inc.
|
|
|
|
// +build integration
|
|
|
|
package client
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/osbuild/osbuild-composer/internal/test"
|
|
)
|
|
|
|
// Hold test state to share between tests
|
|
var testState *TestState
|
|
|
|
// Setup the socket to use for running the tests
|
|
// Also makes sure there is a running server to test against
|
|
func executeTests(m *testing.M) int {
|
|
var err error
|
|
testState, err = setUpTestState("/run/weldr/api.socket", 60*time.Second, false)
|
|
if err != nil {
|
|
fmt.Printf("ERROR: Test setup failed: %s\n", err)
|
|
panic(err)
|
|
}
|
|
|
|
// Setup the test repo
|
|
dir, err := test.SetUpTemporaryRepository()
|
|
if err != nil {
|
|
fmt.Printf("ERROR: Test repo setup failed: %s\n", err)
|
|
panic(err)
|
|
}
|
|
|
|
// Cleanup after the tests
|
|
defer func() {
|
|
err := test.TearDownTemporaryRepository(dir)
|
|
if err != nil {
|
|
fmt.Printf("ERROR: Failed to clean up temporary repository: %s\n", err)
|
|
}
|
|
}()
|
|
|
|
testState.repoDir = dir
|
|
|
|
// Run the tests
|
|
return m.Run()
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
os.Exit(executeTests(m))
|
|
}
|