fission-src/pkg/executor/fscache/queue_test.go

package fscache

import (
	"context"
	"sync"
	"testing"
)

func TestNewQueue(t *testing.T) {
	q := NewQueue()
	if q == nil {
		t.Error("NewQueue returned nil")
	}
}

func TestQueuePushWithSingleRequest(t *testing.T) {
	q := NewQueue()
	item := &svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        nil,
	}
	q.Push(item)
	if q.Len() != 1 {
		t.Errorf("Expected queue length to be 1, got %d", q.Len())
	}
}

func TestQueuePopWithSingleRequest(t *testing.T) {
	q := NewQueue()
	item := &svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        nil,
	}
	q.Push(item)
	popped := q.Pop()
	if popped == nil {
		t.Error("Expected Pop to return a non-nil value")
	}
	if popped != item {
		t.Error("Expected Pop to return the same element that was pushed")
	}
	if q.Len() != 0 {
		t.Errorf("Expected queue length to be 0, got %d", q.Len())
	}
}

func TestQueuePushWithConcurrentRequest(t *testing.T) {
	q := NewQueue()
	noOfRequests := 20
	var wg sync.WaitGroup
	wg.Add(noOfRequests)
	for i := 0; i < noOfRequests; i++ {
		go func() {
			defer wg.Done()
			item := &svcWait{
				svcChannel: make(chan *FuncSvc),
				ctx:        nil,
			}
			q.Push(item)
		}()
	}

	wg.Wait()

	if q.Len() != noOfRequests {
		t.Errorf("Expected queue length to be 20, got %d", q.Len())
	}
}

func TestQueuePopWithConcurrentRequest(t *testing.T) {
	q := NewQueue()
	noOfPush := 20
	noOfPop := 15

	var wg sync.WaitGroup
	wg.Add(noOfPush + noOfPop)

	for i := 0; i < noOfPush; i++ {
		go func() {
			defer wg.Done()
			item := &svcWait{
				svcChannel: make(chan *FuncSvc),
				ctx:        nil,
			}
			q.Push(item)
		}()
	}

	for i := 0; i < noOfPop; i++ {
		go func() {
			defer wg.Done()
			q.Pop()
		}()
	}

	wg.Wait()

	if q.Len() != 5 {
		t.Errorf("Expected queue length to be 5, got %d", q.Len())
	}
}

func TestQueueLen(t *testing.T) {
	q := NewQueue()
	if q.Len() != 0 {
		t.Errorf("Expected queue length to be 0, got %d", q.Len())
	}
	item := &svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        nil,
	}
	q.Push(item)
	if q.Len() != 1 {
		t.Errorf("Expected queue length to be 1, got %d", q.Len())
	}
}

func TestExpiredWhenAllItemsExpired(t *testing.T) {
	q := NewQueue()
	if q.Expired() != 0 {
		t.Errorf("Expected Expired to return 0, got %d", q.Expired())
	}
	ctx, cancel := context.WithCancel(context.Background())
	item := &svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        ctx,
	}
	q.Push(item)
	if q.Len() != 1 {
		t.Errorf("Expected queue length to be 1, got %d", q.Len())
	}
	cancel()
	if q.Expired() != 1 {
		t.Errorf("Expected Expired to return 1, got %d", q.Expired())
	}
	if q.Len() != 0 {
		t.Errorf("Expected queue length to be 0, got %d", q.Len())
	}
}

func TestExpiredWhenFewItemsExpired(t *testing.T) {
	q := NewQueue()
	if q.Expired() != 0 {
		t.Errorf("Expected Expired to return 0, got %d", q.Expired())
	}
	ctx, cancel := context.WithCancel(context.Background())

	q.Push(&svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        ctx,
	})
	q.Push(&svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        context.Background(),
	})
	if q.Len() != 2 {
		t.Errorf("Expected queue length to be 1, got %d", q.Len())
	}
	cancel()
	if q.Expired() != 1 {
		t.Errorf("Expected Expired to return 1, got %d", q.Expired())
	}
	if q.Len() != 1 {
		t.Errorf("Expected queue length to be 0, got %d", q.Len())
	}
}

func TestExpiredWhenNoItemsExpired(t *testing.T) {
	q := NewQueue()
	if q.Expired() != 0 {
		t.Errorf("Expected Expired to return 0, got %d", q.Expired())
	}

	q.Push(&svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        context.Background(),
	})
	q.Push(&svcWait{
		svcChannel: make(chan *FuncSvc),
		ctx:        context.Background(),
	})
	if q.Len() != 2 {
		t.Errorf("Expected queue length to be 1, got %d", q.Len())
	}
	if q.Expired() != 0 {
		t.Errorf("Expected Expired to return 1, got %d", q.Expired())
	}
	if q.Len() != 2 {
		t.Errorf("Expected queue length to be 0, got %d", q.Len())
	}
}