mirror of
https://github.com/holos-run/holos.git
synced 2026-04-07 01:54:58 +00:00
012de360ac
Makes it easier to test and know for certain where the value is being set from. Previously it wasn't clear how the write to directory was being modified or read in the global config object.
133 lines
3.2 KiB
Go
133 lines
3.2 KiB
Go
package component_test
|
|
|
|
// "github.com/stretchr/testify/require"
|
|
import (
|
|
"context"
|
|
"embed"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/holos-run/holos/internal/component"
|
|
"github.com/holos-run/holos/internal/generate"
|
|
"github.com/holos-run/holos/internal/holos"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
//go:embed all:platform
|
|
var f embed.FS
|
|
|
|
// must align with embed all:platform directory
|
|
const platform string = "platform"
|
|
|
|
func TestComponentAlpha5(t *testing.T) {
|
|
h := newHarness(t, "components/v1alpha5")
|
|
t.Run("WriteToDefault", func(t *testing.T) {
|
|
err := h.c.Render(h.ctx, holos.WriteToDefault, os.Stderr, 1, nil)
|
|
assert.NoError(t, err)
|
|
|
|
// Verify the file was written to the expected path
|
|
expectedPath := filepath.Join(h.c.Root, holos.WriteToDefault, "v1alpha5/example/example.gen.yaml")
|
|
_, err = os.Stat(expectedPath)
|
|
assert.NoError(t, err, "Expected manifest file to exist at %s", expectedPath)
|
|
})
|
|
|
|
t.Run("WriteToCustom", func(t *testing.T) {
|
|
err := h.c.Render(h.ctx, "release", os.Stderr, 1, nil)
|
|
assert.NoError(t, err)
|
|
|
|
// Verify the file was written to the expected path
|
|
expectedPath := filepath.Join(h.c.Root, "release", "v1alpha5/example/example.gen.yaml")
|
|
_, err = os.Stat(expectedPath)
|
|
assert.NoError(t, err, "Expected manifest file to exist at %s", expectedPath)
|
|
})
|
|
}
|
|
|
|
func TestComponentAlpha6(t *testing.T) {
|
|
h := newHarness(t, "components/v1alpha6")
|
|
err := h.c.Render(h.ctx, holos.WriteToDefault, os.Stderr, 1, nil)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
type harness struct {
|
|
c *component.Component
|
|
ctx context.Context
|
|
}
|
|
|
|
func newHarness(t testing.TB, leaf string) *harness {
|
|
t.Helper()
|
|
|
|
// temp directory for the test, contains the platform and components for the
|
|
// test cases.
|
|
tempDir := t.TempDir()
|
|
root := filepath.Join(tempDir, platform)
|
|
if err := os.MkdirAll(root, 0o777); err != nil {
|
|
t.Fatalf("could not make platform directory: %v", err)
|
|
}
|
|
|
|
// test cancellation
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
t.Cleanup(cancel)
|
|
|
|
// Initialize the platform
|
|
if err := generate.GeneratePlatform(ctx, root, "v1alpha6"); err != nil {
|
|
t.Fatalf("could not generate platform: %v", err)
|
|
}
|
|
|
|
// Copy the components for the test cases
|
|
if err := fs.WalkDir(f, platform, walkDirFunc(ctx, tempDir)); err != nil {
|
|
t.Fatalf("could not prepare test directory: %v", err)
|
|
}
|
|
|
|
return &harness{
|
|
c: component.New(root, leaf),
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
func walkDirFunc(ctx context.Context, tempDir string) fs.WalkDirFunc {
|
|
return func(path string, d fs.DirEntry, err error) error {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
default:
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if path == "." {
|
|
return nil
|
|
}
|
|
fullPath := filepath.Join(tempDir, path)
|
|
|
|
switch {
|
|
case d.IsDir():
|
|
if err := os.MkdirAll(fullPath, 0o777); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
if err := os.MkdirAll(filepath.Dir(fullPath), 0o777); err != nil {
|
|
return err
|
|
}
|
|
srcFile, err := f.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer srcFile.Close()
|
|
|
|
dstFile, err := os.Create(fullPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer dstFile.Close()
|
|
|
|
if _, err := io.Copy(dstFile, srcFile); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
}
|