1 // Copyright 2011 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.
13 func TestReadVersion(t *testing.T) {
14 buf := []byte(serverVersion)
15 result, ok := readVersion(bufio.NewReader(bytes.NewBuffer(buf)))
17 t.Error("readVersion didn't read version correctly")
19 if !bytes.Equal(buf[:len(buf)-2], result) {
20 t.Error("version read did not match expected")
24 func TestReadVersionTooLong(t *testing.T) {
25 buf := make([]byte, maxVersionStringBytes+1)
26 if _, ok := readVersion(bufio.NewReader(bytes.NewBuffer(buf))); ok {
27 t.Errorf("readVersion consumed %d bytes without error", len(buf))
31 func TestReadVersionWithoutCRLF(t *testing.T) {
32 buf := []byte(serverVersion)
33 buf = buf[:len(buf)-1]
34 if _, ok := readVersion(bufio.NewReader(bytes.NewBuffer(buf))); ok {
35 t.Error("readVersion did not notice \\n was missing")