heapdump_test.go

Documentation: runtime/debug

		 1  // Copyright 2014 The Go Authors. All rights reserved.
		 2  // Use of this source code is governed by a BSD-style
		 3  // license that can be found in the LICENSE file.
		 4  
		 5  package debug_test
		 6  
		 7  import (
		 8  	"os"
		 9  	"runtime"
		10  	. "runtime/debug"
		11  	"testing"
		12  )
		13  
		14  func TestWriteHeapDumpNonempty(t *testing.T) {
		15  	if runtime.GOOS == "js" {
		16  		t.Skipf("WriteHeapDump is not available on %s.", runtime.GOOS)
		17  	}
		18  	f, err := os.CreateTemp("", "heapdumptest")
		19  	if err != nil {
		20  		t.Fatalf("TempFile failed: %v", err)
		21  	}
		22  	defer os.Remove(f.Name())
		23  	defer f.Close()
		24  	WriteHeapDump(f.Fd())
		25  	fi, err := f.Stat()
		26  	if err != nil {
		27  		t.Fatalf("Stat failed: %v", err)
		28  	}
		29  	const minSize = 1
		30  	if size := fi.Size(); size < minSize {
		31  		t.Fatalf("Heap dump size %d bytes, expected at least %d bytes", size, minSize)
		32  	}
		33  }
		34  
		35  type Obj struct {
		36  	x, y int
		37  }
		38  
		39  func objfin(x *Obj) {
		40  	//println("finalized", x)
		41  }
		42  
		43  func TestWriteHeapDumpFinalizers(t *testing.T) {
		44  	if runtime.GOOS == "js" {
		45  		t.Skipf("WriteHeapDump is not available on %s.", runtime.GOOS)
		46  	}
		47  	f, err := os.CreateTemp("", "heapdumptest")
		48  	if err != nil {
		49  		t.Fatalf("TempFile failed: %v", err)
		50  	}
		51  	defer os.Remove(f.Name())
		52  	defer f.Close()
		53  
		54  	// bug 9172: WriteHeapDump couldn't handle more than one finalizer
		55  	println("allocating objects")
		56  	x := &Obj{}
		57  	runtime.SetFinalizer(x, objfin)
		58  	y := &Obj{}
		59  	runtime.SetFinalizer(y, objfin)
		60  
		61  	// Trigger collection of x and y, queueing of their finalizers.
		62  	println("starting gc")
		63  	runtime.GC()
		64  
		65  	// Make sure WriteHeapDump doesn't fail with multiple queued finalizers.
		66  	println("starting dump")
		67  	WriteHeapDump(f.Fd())
		68  	println("done dump")
		69  }
		70
View as plain text