slice_chan_bytes_test.go

package concurrent

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestSliceChanBytes(t *testing.T) {
	s := NewSliceChanBytes()

	ch := make(chan []byte, 5)
	s.Add(ch)
	assert.Equal(t, 1, s.Len())

	ch2 := make(chan []byte, 5)
	s.Add(ch2)
	assert.Equal(t, 2, s.Len())

	ch3 := make(chan []byte, 5)
	s.Add(ch3)
	assert.Equal(t, 3, s.Len())

	ch4 := make(chan []byte, 5)
	s.Add(ch4)
	assert.Equal(t, 4, s.Len())

	b := []byte("bla")
	s.Send(b)
	// TODO: remove?
	time.Sleep(200 * time.Millisecond)
	assert.Equal(t, len(ch), 1)
	assert.Equal(t, len(ch2), 1)
	assert.Equal(t, len(ch3), 1)
	assert.Equal(t, len(ch3), 1)
	assert.Equal(t, len(ch4), 1)
	assert.Equal(t, b, <-ch)
	assert.Equal(t, b, <-ch2)
	assert.Equal(t, b, <-ch3)
	assert.Equal(t, b, <-ch4)

	assert.True(t, s.Remove(ch2))
	assert.Equal(t, 3, s.Len())

	assert.True(t, s.Remove(ch4))
	assert.Equal(t, 2, s.Len())

	s.RemoveAll()
	assert.Equal(t, 0, s.Len())
}