mirror of
https://github.com/siderolabs/image-factory.git
synced 2025-12-09 03:21:36 +01:00
cf250cd103
No changes, just tons of renames. Fixes #11 Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
//go:build integration
|
|
|
|
package integration_test
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"net"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/siderolabs/image-service/cmd/image-service/cmd"
|
|
)
|
|
|
|
func setupService(t *testing.T) (context.Context, string) {
|
|
t.Helper()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
logger := zaptest.NewLogger(t)
|
|
|
|
options := cmd.DefaultOptions
|
|
options.HTTPListenAddr = findListenAddr(t)
|
|
options.ImagePrefix = imagePrefixFlag
|
|
options.ExternalURL = "http://" + options.HTTPListenAddr + "/"
|
|
options.FlavorServiceRepository = flavorServiceRepositoryFlag
|
|
options.InstallerExternalRepository = installerExternalRepository
|
|
options.InstallerInternalRepository = installerInternalRepository
|
|
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
|
|
eg.Go(func() error {
|
|
return cmd.RunService(ctx, logger, options)
|
|
})
|
|
|
|
t.Cleanup(func() {
|
|
require.NoError(t, eg.Wait())
|
|
})
|
|
t.Cleanup(cancel)
|
|
t.Cleanup(http.DefaultClient.CloseIdleConnections)
|
|
|
|
// wait for the endpoint to be ready
|
|
require.Eventually(t, func() bool {
|
|
d, err := net.Dial("tcp", options.HTTPListenAddr)
|
|
if d != nil {
|
|
require.NoError(t, d.Close())
|
|
}
|
|
|
|
return err == nil
|
|
}, 10*time.Second, 10*time.Millisecond)
|
|
|
|
return ctx, options.HTTPListenAddr
|
|
}
|
|
|
|
func findListenAddr(t *testing.T) string {
|
|
t.Helper()
|
|
|
|
l, err := net.Listen("tcp", "127.0.0.1:0")
|
|
require.NoError(t, err)
|
|
|
|
addr := l.Addr().String()
|
|
|
|
require.NoError(t, l.Close())
|
|
|
|
return addr
|
|
}
|
|
|
|
func TestIntegration(t *testing.T) {
|
|
ctx, listenAddr := setupService(t)
|
|
baseURL := "http://" + listenAddr
|
|
|
|
t.Run("TestFlavor", func(t *testing.T) {
|
|
// flavor should be created first, thus no t.Parallel
|
|
testFlavor(ctx, t, baseURL)
|
|
})
|
|
|
|
t.Run("TestDownloadFrontend", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testDownloadFrontend(ctx, t, baseURL)
|
|
})
|
|
|
|
t.Run("TestPXEFrontend", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testPXEFrontend(ctx, t, baseURL)
|
|
})
|
|
|
|
t.Run("TestRegistryFrontend", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
testRegistryFrontend(ctx, t, listenAddr)
|
|
})
|
|
}
|
|
|
|
var (
|
|
imagePrefixFlag string
|
|
flavorServiceRepositoryFlag string
|
|
installerExternalRepository string
|
|
installerInternalRepository string
|
|
)
|
|
|
|
func init() {
|
|
flag.StringVar(&imagePrefixFlag, "test.image-prefix", cmd.DefaultOptions.ImagePrefix, "image prefix")
|
|
flag.StringVar(&flavorServiceRepositoryFlag, "test.flavor-service-repository", cmd.DefaultOptions.FlavorServiceRepository, "flavor service repository")
|
|
flag.StringVar(&installerExternalRepository, "test.installer-external-repository", cmd.DefaultOptions.InstallerExternalRepository, "image repository for the installer (external)")
|
|
flag.StringVar(&installerInternalRepository, "test.installer-internal-repository", cmd.DefaultOptions.InstallerInternalRepository, "image repository for the installer (internal)")
|
|
}
|