attest/pkg/tuf/mock.go

package tuf

import (
	"io"
	"os"
	"path/filepath"

	"github.com/docker/attest/internal/util"
)

type MockTufClient struct {
	srcPath string
}

func NewMockTufClient(srcPath string) *MockTufClient {
	if srcPath == "" {
		panic("srcPath must be set")
	}
	return &MockTufClient{
		srcPath: srcPath,
	}
}

func (dc *MockTufClient) DownloadTarget(target string, _ string) (file *TargetFile, err error) {
	targetPath := filepath.Join(dc.srcPath, target)
	src, err := os.Open(targetPath)
	if err != nil {
		return nil, err
	}
	defer src.Close()

	b, err := io.ReadAll(src)
	if err != nil {
		return nil, err
	}

	return &TargetFile{TargetURI: targetPath, Data: b, Digest: util.SHA256Hex(b)}, nil
}

type MockVersionChecker struct {
	err error
}

func NewMockVersionChecker() *MockVersionChecker {
	return &MockVersionChecker{}
}

func (vc *MockVersionChecker) CheckVersion(_ Downloader) error {
	return vc.err
}