// Copyright 2011 The Go 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 ssh

import (
	"bufio"
	"bytes"
	"testing"
)

func TestReadVersion(t *testing.T) {
	buf := serverVersion
	result, err := readVersion(bufio.NewReader(bytes.NewBuffer(buf)))
	if err != nil {
		t.Errorf("readVersion didn't read version correctly: %s", err)
	}
	if !bytes.Equal(buf[:len(buf)-2], result) {
		t.Error("version read did not match expected")
	}
}

func TestReadVersionWithJustLF(t *testing.T) {
	var buf []byte
	buf = append(buf, serverVersion...)
	buf = buf[:len(buf)-1]
	buf[len(buf)-1] = '\n'
	result, err := readVersion(bufio.NewReader(bytes.NewBuffer(buf)))
	if err != nil {
		t.Error("readVersion failed to handle just a \n")
	}
	if !bytes.Equal(buf[:len(buf)-1], result) {
		t.Errorf("version read did not match expected: got %x, want %x", result, buf[:len(buf)-1])
	}
}

func TestReadVersionTooLong(t *testing.T) {
	buf := make([]byte, maxVersionStringBytes+1)
	if _, err := readVersion(bufio.NewReader(bytes.NewBuffer(buf))); err == nil {
		t.Errorf("readVersion consumed %d bytes without error", len(buf))
	}
}

func TestReadVersionWithoutCRLF(t *testing.T) {
	buf := serverVersion
	buf = buf[:len(buf)-1]
	if _, err := readVersion(bufio.NewReader(bytes.NewBuffer(buf))); err == nil {
		t.Error("readVersion did not notice \\n was missing")
	}
}