43 lines
757 B
Go
43 lines
757 B
Go
package meter
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestFromNilContext(t *testing.T) {
|
|
// nolint: staticcheck
|
|
c, ok := FromContext(nil)
|
|
if ok || c != nil {
|
|
t.Fatal("FromContext not works")
|
|
}
|
|
}
|
|
|
|
func TestNewNilContext(t *testing.T) {
|
|
// nolint: staticcheck
|
|
ctx := NewContext(nil, NewMeter())
|
|
|
|
c, ok := FromContext(ctx)
|
|
if c == nil || !ok {
|
|
t.Fatal("NewContext not works")
|
|
}
|
|
}
|
|
|
|
func TestFromContext(t *testing.T) {
|
|
ctx := context.WithValue(context.TODO(), meterKey{}, NewMeter())
|
|
|
|
c, ok := FromContext(ctx)
|
|
if c == nil || !ok {
|
|
t.Fatal("FromContext not works")
|
|
}
|
|
}
|
|
|
|
func TestNewContext(t *testing.T) {
|
|
ctx := NewContext(context.TODO(), NewMeter())
|
|
|
|
c, ok := FromContext(ctx)
|
|
if c == nil || !ok {
|
|
t.Fatal("NewContext not works")
|
|
}
|
|
}
|