banner_test.go 704 B

12345678910111213141516171819202122232425262728293031
  1. // Copyright 2014 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. //go:build aix || darwin || dragonfly || freebsd || linux || netbsd || openbsd || solaris
  5. package test
  6. import (
  7. "testing"
  8. )
  9. func TestBannerCallbackAgainstOpenSSH(t *testing.T) {
  10. server := newServer(t)
  11. clientConf := clientConfig()
  12. var receivedBanner string
  13. clientConf.BannerCallback = func(message string) error {
  14. receivedBanner = message
  15. return nil
  16. }
  17. conn := server.Dial(clientConf)
  18. defer conn.Close()
  19. expected := "Server Banner"
  20. if receivedBanner != expected {
  21. t.Fatalf("got %v; want %v", receivedBanner, expected)
  22. }
  23. }