// Copyright 2024 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.

package rpcserver

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestLastExecutingEmpty(t *testing.T) {
	last := MakeLastExecuting(10, 10)
	assert.Empty(t, last.Collect())
}

func TestLastExecuting(t *testing.T) {
	last := MakeLastExecuting(21, 3)
	last.Note(1, 0, []byte("prog1"), 1)

	last.Note(2, 1, []byte("prog2"), 2)
	last.Note(3, 1, []byte("prog3"), 3)

	last.Note(4, 3, []byte("prog4"), 4)
	last.Note(5, 3, []byte("prog5"), 5)
	last.Note(6, 3, []byte("prog6"), 6)

	last.Note(7, 7, []byte("prog7"), 7)
	last.Note(8, 7, []byte("prog8"), 8)
	last.Note(9, 7, []byte("prog9"), 9)
	last.Note(10, 7, []byte("prog10"), 10)
	last.Note(11, 7, []byte("prog11"), 11)

	last.Note(12, 9, []byte("prog12"), 12)

	last.Note(13, 8, []byte("prog13"), 13)

	assert.Equal(t, last.Collect(), []ExecRecord{
		{ID: 1, Proc: 0, Prog: []byte("prog1"), Time: 12},

		{ID: 2, Proc: 1, Prog: []byte("prog2"), Time: 11},
		{ID: 3, Proc: 1, Prog: []byte("prog3"), Time: 10},

		{ID: 4, Proc: 3, Prog: []byte("prog4"), Time: 9},
		{ID: 5, Proc: 3, Prog: []byte("prog5"), Time: 8},
		{ID: 6, Proc: 3, Prog: []byte("prog6"), Time: 7},

		{ID: 9, Proc: 7, Prog: []byte("prog9"), Time: 4},
		{ID: 10, Proc: 7, Prog: []byte("prog10"), Time: 3},
		{ID: 11, Proc: 7, Prog: []byte("prog11"), Time: 2},

		{ID: 12, Proc: 9, Prog: []byte("prog12"), Time: 1},

		{ID: 13, Proc: 8, Prog: []byte("prog13"), Time: 0},
	})
}

func TestLastExecutingHanged(t *testing.T) {
	last := MakeLastExecuting(1, 3)
	last.Note(1, 0, []byte("prog1"), 10)
	last.Note(2, 0, []byte("prog2"), 20)
	last.Hanged(2, 0, []byte("prog2"), 25)
	last.Note(3, 0, []byte("prog3"), 30)
	last.Note(4, 0, []byte("prog4"), 40)
	last.Note(5, 0, []byte("prog5"), 50)
	last.Hanged(5, 0, []byte("prog5"), 55)
	last.Note(6, 0, []byte("prog6"), 60)
	last.Note(7, 0, []byte("prog7"), 70)
	last.Note(8, 0, []byte("prog8"), 80)
	last.Note(9, 0, []byte("prog9"), 90)
	assert.Equal(t, last.Collect(), []ExecRecord{
		{ID: 2, Proc: 32, Prog: []byte("prog2"), Time: 65},
		{ID: 5, Proc: 33, Prog: []byte("prog5"), Time: 35},
		{ID: 7, Proc: 0, Prog: []byte("prog7"), Time: 20},
		{ID: 8, Proc: 0, Prog: []byte("prog8"), Time: 10},
		{ID: 9, Proc: 0, Prog: []byte("prog9"), Time: 0},
	})
}