@@ -7,7 +7,11 @@ import (
77 "reflect"
88 "testing"
99
10+ "github.com/go-logr/logr"
11+ "github.com/go-logr/logr/funcr"
1012 "github.com/stretchr/testify/assert"
13+
14+ "go.opentelemetry.io/otel/internal/global"
1115)
1216
1317func init () {
@@ -36,18 +40,25 @@ func TestCopy(t *testing.T) {
3640
3741func TestDropCount (t * testing.T ) {
3842 q := newEvictedQueueEvent (3 )
39- var called bool
40- q .logDropped = func () { called = true }
43+
44+ var called int
45+ t .Cleanup (func (l logr.Logger ) func () {
46+ return func () { global .SetLogger (l ) }
47+ }(global .GetLogger ()))
48+ global .SetLogger (funcr .New (func (prefix , args string ) {
49+ called ++
50+ }, funcr.Options {Verbosity : 1 }))
4151
4252 q .add (Event {Name : "value1" })
43- assert .False ( t , called , `"value1" logged as dropped` )
53+ assert .Equal ( t , 0 , called , `"value1" logged as dropped` )
4454 q .add (Event {Name : "value2" })
45- assert .False ( t , called , `"value2" logged as dropped` )
55+ assert .Equal ( t , 0 , called , `"value2" logged as dropped` )
4656 q .add (Event {Name : "value3" })
47- assert .False ( t , called , `"value3" logged as dropped` )
57+ assert .Equal ( t , 0 , called , `"value3" logged as dropped` )
4858 q .add (Event {Name : "value1" })
49- assert .True ( t , called , `"value2" not logged as dropped` )
59+ assert .Equal ( t , 1 , called , `"value2" not logged as dropped` )
5060 q .add (Event {Name : "value4" })
61+ assert .Equal (t , 1 , called , `"value4" logged as dropped` )
5162 if wantLen , gotLen := 3 , len (q .queue ); wantLen != gotLen {
5263 t .Errorf ("got queue length %d want %d" , gotLen , wantLen )
5364 }
0 commit comments