github.com/annwntech/go-micro/v2@v2.9.5/debug/log/memory/memory_test.go (about)

     1  package memory
     2  
     3  import (
     4  	"reflect"
     5  	"testing"
     6  
     7  	"github.com/annwntech/go-micro/v2/debug/log"
     8  )
     9  
    10  func TestLogger(t *testing.T) {
    11  	// set size to some value
    12  	size := 100
    13  	// override the global logger
    14  	lg := NewLog(log.Size(size))
    15  	// make sure we have the right size of the logger ring buffer
    16  	if lg.(*memoryLog).Size() != size {
    17  		t.Errorf("expected buffer size: %d, got: %d", size, lg.(*memoryLog).Size())
    18  	}
    19  
    20  	// Log some cruft
    21  	lg.Write(log.Record{Message: "foobar"})
    22  	lg.Write(log.Record{Message: "foo bar"})
    23  
    24  	// Check if the logs are stored in the logger ring buffer
    25  	expected := []string{"foobar", "foo bar"}
    26  	entries, _ := lg.Read(log.Count(len(expected)))
    27  	for i, entry := range entries {
    28  		if !reflect.DeepEqual(entry.Message, expected[i]) {
    29  			t.Errorf("expected %s, got %s", expected[i], entry.Message)
    30  		}
    31  	}
    32  }