github.com/dubbogo/gost@v1.14.0/container/chan/unbounded_chan_test.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 gxchan

import (
	"sync"
	"testing"
	"time"
)

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

func TestUnboundedChan(t *testing.T) {
	ch := NewUnboundedChan(300)

	var count int

	for i := 1; i < 200; i++ {
		ch.In() <- i
	}

	for i := 1; i < 60; i++ {
		v, _ := <-ch.Out()
		count += v.(int)
	}

	assert.Equal(t, 100, ch.queue.Cap())

	for i := 200; i <= 1200; i++ {
		ch.In() <- i
	}
	assert.Equal(t, 1600, ch.queue.Cap())

	wg := sync.WaitGroup{}
	wg.Add(1)
	go func() {
		defer wg.Done()
		var icount int
		for v := range ch.Out() {
			count += v.(int)
			icount++
			if icount == 900 {
				break
			}
		}
	}()

	wg.Wait()

	close(ch.In())

	// buffer should be empty
	wg.Add(1)
	go func() {
		defer wg.Done()
		for v := range ch.Out() {
			count += v.(int)
		}
	}()

	wg.Wait()

	assert.Equal(t, 720600, count)

}

func TestUnboundedChan_Quota(t *testing.T) {
	t.Run("testQuota1", testQuota1)
	t.Run("testQuota2", testQuota2)
}

func testQuota1(t *testing.T) {
	ch := NewUnboundedChanWithQuota(10, 15)
	assert.Equal(t, 2, cap(ch.in))
	assert.Equal(t, 3, cap(ch.out))
	assert.Equal(t, 4, ch.queue.Cap())
	assert.Equal(t, 0, ch.Len())
	assert.Equal(t, 10, ch.Cap())

	var count int

	for i := 0; i < 10; i++ {
		ch.In() <- i
	}

	assert.True(t, 14 >= ch.Cap())
	assert.True(t, 10 >= ch.Len())

	for i := 0; i < 10; i++ {
		v, ok := <-ch.Out()
		assert.True(t, ok)
		count += v.(int)
	}

	assert.Equal(t, 45, count)

	for i := 0; i < 15; i++ {
		ch.In() <- i
	}

	wg := sync.WaitGroup{}
	wg.Add(1)
	go func() {
		defer wg.Done()
		ch.In() <- 15
	}()

	for i := 0; i < 16; i++ {
		v, ok := <-ch.Out()
		assert.True(t, ok)
		count += v.(int)
	}

	wg.Wait()

	assert.Equal(t, 165, count)
}

// testQuota2 tests `ch.in` has no space
func testQuota2(t *testing.T) {
	ch := NewUnboundedChanWithQuota(1, 1)

	for i := 0; i < 1; i++ {
		ch.In() <- i
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 1:
	default:
		assert.Fail(t, "the chan shouldn't be blocked")
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 2:
		assert.Fail(t, "the chan should be blocked")
	default:
	}

	ch = NewUnboundedChanWithQuota(1, 0)

	for i := 0; i < 2; i++ {
		ch.In() <- i
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 2:
		assert.True(t, ch.Len() <= 3)
	default:
		assert.Fail(t, "the chan shouldn't be blocked")
	}

	ch = NewUnboundedChanWithQuota(1, 2)

	for i := 0; i < 1; i++ {
		ch.In() <- i
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 1:
	default:
		assert.Fail(t, "the chan shouldn't be blocked")
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 1:
		assert.Fail(t, "the chan should be blocked")
	default:
	}

	ch = NewUnboundedChanWithQuota(1, 3)

	for i := 0; i < 2; i++ {
		ch.In() <- i
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 2:
	default:
		assert.Fail(t, "the chan shouldn't be blocked")
	}

	time.Sleep(10 * time.Millisecond)

	select {
	case ch.In() <- 2:
		assert.Fail(t, "the chan should be blocked")
	default:
	}
}

func BenchmarkUnboundedChan_Fixed(b *testing.B) {
	ch := NewUnboundedChanWithQuota(1000, 1000)

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			select {
			case ch.In() <- 1:
			}

			<-ch.Out()
		}
	})

	close(ch.In())
}

func BenchmarkUnboundedChan_Extension(b *testing.B) {
	ch := NewUnboundedChanWithQuota(1000, 100000)

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			select {
			case ch.In() <- 1:
			}

			<-ch.Out()
		}
	})

	close(ch.In())
}

func BenchmarkUnboundedChan_ExtensionUnlimited(b *testing.B) {
	ch := NewUnboundedChanWithQuota(1000, 0)

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			select {
			case ch.In() <- 1:
			}

			<-ch.Out()
		}
	})

	close(ch.In())
}