update

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
appleboy · Dec 6, 2024 · ded1028 · ded1028
1 parent 2df67ce
commit ded1028
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/easyssh_test.go b/easyssh_test.go
@@ -1,6 +1,7 @@
 package easyssh
 
 import (
+	"context"
 	"os"
 	"os/user"
 	"path"
@@ -508,5 +509,6 @@ func TestCommandTimeout(t *testing.T) {
 	assert.Equal(t, "root\n", outStr)
 	assert.Equal(t, "", errStr)
 	assert.False(t, isTimeout)
-	assert.NoError(t, err)
+	assert.NotNil(t, err)
+	assert.Equal(t, "Run Command Timeout: "+context.DeadlineExceeded.Error(), err.Error())
 }