Function TestDB_Close_WithMergeInProgress

db_test.go:1213–1254

TestDB_Close_WithMergeInProgress tests closing database during merge operation

(t *testing.T)

// TestDB_Close_WithMergeInProgress tests closing database during merge operation
func TestDB_Close_WithMergeInProgress(t *testing.T) {
	opts := DefaultOptions
	opts.Dir = filepath.Join(t.TempDir(), "test-close-merge")
	opts.SegmentSize = 1024 // Small segment to trigger merge easily
	defer func() { _ = os.RemoveAll(opts.Dir) }()

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

	bucket := "test_bucket"
	err = db.Update(func(tx *Tx) error {
		return tx.NewBucket(DataStructureBTree, bucket)
	})
	require.NoError(t, err)

	// Add enough data to potentially trigger merge
	for i := 0; i < 100; i++ {
		key := testutils.GetTestBytes(i)
		val := testutils.GetRandomBytes(100)
		err = db.Update(func(tx *Tx) error {
			return tx.Put(bucket, key, val, Persistent)
		})
		require.NoError(t, err)
	}

	// Try to trigger merge (may or may not actually start)
	go func() {
		_ = db.Merge()
	}()

	// Give merge a moment to potentially start
	time.Sleep(100 * time.Millisecond)

	// Close should handle merge gracefully
	err = db.Close()
	require.NoError(t, err)

	// Verify database is closed
	require.True(t, db.IsClose())
	require.True(t, db.statusMgr.isClosed())
}

// TestDB_Close_ResourceCleanup tests that all resources are properly cleaned up
func TestDB_Close_ResourceCleanup(t *testing.T) {