maintenance_test.go RawHistoryFolder

1package daemon
2
3import (
4	"context"
5	"errors"
6	"testing"
7
8	"github.com/bnema/vev/internal/domain"
9	portsmocks "github.com/bnema/vev/internal/ports/mocks"
10	"github.com/stretchr/testify/mock"
11	"github.com/stretchr/testify/require"
12)
13
14func TestStartupGarbageCollectionUsesValidatedCatalogue(t *testing.T) {
15	committed := domain.CheckpointRef{Generation: 3, ManifestDigest: [32]byte{3}}
16	records := []domain.CatalogueRecord{
17		{Name: "work", IncarnationID: domain.IncarnationID{1}, Committed: &committed},
18		{Name: "fresh", IncarnationID: domain.IncarnationID{2}},
19	}
20	catalogue := portsmocks.NewMockCatalogue(t)
21	catalogue.EXPECT().Records().Return(records, nil).Once()
22	repository := portsmocks.NewMockSnapshotRepository(t)
23	repository.EXPECT().CollectGarbage(mock.Anything, mock.MatchedBy(func(keep map[domain.IncarnationID]domain.CheckpointRef) bool {
24		return len(keep) == 2 && keep[records[0].IncarnationID] == committed && keep[records[1].IncarnationID] == (domain.CheckpointRef{})
25	})).Return(nil).Once()
26	d := newTestDaemon(t, portsmocks.NewMockPTYFactory(t), stubClock{})
27	WithDurableMaintenance(catalogue, repository)(d)
28
29	require.NoError(t, d.CollectStartupGarbage(context.Background()))
30}
31
32func TestStartupGarbageCollectionSkipsUnknownCatalogueState(t *testing.T) {
33	catalogue := portsmocks.NewMockCatalogue(t)
34	catalogue.EXPECT().Records().Return(nil, errors.New("catalogue read failed")).Once()
35	repository := portsmocks.NewMockSnapshotRepository(t)
36	d := newTestDaemon(t, portsmocks.NewMockPTYFactory(t), stubClock{})
37	WithDurableMaintenance(catalogue, repository)(d)
38
39	require.Error(t, d.CollectStartupGarbage(context.Background()))
40	repository.AssertNotCalled(t, "CollectGarbage", mock.Anything, mock.Anything)
41}