Function TestCacheVolumeLockKeyUsesSourceID

core/cache_test.go:464–499

(t *testing.T)

}

func TestCacheVolumeLockKeyUsesSourceID(t *testing.T) {
	t.Parallel()

	ctx := context.Background()
	query := &Query{Server: &cacheVolumeTestQueryServer{mockServer: &mockServer{}}}
	srv := newCoreDagqlServerForTest(t, query)
	srv.InstallObject(dagql.NewClass(srv, dagql.ClassOpts[*Directory]{}))
	dagCache, err := dagql.NewCache(ctx, "", nil, nil)
	require.NoError(t, err)
	t.Cleanup(func() {
		require.NoError(t, dagCache.Close(context.Background()))
	})
	ctx = ContextWithQuery(dagql.ContextWithCache(ctx, dagCache), query)

	sourceCall := &dagql.ResultCall{
		Kind:        dagql.ResultCallKindSynthetic,
		SyntheticOp: "cache-volume-source",
		Type:        dagql.NewResultCallType((&Directory{}).Type()),
	}
	initialSource := cacheVolumeTestDirectoryResult(t, srv, "cache-volume-source")
	sourceAny, err := dagCache.GetOrInitCall(ctx, "session", srv, &dagql.CallRequest{
		ResultCall: sourceCall,
	}, dagql.ValueFunc(initialSource))
	require.NoError(t, err)
	source, ok := sourceAny.(dagql.ObjectResult[*Directory])
	require.True(t, ok)
	sourceID, err := source.ID()
	require.NoError(t, err)
	encodedSourceID, err := sourceID.Encode()
	require.NoError(t, err)

	cache := NewCache("cache-key", "ns", dagql.NonNull(source), CacheSharingModeLocked, "")
	lockKey, err := cache.lockKey()
	require.NoError(t, err)
	require.Contains(t, lockKey, encodedSourceID)
}

func TestCacheVolumeInitializeSnapshotCreatesMutableSnapshot(t *testing.T) {
	t.Parallel()