harness-drone/livelog/sub_test.go
2019-02-27 23:07:13 -08:00

83 lines
1.8 KiB
Go

// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.
// +build !oss
package livelog
import (
"testing"
"github.com/drone/drone/core"
)
func TestSubscription_publish(t *testing.T) {
s := &subscriber{
handler: make(chan *core.Line, 5),
closec: make(chan struct{}),
}
e := new(core.Line)
s.publish(e)
if got, want := len(s.handler), 1; got != want {
t.Errorf("Want buffered channel size %d, got %d", want, got)
}
if got, want := <-s.handler, e; got != want {
t.Errorf("Want log entry received from channel")
}
if got, want := len(s.handler), 0; got != want {
t.Errorf("Want buffered channel size %d, got %d", want, got)
}
}
func TestSubscription_buffer(t *testing.T) {
s := &subscriber{
handler: make(chan *core.Line, 1),
closec: make(chan struct{}),
}
// the buffer size is 1 to simulate what happens
// if the subscriber cannot keep up with processing
// and the buffer fills up. In this case, lines
// should be ignored until pending lines are
// processed.
e := new(core.Line)
s.publish(e)
s.publish(e)
s.publish(e)
s.publish(e)
s.publish(e)
if got, want := len(s.handler), 1; got != want {
t.Errorf("Want buffered channel size %d, got %d", want, got)
}
}
func TestSubscription_stop(t *testing.T) {
s := &subscriber{
handler: make(chan *core.Line, 1),
closec: make(chan struct{}),
}
if got, want := s.closed, false; got != want {
t.Errorf("Want subscription open")
}
s.close()
if got, want := s.closed, true; got != want {
t.Errorf("Want subscription closed")
}
// if the subscription is closed we should
// ignore any new events being published.
e := new(core.Line)
s.publish(e)
s.publish(e)
s.publish(e)
s.publish(e)
s.publish(e)
}