git » go-net » unix-separados » tree

[unix-separados] / http2 / fixed_buffer_test.go 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// Copyright 2014 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 http2

import (
	"reflect"
	"testing"
)

var bufferReadTests = []struct {
	buf      fixedBuffer
	read, wn int
	werr     error
	wp       []byte
	wbuf     fixedBuffer
}{
	{
		fixedBuffer{[]byte{'a', 0}, 0, 1},
		5, 1, nil, []byte{'a'},
		fixedBuffer{[]byte{'a', 0}, 0, 0},
	},
	{
		fixedBuffer{[]byte{0, 'a'}, 1, 2},
		5, 1, nil, []byte{'a'},
		fixedBuffer{[]byte{0, 'a'}, 0, 0},
	},
	{
		fixedBuffer{[]byte{'a', 'b'}, 0, 2},
		1, 1, nil, []byte{'a'},
		fixedBuffer{[]byte{'a', 'b'}, 1, 2},
	},
	{
		fixedBuffer{[]byte{}, 0, 0},
		5, 0, errReadEmpty, []byte{},
		fixedBuffer{[]byte{}, 0, 0},
	},
}

func TestBufferRead(t *testing.T) {
	for i, tt := range bufferReadTests {
		read := make([]byte, tt.read)
		n, err := tt.buf.Read(read)
		if n != tt.wn {
			t.Errorf("#%d: wn = %d want %d", i, n, tt.wn)
			continue
		}
		if err != tt.werr {
			t.Errorf("#%d: werr = %v want %v", i, err, tt.werr)
			continue
		}
		read = read[:n]
		if !reflect.DeepEqual(read, tt.wp) {
			t.Errorf("#%d: read = %+v want %+v", i, read, tt.wp)
		}
		if !reflect.DeepEqual(tt.buf, tt.wbuf) {
			t.Errorf("#%d: buf = %+v want %+v", i, tt.buf, tt.wbuf)
		}
	}
}

var bufferWriteTests = []struct {
	buf       fixedBuffer
	write, wn int
	werr      error
	wbuf      fixedBuffer
}{
	{
		buf: fixedBuffer{
			buf: []byte{},
		},
		wbuf: fixedBuffer{
			buf: []byte{},
		},
	},
	{
		buf: fixedBuffer{
			buf: []byte{1, 'a'},
		},
		write: 1,
		wn:    1,
		wbuf: fixedBuffer{
			buf: []byte{0, 'a'},
			w:   1,
		},
	},
	{
		buf: fixedBuffer{
			buf: []byte{'a', 1},
			r:   1,
			w:   1,
		},
		write: 2,
		wn:    2,
		wbuf: fixedBuffer{
			buf: []byte{0, 0},
			w:   2,
		},
	},
	{
		buf: fixedBuffer{
			buf: []byte{},
		},
		write: 5,
		werr:  errWriteFull,
		wbuf: fixedBuffer{
			buf: []byte{},
		},
	},
}

func TestBufferWrite(t *testing.T) {
	for i, tt := range bufferWriteTests {
		n, err := tt.buf.Write(make([]byte, tt.write))
		if n != tt.wn {
			t.Errorf("#%d: wrote %d bytes; want %d", i, n, tt.wn)
			continue
		}
		if err != tt.werr {
			t.Errorf("#%d: error = %v; want %v", i, err, tt.werr)
			continue
		}
		if !reflect.DeepEqual(tt.buf, tt.wbuf) {
			t.Errorf("#%d: buf = %+v; want %+v", i, tt.buf, tt.wbuf)
		}
	}
}