mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 15:05:31 +08:00
18 lines
273 B
Go
18 lines
273 B
Go
package message
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMessageHandler(t *testing.T) {
|
|
ch := make(chan ImmutableMessage, 100)
|
|
h := ChanMessageHandler(ch)
|
|
h.Handle(nil)
|
|
assert.Nil(t, <-ch)
|
|
h.Close()
|
|
_, ok := <-ch
|
|
assert.False(t, ok)
|
|
}
|