Function TestHintFileIntegration_EnableDisableToggle

hintfile_integration_test.go:109–185

TestHintFileIntegration_EnableDisableToggle tests enabling and disabling hint files

(t *testing.T)

// TestHintFileIntegration_EnableDisableToggle tests enabling and disabling hint files
func TestHintFileIntegration_EnableDisableToggle(t *testing.T) {
	runForMergeModes(t, func(t *testing.T, mode mergeTestMode) {
		bucket := "bucket"
		opts := DefaultOptions
		opts.SegmentSize = 64 * KB
		opts.Dir = filepath.Join(t.TempDir(), fmt.Sprintf("test-hintfile-toggle-%s", mode.name))
		opts.EnableMergeV2 = mode.enableMergeV2

		removeDir(opts.Dir)

		// Test with hint files enabled
		opts.EnableHintFile = true

		db, err := Open(opts)
		require.NoError(t, err)
		txCreateBucket(t, db, DataStructureBTree, bucket, nil)

		// Write enough data to generate at least 2 data files (for merge to work)
		n := 2000
		for i := 0; i < n; i++ {
			txPut(t, db, bucket, testutils.GetTestBytes(i), testutils.GetTestBytes(i), Persistent, nil, nil)
		}

		// Perform merge
		require.NoError(t, db.Merge())

		// Verify hint files are created for all merged files
		setsEnabled := collectMergeFileSets(t, opts.Dir)
		firstIDs := dataFileIDsForMode(mode, setsEnabled)
		require.Greater(t, len(firstIDs), 0)

		for _, fid := range firstIDs {
			hintPath := getHintPath(fid, opts.Dir)
			_, err = os.Stat(hintPath)
			require.NoError(t, err, "Hint file should exist when EnableHintFile is true for file %d", fid)
		}

		require.NoError(t, db.Close())

		// Test with hint files disabled
		opts.EnableHintFile = false

		db, err = Open(opts)
		require.NoError(t, err)

		// Write more data to generate additional files for the second merge
		for i := n; i < n+1500; i++ {
			txPut(t, db, bucket, testutils.GetTestBytes(i), testutils.GetTestBytes(i), Persistent, nil, nil)
		}

		// Perform merge (should not create hint files)
		require.NoError(t, db.Merge())

		setsDisabled := collectMergeFileSets(t, opts.Dir)
		secondIDs := unionDataFileIDs(setsDisabled)
		require.Greater(t, len(secondIDs), 0)

		// Check that hint files do not exist for any merged files