webrtc/operations_test.go

// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT

package webrtc

import (
	"sync"
	"sync/atomic"
	"testing"

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

func TestOperations_Enqueue(t *testing.T) {
	updateNegotiationNeededFlagOnEmptyChain := &atomic.Bool{}
	onNegotiationNeededCalledCount := 0
	var onNegotiationNeededCalledCountMu sync.Mutex
	ops := newOperations(updateNegotiationNeededFlagOnEmptyChain, func() {
		onNegotiationNeededCalledCountMu.Lock()
		onNegotiationNeededCalledCount++
		onNegotiationNeededCalledCountMu.Unlock()
	})
	defer ops.GracefulClose()

	for resultSet := 0; resultSet < 100; resultSet++ {
		results := make([]int, 16)
		resultSetCopy := resultSet
		for i := range results {
			func(j int) {
				ops.Enqueue(func() {
					results[j] = j * j
					if resultSetCopy > 50 {
						updateNegotiationNeededFlagOnEmptyChain.Store(true)
					}
				})
			}(i)
		}

		ops.Done()
		expected := []int{0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100, 121, 144, 169, 196, 225}
		assert.Equal(t, len(expected), len(results))
		assert.Equal(t, expected, results)
	}
	onNegotiationNeededCalledCountMu.Lock()
	defer onNegotiationNeededCalledCountMu.Unlock()
	assert.NotEqual(t, onNegotiationNeededCalledCount, 0)
}

func TestOperations_Done(*testing.T) {
	ops := newOperations(&atomic.Bool{}, func() {
	})
	defer ops.GracefulClose()
	ops.Done()
}

func TestOperations_GracefulClose(t *testing.T) {
	ops := newOperations(&atomic.Bool{}, func() {
	})

	counter := 0
	var counterMu sync.Mutex
	incFunc := func() {
		counterMu.Lock()
		counter++
		counterMu.Unlock()
	}
	const times = 25
	for i := 0; i < times; i++ {
		ops.Enqueue(incFunc)
	}
	ops.Done()
	counterMu.Lock()
	counterCur := counter
	counterMu.Unlock()
	assert.Equal(t, counterCur, times)

	ops.GracefulClose()
	for i := 0; i < times; i++ {
		ops.Enqueue(incFunc)
	}
	ops.Done()
	assert.Equal(t, counterCur, times)
}