golang expect包怎么使用

# Golang expect包怎么使用

## 什么是expect包

`expect`是一個用于自動化交互式命令行程序的Go語言庫，靈感來源于Unix的`expect`工具。它允許開發者模擬用戶輸入，并根據程序輸出進行條件匹配和響應，常用于測試或自動化運維場景。

## 安裝expect包

通過go get安裝：
```bash
go get github.com/Netflix/go-expect

基本使用方法

1. 創建expect會話

c, err := expect.NewConsole(expect.WithStdout(os.Stdout))
if err != nil {
    log.Fatal(err)
}
defer c.Close()

2. 發送命令和期望輸出

// 發送命令
_, err = c.Send("ls -l\n")
if err != nil {
    log.Fatal(err)
}

// 等待特定輸出
_, err = c.ExpectString("total")
if err != nil {
    log.Fatal(err)
}

3. 處理交互式提示

// 匹配密碼提示并自動響應
_, err = c.ExpectString("Password:")
if err != nil {
    log.Fatal(err)
}
_, err = c.SendLine("mypassword")

高級功能

1. 正則表達式匹配

// 使用正則匹配輸出
match, err := c.Expect(expect.RegexpPattern(`\d+ files`))
fmt.Printf("Found: %s\n", match)

2. 超時設置

// 設置5秒超時
_, err = c.ExpectString("login:", expect.WithTimeout(5*time.Second))

3. 多條件匹配

// 匹配成功或失敗
result, err := c.Expect(
    expect.String("Login successful"),
    expect.String("Login failed"),
)

實際應用示例

自動化SSH登錄

func TestSSHLogin(t *testing.T) {
    c, err := expect.NewConsole()
    // 錯誤處理省略...
    
    go func() {
        c.Send("ssh user@host\n")
        c.ExpectString("password:")
        c.SendLine("secret")
        c.ExpectString("$ ")
        c.Send("exit\n")
        c.ExpectEOF()
    }()
    
    cmd := exec.Command("bash")
    cmd.Stdin = c.Tty()
    cmd.Stdout = c.Tty()
    cmd.Stderr = c.Tty()
    cmd.Run()
}

常見問題解決

1. 終端模式問題：

   • 使用expect.WithTermSize(cols, rows)設置虛擬終端尺寸

2. 特殊字符處理：

   • 對于控制字符，使用Send(string)而非SendLine()

3. 并發安全：

   • 所有expect方法都是線程安全的，可在goroutine中使用

替代方案比較

總結

go-expect包為Golang提供了強大的交互式命令行自動化能力，特別適合： - CLI工具測試 - 自動化運維腳本 - 復雜交互流程驗證

通過合理的輸出匹配和輸入模擬，可以構建穩定的自動化測試套件。其API設計簡潔，與Go的標準庫良好集成，是Golang生態中處理交互式命令的首選方案。

注意：實際使用時請根據具體需求添加錯誤處理和日志記錄 “`