Optimize error handling methods (#403)

no-src · Dec 31, 2024 · ede1cd9 · ede1cd9
1 parent a6502b0
commit ede1cd9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/driver/sftp/sftp.go b/driver/sftp/sftp.go
@@ -161,7 +161,7 @@ func (sd *sftpDriver) reconnectIfLost(f func() error) error {
 }
 
 func (sd *sftpDriver) isClosed(err error) bool {
-	return err == sftp.ErrSSHFxConnectionLost
+	return errors.Is(err, sftp.ErrSSHFxConnectionLost)
 }
 
 func (sd *sftpDriver) MkdirAll(path string) error {

diff --git a/internal/toplist/toplist_test.go b/internal/toplist/toplist_test.go
@@ -2,6 +2,7 @@ package toplist
 
 import (
 	"bytes"
+	"errors"
 	"sync"
 	"testing"
 
@@ -12,14 +13,14 @@ func TestTopList_WithCapZero(t *testing.T) {
 	// desc
 	capacity := 0
 	_, err := New(capacity)
-	if err != errInvalidCapacity {
+	if !errors.Is(err, errInvalidCapacity) {
 		t.Errorf("[desc] test toplist with zero capacity error, expect get an error")
 	}
 
 	// asc
 	capacity = 0
 	_, err = NewOrderByAsc(capacity)
-	if err != errInvalidCapacity {
+	if !errors.Is(err, errInvalidCapacity) {
 		t.Errorf("[asc] test toplist with zero capacity error, expect get an error")
 	}
 }