// Copyright 2015 The etcd Authors
//
// 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"

	"go.etcd.io/raft/v3/raftpb"
)

func TestPeerPick(t *testing.T) {
	tests := []struct {
		msgappWorking  bool
		messageWorking bool
		m              *raftpb.Message
		wpicked        string
	}{
		{
			true, true,
			&raftpb.Message{Type: raftpb.MsgSnap.Enum()},
			pipelineMsg,
		},
		{
			true, true,
			&raftpb.Message{Type: raftpb.MsgApp.Enum(), Term: new(uint64(1)), LogTerm: new(uint64(1))},
			streamAppV2,
		},
		{
			true, true,
			&raftpb.Message{Type: raftpb.MsgProp.Enum()},
			streamMsg,
		},
		{
			true, true,
			&raftpb.Message{Type: raftpb.MsgHeartbeat.Enum()},
			streamMsg,
		},
		{
			false, true,
			&raftpb.Message{Type: raftpb.MsgApp.Enum(), Term: new(uint64(1)), LogTerm: new(uint64(1))},
			streamMsg,
		},
		{
			false, false,
			&raftpb.Message{Type: raftpb.MsgApp.Enum(), Term: new(uint64(1)), LogTerm: new(uint64(1))},
			pipelineMsg,
		},
		{
			false, false,
			&raftpb.Message{Type: raftpb.MsgProp.Enum()},
			pipelineMsg,
		},
		{
			false, false,
			&raftpb.Message{Type: raftpb.MsgSnap.Enum()},
			pipelineMsg,
		},
		{
			false, false,
			&raftpb.Message{Type: raftpb.MsgHeartbeat.Enum()},
			pipelineMsg,
		},
	}
	for i, tt := range tests {
		peer := &peer{
			msgAppV2Writer: &streamWriter{working: tt.msgappWorking},
			writer:         &streamWriter{working: tt.messageWorking},
			pipeline:       &pipeline{},
		}
		_, picked := peer.pick(tt.m)
		if picked != tt.wpicked {
			t.Errorf("#%d: picked = %v, want %v", i, picked, tt.wpicked)
		}
	}
}