Function TestHintFileIntegration_MixedDataStructures

hintfile_integration_test.go:276–384

TestHintFileIntegration_MixedDataStructures tests hint files with mixed data structures

(t *testing.T)

// TestHintFileIntegration_MixedDataStructures tests hint files with mixed data structures
func TestHintFileIntegration_MixedDataStructures(t *testing.T) {
	runForMergeModes(t, func(t *testing.T, mode mergeTestMode) {
		opts := DefaultOptions
		opts.SegmentSize = 64 * KB
		opts.Dir = filepath.Join(t.TempDir(), fmt.Sprintf("test-hintfile-mixed-ds-%s", mode.name))
		opts.EnableHintFile = true
		opts.EnableMergeV2 = mode.enableMergeV2

		removeDir(opts.Dir)

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

		// Create buckets for different data structures
		bucketBTree := "bucket_btree"
		bucketSet := "bucket_set"
		bucketList := "bucket_list"
		bucketZSet := "bucket_zset"

		txCreateBucket(t, db, DataStructureBTree, bucketBTree, nil)
		txCreateBucket(t, db, DataStructureSet, bucketSet, nil)
		txCreateBucket(t, db, DataStructureList, bucketList, nil)
		txCreateBucket(t, db, DataStructureSortedSet, bucketZSet, nil)

		// Add data to each structure - increase data size to ensure 2 files are created
		for i := 0; i < 2000; i++ {
			txPut(t, db, bucketBTree, testutils.GetTestBytes(i), testutils.GetTestBytes(i), Persistent, nil, nil)
		}

		setKey := testutils.GetTestBytes(0)
		for i := 0; i < 1000; i++ {
			txSAdd(t, db, bucketSet, setKey, testutils.GetTestBytes(i), nil, nil)
		}

		listKey := testutils.GetTestBytes(0)
		for i := 0; i < 500; i++ {
			txPush(t, db, bucketList, listKey, testutils.GetTestBytes(i), true, nil, nil)
		}

		zsetKey := testutils.GetTestBytes(0)
		for i := 0; i < 300; i++ {
			txZAdd(t, db, bucketZSet, zsetKey, testutils.GetTestBytes(i), float64(i), nil, nil)
		}

		// Delete some data from each structure
		for i := 0; i < 500; i++ {
			txDel(t, db, bucketBTree, testutils.GetTestBytes(i), nil)
		}
		for i := 0; i < 250; i++ {
			txSRem(t, db, bucketSet, setKey, testutils.GetTestBytes(i), nil)
		}
		for i := 0; i < 100; i++ {
			txPop(t, db, bucketList, listKey, testutils.GetTestBytes(i), nil, false)
		}
		for i := 0; i < 50; i++ {
			txZRem(t, db, bucketZSet, zsetKey, testutils.GetTestBytes(i), nil)
		}