src/testing/host-target-testing/packages/repo_test.go

// Copyright 2023 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package packages

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"testing"
)

func TestCreateAndEditPackage(t *testing.T) {
	ctx := context.Background()

	parentDir := t.TempDir()
	pkgRepo, expandDir := createAndExpandPackage(t, parentDir)
	pkgBuilder, err := NewPackageBuilderFromDir(expandDir, "testpackage", "0", "testrepository.com")
	if err != nil {
		t.Fatalf("Failed to parse package from %s. %s", expandDir, err)
	}
	defer pkgBuilder.Close()

	fullPkgName := pkgBuilder.Name + "/" + pkgBuilder.Version
	newResource := "blah/z"

	srcPkg, err := pkgRepo.OpenPackage(ctx, fullPkgName)
	if err != nil {
		t.Fatalf("Repo does not contain '%s'. %s", fullPkgName, err)
	}
	if _, err := srcPkg.ReadFile(ctx, newResource); err == nil {
		t.Fatalf("%s should not be found in package", newResource)
	}

	pkg, err := pkgRepo.EditPackage(ctx, srcPkg, "testpackage_prime/0", func(tempDir string) error {
		contents := bytes.NewReader([]byte(newResource))
		data, err := io.ReadAll(contents)
		if err != nil {
			return fmt.Errorf("failed to read file: %w", err)
		}

		tempPath := filepath.Join(tempDir, newResource)
		if err := os.MkdirAll(filepath.Dir(tempPath), os.ModePerm); err != nil {
			return fmt.Errorf("failed to create parent directories for %q: %w", tempPath, err)
		}

		if err := os.WriteFile(tempPath, data, 0644); err != nil {
			return fmt.Errorf("failed to write data to %q: %w", tempDir, err)
		}

		return nil
	})

	if err != nil {
		t.Fatalf("Failed to edit the package '%s'. %s", fullPkgName, err)
	}
	if data, err := pkg.ReadFile(ctx, newResource); err != nil {
		t.Fatalf("%s should be in package.", newResource)
	} else {
		if string(data) != newResource {
			t.Fatalf("%s should have value %s but is %s", newResource, newResource, data)
		}
	}
}