mirror of
https://github.com/siderolabs/image-factory.git
synced 2025-12-10 03:51:12 +01:00
cf250cd103
No changes, just tons of renames. Fixes #11 Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
125 lines
3.1 KiB
Go
125 lines
3.1 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 (
|
|
"bytes"
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/siderolabs/image-service/pkg/flavor"
|
|
)
|
|
|
|
// well known flavor IDs, they will be created with the test run
|
|
const (
|
|
emptyFlavorID = "376567988ad370138ad8b2698212367b8edcb69b5fd68c80be1f2ec7d603b4ba"
|
|
extraArgsFlavorID = "e0fb1129bbbdfb5d002e94af4cdce712a8370e850950a33a242d4c3f178c532d"
|
|
)
|
|
|
|
func createFlavor(ctx context.Context, t *testing.T, baseURL string, marshalled []byte) *http.Response {
|
|
t.Helper()
|
|
|
|
req, err := http.NewRequestWithContext(ctx, "POST", baseURL+"/flavor", bytes.NewReader(marshalled))
|
|
require.NoError(t, err)
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
require.NoError(t, err)
|
|
|
|
t.Cleanup(func() {
|
|
resp.Body.Close()
|
|
})
|
|
|
|
return resp
|
|
}
|
|
|
|
func createFlavorGetID(ctx context.Context, t *testing.T, baseURL string, flavor flavor.Flavor) string {
|
|
t.Helper()
|
|
|
|
marshalled, err := flavor.Marshal()
|
|
require.NoError(t, err)
|
|
|
|
resp := createFlavor(ctx, t, baseURL, marshalled)
|
|
|
|
require.Equal(t, http.StatusCreated, resp.StatusCode)
|
|
|
|
var respBody struct {
|
|
ID string `json:"id"`
|
|
}
|
|
|
|
require.NoError(t, json.NewDecoder(resp.Body).Decode(&respBody))
|
|
|
|
return respBody.ID
|
|
}
|
|
|
|
func createFlavorInvalid(ctx context.Context, t *testing.T, baseURL string, marshalled []byte) string {
|
|
t.Helper()
|
|
|
|
resp := createFlavor(ctx, t, baseURL, marshalled)
|
|
|
|
require.Equal(t, http.StatusBadRequest, resp.StatusCode)
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err)
|
|
|
|
return string(body)
|
|
}
|
|
|
|
func testFlavor(ctx context.Context, t *testing.T, baseURL string) {
|
|
t.Run("empty", func(t *testing.T) {
|
|
assert.Equal(t, emptyFlavorID, createFlavorGetID(ctx, t, baseURL, flavor.Flavor{}))
|
|
})
|
|
|
|
t.Run("kernel args", func(t *testing.T) {
|
|
assert.Equal(t, extraArgsFlavorID, createFlavorGetID(ctx, t, baseURL,
|
|
flavor.Flavor{
|
|
Customization: flavor.Customization{
|
|
ExtraKernelArgs: []string{"nolapic", "nomodeset"},
|
|
},
|
|
},
|
|
))
|
|
})
|
|
|
|
t.Run("empty once again", func(t *testing.T) {
|
|
assert.Equal(t, emptyFlavorID, createFlavorGetID(ctx, t, baseURL, flavor.Flavor{}))
|
|
})
|
|
|
|
t.Run("invalid", func(t *testing.T) {
|
|
assert.Equal(t, "yaml: unmarshal errors:\n line 1: field something not found in type flavor.Flavor\n", createFlavorInvalid(ctx, t, baseURL, []byte(`something:`)))
|
|
})
|
|
|
|
t.Run("new flavor", func(t *testing.T) {
|
|
// create a new random flavor, as the flavor is persisted, and we want to test uploading new config
|
|
randomKernelArg := hex.EncodeToString(randomBytes(t, 32))
|
|
|
|
assert.Len(t, createFlavorGetID(ctx, t, baseURL,
|
|
flavor.Flavor{
|
|
Customization: flavor.Customization{
|
|
ExtraKernelArgs: []string{randomKernelArg},
|
|
},
|
|
},
|
|
), 64)
|
|
})
|
|
}
|
|
|
|
func randomBytes(t *testing.T, n int) []byte {
|
|
t.Helper()
|
|
|
|
b := make([]byte, n)
|
|
_, err := rand.Read(b)
|
|
require.NoError(t, err)
|
|
|
|
return b
|
|
}
|