Merge pull request ipfs/go-blockservice#28 from ipfs/fix/bs-131

set the session context This commit was moved from ipfs/go-blockservice@87a79d5
ipfs · May 24, 2019 · bd53d87 · bd53d87
2 parents a13f5d6 + 6bc6552
commit bd53d87
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/blockservice/blockservice.go b/blockservice/blockservice.go
@@ -115,14 +115,16 @@ func NewSession(ctx context.Context, bs BlockService) *Session {
 	exch := bs.Exchange()
 	if sessEx, ok := exch.(exchange.SessionExchange); ok {
 		return &Session{
-			ses:    nil,
-			sessEx: sessEx,
-			bs:     bs.Blockstore(),
+			sessCtx: ctx,
+			ses:     nil,
+			sessEx:  sessEx,
+			bs:      bs.Blockstore(),
 		}
 	}
 	return &Session{
-		ses: exch,
-		bs:  bs.Blockstore(),
+		ses:     exch,
+		sessCtx: ctx,
+		bs:      bs.Blockstore(),
 	}
 }
 

diff --git a/blockservice/blockservice_test.go b/blockservice/blockservice_test.go
@@ -102,6 +102,9 @@ type fakeSessionExchange struct {
 	session exchange.Fetcher
 }
 
-func (fe *fakeSessionExchange) NewSession(context.Context) exchange.Fetcher {
+func (fe *fakeSessionExchange) NewSession(ctx context.Context) exchange.Fetcher {
+	if ctx == nil {
+		panic("nil context")
+	}
 	return fe.session
 }
diff --git a/blockservice/test/mock.go b/blockservice/test/mock.go
@@ -3,7 +3,7 @@ package bstest
 import (
 	. "github.com/ipfs/go-blockservice"
 
-	bitswap "github.com/ipfs/go-bitswap"
+	testinstance "github.com/ipfs/go-bitswap/testinstance"
 	tn "github.com/ipfs/go-bitswap/testnet"
 	delay "github.com/ipfs/go-ipfs-delay"
 	mockrouting "github.com/ipfs/go-ipfs-routing/mock"
@@ -12,7 +12,7 @@ import (
 // Mocks returns |n| connected mock Blockservices
 func Mocks(n int) []BlockService {
 	net := tn.VirtualNetwork(mockrouting.NewServer(), delay.Fixed(0))
-	sg := bitswap.NewTestSessionGenerator(net)
+	sg := testinstance.NewTestInstanceGenerator(net)
 
 	instances := sg.Instances(n)