mirror of
https://github.com/samuelncui/yatm.git
synced 2026-11-08 04:18:03 +00:00
Samuel Cui
4fef653b71
Unify file queries and Scan execution, retain indexed Identical results, improve Job and Preview workflows, and verify release packaging.
96 lines
3.5 KiB
Go
96 lines
3.5 KiB
Go
package executor
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/samuelncui/acp"
|
|
"github.com/samuelncui/yatm/entity"
|
|
"github.com/samuelncui/yatm/internal/library"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func writeObservedFile(t *testing.T, name string, content []byte) (string, os.FileInfo) {
|
|
t.Helper()
|
|
filename := filepath.Join(t.TempDir(), name)
|
|
require.NoError(t, os.WriteFile(filename, content, 0o644))
|
|
info, err := os.Lstat(filename)
|
|
require.NoError(t, err)
|
|
return filename, info
|
|
}
|
|
|
|
func TestContentHashUsesTheObservedFileAndRefreshesItsCache(t *testing.T) {
|
|
ctx := context.Background()
|
|
content := []byte("observed content")
|
|
contentHash := sha256.Sum256(content)
|
|
filename, observed := writeObservedFile(t, "file.txt", content)
|
|
|
|
// An unknown file is read and hashed, and the result becomes reusable cache state.
|
|
facts, err := HashObservedContent(ctx, filename, observed)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int64(len(content)), facts.SizeBytes)
|
|
require.Equal(t, contentHash[:], facts.Sha256)
|
|
require.Equal(t, uint32(observed.Mode()), facts.Mode)
|
|
require.Equal(t, observed.ModTime().UnixNano(), facts.MtimeNs)
|
|
signature, err := library.NewFileSignature(facts.Sha256, facts.SizeBytes)
|
|
require.NoError(t, err)
|
|
require.Equal(t, signature, facts.Signature)
|
|
cached, valid, err := acp.ReadCachedSignature(filename)
|
|
require.NoError(t, err)
|
|
require.True(t, valid)
|
|
require.Equal(t, facts.Sha256, cached.SHA256[:])
|
|
|
|
// The cached value is reused only while a later read would still observe the same file.
|
|
require.NoError(t, os.Chtimes(filename, observed.ModTime().Add(time.Hour), observed.ModTime().Add(time.Hour)))
|
|
stale, err := os.Lstat(filename)
|
|
require.NoError(t, err)
|
|
reused, err := HashObservedContent(ctx, filename, stale)
|
|
require.NoError(t, err)
|
|
require.Equal(t, facts.Sha256, reused.Sha256)
|
|
}
|
|
|
|
func TestContentVerificationReadsContentInsteadOfTrustingTheCache(t *testing.T) {
|
|
ctx := context.Background()
|
|
filename, observed := writeObservedFile(t, "file.txt", []byte("saved"))
|
|
_, err := HashObservedContent(ctx, filename, observed)
|
|
require.NoError(t, err)
|
|
|
|
// Equal metadata with different bytes must still be observed as the bytes on disk.
|
|
require.NoError(t, os.WriteFile(filename, []byte("other"), 0o644))
|
|
require.NoError(t, os.Chtimes(filename, observed.ModTime(), observed.ModTime()))
|
|
require.NoError(t, os.Chmod(filename, observed.Mode()))
|
|
current, err := os.Lstat(filename)
|
|
require.NoError(t, err)
|
|
require.Equal(t, observed.Size(), current.Size())
|
|
|
|
otherHash := sha256.Sum256([]byte("other"))
|
|
verified, err := VerifyObservedContent(ctx, filename, current)
|
|
require.NoError(t, err)
|
|
require.Equal(t, otherHash[:], verified.Sha256)
|
|
}
|
|
|
|
func TestOriginalContentUsesConfiguredMappedRead(t *testing.T) {
|
|
content := []byte("mapped original content")
|
|
filename, observed := writeObservedFile(t, "file.txt", content)
|
|
source := &library.Location{Config: &entity.LocationConfig{UseMmap: true}}
|
|
facts, err := HashOriginalContent(context.Background(), filename, observed, source)
|
|
require.NoError(t, err)
|
|
want := sha256.Sum256(content)
|
|
require.Equal(t, want[:], facts.Sha256)
|
|
}
|
|
|
|
func TestContentObservationKeepsRetryableSourceErrors(t *testing.T) {
|
|
ctx := context.Background()
|
|
filename, observed := writeObservedFile(t, "file.txt", []byte("saved"))
|
|
|
|
// A source ACP cannot describe stays an error with its original identity, never a silent gap.
|
|
require.NoError(t, os.Remove(filename))
|
|
_, err := HashObservedContent(ctx, filename, observed)
|
|
require.ErrorIs(t, err, fs.ErrNotExist)
|
|
}
|