// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package sshutil

import (
	"errors"
	"fmt"
	"testing"

	"golang.org/x/crypto/ssh"
)

func TestConnectionError(t *testing.T) {
	if IsConnectionError(errors.New("not a connection error")) {
		t.Errorf("IsConnectionError returned true for a non-ConnectionError")
	}

	wrapped := new(ssh.ExitMissingError)
	connErr := ConnectionError{Err: wrapped}
	if !IsConnectionError(connErr) {
		t.Errorf("IsConnectionError returned false for a ConnectionError")
	}
	var eme *ssh.ExitMissingError
	if !errors.As(connErr, &eme) {
		t.Errorf("connErr should wrap ssh.ExitMissingError")
	}

	wrapper := fmt.Errorf("something failed: %w", connErr)
	if !IsConnectionError(wrapper) {
		t.Errorf("a wrapped ConnectionError should still be considered a ConnectionError")
	}
}