Function TestConcurrentQueue

10-concurrent-queue/task_test.go:82–131

(t *testing.T)

}

func TestConcurrentQueue(t *testing.T) {
	queue := NewQueue(1000)
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	var wg sync.WaitGroup
	wg.Add(1000)
	for i := range 1000 {
		go func() {
			defer wg.Done()
			queue.Push(i + 1)
		}()
	}

	ch := make(chan int, 1000)
	go func() {
		defer close(ch)
		for {
			select {
			case <-ctx.Done():
				return
			default:
				if val := queue.Pop(); val != -1 {
					ch <- val
				}
			}
		}
	}()

	go func() {
		wg.Wait()
		time.Sleep(100 * time.Millisecond)
		cancel()
	}()

	var result []int
	for val := range ch {
		result = append(result, val)
	}

	expected := make([]int, 1000)
	for i := range 1000 {
		expected[i] = i + 1
	}

	sort.Ints(result)
	if !reflect.DeepEqual(result, expected) {
		t.Errorf("Expected: 1..1000, got: %v", result)
	}
}