blob: 892287b4d86afe9459db0f7b0c6fa29b1e6a3079 [file] [log] [blame]
// Copyright 2020 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// +build !disable_events
package event_test
import (
"context"
"testing"
"golang.org/x/exp/event"
)
func TestCommon(t *testing.T) {
h := &catchHandler{}
ctx := event.WithExporter(context.Background(), event.NewExporter(h))
const simple = "simple message"
const trace = "a trace"
event.To(ctx).Log(simple)
checkFind(t, h, "Log", event.Message, true, simple)
checkFind(t, h, "Log", event.Trace, false, "")
event.To(ctx).Metric()
checkFind(t, h, "Metric", event.Message, false, "")
checkFind(t, h, "Metric", event.Trace, false, "")
event.To(ctx).Annotate()
checkFind(t, h, "Annotate", event.Message, false, "")
checkFind(t, h, "Annotate", event.Trace, false, "")
_, end := event.To(ctx).Start(trace)
checkFind(t, h, "Start", event.Message, false, "")
checkFind(t, h, "Start", event.Trace, true, trace)
end()
checkFind(t, h, "End", event.Message, false, "")
checkFind(t, h, "End", event.Trace, false, "")
}
type finder interface {
Find(*event.Event) (string, bool)
}
func checkFind(t *testing.T, h *catchHandler, method string, key finder, match bool, text string) {
m, ok := key.Find(&h.ev)
if ok && !match {
t.Errorf("%s produced an event with a %v", method, key)
}
if !ok && match {
t.Errorf("%s did not produce an event with a %v", method, key)
}
if m != text {
t.Errorf("Expected event with %v %q from %s got %q", key, text, method, m)
}
}
type catchHandler struct {
ev event.Event
}
func (h *catchHandler) Log(ctx context.Context, ev *event.Event) {
h.event(ctx, ev)
}
func (h *catchHandler) Metric(ctx context.Context, ev *event.Event) {
h.event(ctx, ev)
}
func (h *catchHandler) Annotate(ctx context.Context, ev *event.Event) {
h.event(ctx, ev)
}
func (h *catchHandler) Start(ctx context.Context, ev *event.Event) context.Context {
h.event(ctx, ev)
return ctx
}
func (h *catchHandler) End(ctx context.Context, ev *event.Event) {
h.event(ctx, ev)
}
func (h *catchHandler) event(ctx context.Context, ev *event.Event) {
h.ev = *ev
h.ev.Labels = make([]event.Label, len(ev.Labels))
copy(h.ev.Labels, ev.Labels)
}