Function TestUsagePusher_RunCancelExits

internal/nodeagent/usage_test.go:212–229

(t *testing.T)

}

func TestUsagePusher_RunCancelExits(t *testing.T) {
	t.Parallel()
	api := &mockUsageAPI{}
	p := NewUsagePusher(api, 10*time.Millisecond)
	ctx, cancel := context.WithCancel(context.Background())
	done := make(chan error, 1)
	go func() { done <- p.Run(ctx) }()
	time.Sleep(30 * time.Millisecond)
	cancel()
	select {
	case err := <-done:
		if !errors.Is(err, context.Canceled) {
			t.Fatalf("Run err = %v, want Canceled", err)
		}
	case <-time.After(time.Second):
		t.Fatalf("Run did not return on cancel")
	}
}