Commit ab65a184 authored by Steven Allen's avatar Steven Allen

bitswap: test canceling subscription context after shutting down

License: MIT
Signed-off-by: default avatarSteven Allen <steven@stebalien.com>
parent e39ba627
...@@ -100,6 +100,25 @@ func TestDuplicateSubscribe(t *testing.T) { ...@@ -100,6 +100,25 @@ func TestDuplicateSubscribe(t *testing.T) {
assertBlocksEqual(t, e1, r2) assertBlocksEqual(t, e1, r2)
} }
func TestShutdownBeforeUnsubscribe(t *testing.T) {
e1 := blocks.NewBlock([]byte("1"))
n := New()
ctx, cancel := context.WithCancel(context.Background())
ch := n.Subscribe(ctx, e1.Cid()) // no keys provided
n.Shutdown()
cancel()
select {
case _, ok := <-ch:
if ok {
t.Fatal("channel should have been closed")
}
default:
t.Fatal("channel should have been closed")
}
}
func TestSubscribeIsANoopWhenCalledWithNoKeys(t *testing.T) { func TestSubscribeIsANoopWhenCalledWithNoKeys(t *testing.T) {
n := New() n := New()
defer n.Shutdown() defer n.Shutdown()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment