minor test fixes.

internal/transport/writer.go

@@ -43,7 +43,21 @@ func (c *Reader) begin() {
 	}
 }
 
+func isClosed(ch <-chan int) bool {
+	select {
+	case <-ch:
+		return true
+	default:
+	}
+
+	return false
+}
+
 func (c *Reader) Read(p []byte) (int, error) {
+	if isClosed(c.sizeWant) {
+		return 0, io.EOF
+	}
+
 	select {
 	case <-c.ctx.Done():
 		return 0, c.ctx.Err()

telegram/client.go

@@ -27,7 +27,7 @@ import (
 const (
 	// The Initial DC to connect to, before auth
 	DefaultDataCenter       = 4
-	DisconnectExportedAfter = 5 * time.Minute
+	DisconnectExportedAfter = 15 * time.Minute
 )
 
 type clientData struct {
@@ -364,12 +364,15 @@ func (c *Client) loopForCleaningExpiredSenders() {
 	for {
 		time.Sleep(DisconnectExportedAfter)
 		c.exportedSenders.Lock()
-		for i, s := range c.exportedSenders.senders {
+		newSenders := c.exportedSenders.senders[:0]
+		for _, s := range c.exportedSenders.senders {
 			if time.Since(s.added) > DisconnectExportedAfter {
 				s.client.Terminate()
-				c.exportedSenders.senders = append(c.exportedSenders.senders[:i], c.exportedSenders.senders[i+1:]...)
+			} else {
+				newSenders = append(newSenders, s)
 			}
 		}
+		c.exportedSenders.senders = newSenders
 		c.exportedSenders.Unlock()
 	}
 }

telegram/media.go

@@ -355,6 +355,13 @@ func (mb *Destination) WriteAt(p []byte, off int64) (n int, err error) {
 	return len(p), nil
 }
 
+func (mb *Destination) Close() error {
+	if mb.file != nil {
+		return mb.file.Close()
+	}
+	return nil
+}
+
 func (c *Client) DownloadMedia(file interface{}, Opts ...*DownloadOptions) (string, error) {
 	opts := getVariadic(Opts, &DownloadOptions{})
 	location, dc, size, fileName, err := GetFileLocation(file)
@@ -381,12 +388,10 @@ func (c *Client) DownloadMedia(file interface{}, Opts ...*DownloadOptions) (stri
 		}
 
 		fs.file = file
-
-		defer func() {
-			fs.file.Close()
-		}()
 	}
 
+	defer fs.Close()
+
 	parts := size / int64(partSize)
 	partOver := size % int64(partSize)