etcd/rafthttp/batcher_test.go

// Copyright 2015 CoreOS, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package rafthttp

import (
	"testing"
	"time"
)

func TestBatcherNum(t *testing.T) {
	n := 100
	largeD := time.Minute
	tests := []struct {
		n         int
		wnotbatch int
	}{
		{n - 1, 0},
		{n, 1},
		{n + 1, 1},
		{2*n + 1, 2},
		{3*n + 1, 3},
	}

	for i, tt := range tests {
		b := NewBatcher(n, largeD)
		notbatched := 0
		for j := 0; j < tt.n; j++ {
			if !b.ShouldBatch(time.Now()) {
				notbatched++
			}
		}
		if notbatched != tt.wnotbatch {
			t.Errorf("#%d: notbatched = %d, want %d", i, notbatched, tt.wnotbatch)
		}
	}
}

func TestBatcherTime(t *testing.T) {
	largeN := 10000
	tests := []struct {
		nms       int
		wnotbatch int
	}{
		{0, 0},
		{1, 1},
		{2, 2},
		{3, 3},
	}

	for i, tt := range tests {
		b := NewBatcher(largeN, time.Millisecond)
		baseT := b.batchedT
		notbatched := 0
		for j := 0; j < tt.nms+1; j++ {
			if !b.ShouldBatch(baseT.Add(time.Duration(j) * time.Millisecond)) {
				notbatched++
			}
		}
		if notbatched != tt.wnotbatch {
			t.Errorf("#%d: notbatched = %d, want %d", i, notbatched, tt.wnotbatch)
		}
	}
}