Function TestDB_Close_ResourceCleanup

db_test.go:1257–1312

TestDB_Close_ResourceCleanup tests that all resources are properly cleaned up

(t *testing.T)

// TestDB_Close_ResourceCleanup tests that all resources are properly cleaned up
func TestDB_Close_ResourceCleanup(t *testing.T) {
	opts := DefaultOptions
	opts.Dir = filepath.Join(t.TempDir(), "test-close-cleanup")
	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 some data
	for i := 0; i < 10; 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)
	}

	// Close database
	err = db.Close()
	require.NoError(t, err)

	// Verify all resources are released
	require.Nil(t, db.ActiveFile, "ActiveFile should be nil")
	require.Nil(t, db.Index, "Index should be nil")
	require.Nil(t, db.fm, "FileManager should be nil")
	require.Nil(t, db.commitBuffer, "commitBuffer should be nil")

	// Verify file lock is released
	require.False(t, db.flock.Locked(), "File lock should be released")

	// Verify we can open the database again (proves lock is released)
	db2, err := Open(opts)
	require.NoError(t, err)
	require.NotNil(t, db2)

	// Verify data is still accessible
	err = db2.View(func(tx *Tx) error {
		val, err := tx.Get(bucket, testutils.GetTestBytes(0))
		require.NoError(t, err)
		require.NotNil(t, val)
		return nil
	})
	require.NoError(t, err)

	// Clean up
	err = db2.Close()
	require.NoError(t, err)
}

// TestDB_Close_ErrorHandling tests error handling during component shutdown