fix: pass a context to PinsStream

Author: Stebalien

shell.go

@@ -245,10 +245,10 @@ type PinStreamInfo struct {
 
 // PinsStream is a streamed version of Pins. It returns a channel of the pins
 // with their type, one of DirectPin, RecursivePin, or IndirectPin.
-func (s *Shell) PinsStream() (<-chan PinStreamInfo, error) {
+func (s *Shell) PinsStream(ctx context.Context) (<-chan PinStreamInfo, error) {
 	resp, err := s.Request("pin/ls").
 		Option("stream", true).
-		Send(context.Background())
+		Send(ctx)
 	if err != nil {
 		return nil, err
 	}
@@ -269,7 +269,11 @@ func (s *Shell) PinsStream() (<-chan PinStreamInfo, error) {
 			if err != nil {
 				return
 			}
-			out <- pin
+			select {
+			case out <- pin:
+			case <-ctx.Done():
+				return
+			}
 		}
 	}()
 

shell_test.go

@@ -247,7 +247,7 @@ func TestPinsStream(t *testing.T) {
 	h, err := s.Add(bytes.NewBufferString("go-ipfs-api pins test 0C7023F8-1FEC-4155-A8E0-432A5853F46B"))
 	is.Nil(err)
 
-	pinChan, err := s.PinsStream()
+	pinChan, err := s.PinsStream(context.Background())
 	is.Nil(err)
 
 	pins := accumulatePins(pinChan)
@@ -258,7 +258,7 @@ func TestPinsStream(t *testing.T) {
 	err = s.Unpin(h)
 	is.Nil(err)
 
-	pinChan, err = s.PinsStream()
+	pinChan, err = s.PinsStream(context.Background())
 	is.Nil(err)
 
 	pins = accumulatePins(pinChan)
@@ -269,7 +269,7 @@ func TestPinsStream(t *testing.T) {
 	err = s.Pin(h)
 	is.Nil(err)
 
-	pinChan, err = s.PinsStream()
+	pinChan, err = s.PinsStream(context.Background())
 	is.Nil(err)
 
 	pins = accumulatePins(pinChan)