package common

import (
	"testing"
)

func isClosed(ch chan struct{}) bool {
	select {
	case <-ch:
		return true
	default:
	}
	return false
}

func TestWaitGroupEmpty(t *testing.T) {

	wg := NewWaitGroup()

	if !wg.AddSession(0) {
		t.Fatalf("Add 0 should not fail")
	}

	if isClosed(wg.Closer()) {
		t.Fatalf("Should not be closed yet")
	}

	done := wg.CloseGroupNB()

	// gate-on close
	wg.CloseGroup()

	if !isClosed(wg.Closer()) {
		t.Fatalf("Should be closing state")
	}

	if isClosed(done) {
		t.Fatalf("NB Chan I should be closed")
	}

	done = wg.CloseGroupNB()
	if isClosed(done) {
		t.Fatalf("NB Chan II should be closed")
	}
}

func TestWaitGroupSingle(t *testing.T) {

	wg := NewWaitGroup()

	if isClosed(wg.Closer()) {
		t.Fatalf("Should not be closing state yet")
	}

	if !wg.AddSession(1) {
		t.Fatalf("Add 1 should not fail")
	}

	if isClosed(wg.Closer()) {
		t.Fatalf("Should not be closing state yet")
	}

	if !wg.AddSession(-1) {
		t.Fatalf("Add -1 should not fail")
	}

	// sum should be zero now.

	if !wg.AddSession(2) {
		t.Fatalf("Add 2 should not fail")
	}

	// sum is 2 now
	// initiate shutdown
	done := wg.CloseGroupNB()

	if isClosed(done) {
		t.Fatalf("NB Chan should not be closed yet, since sum is 2")
	}

	if !wg.AddSession(-1) {
		t.Fatalf("Add -1 should not fail")
	}
	if wg.AddSession(1) {
		t.Fatalf("Add 1 should fail (we are shutting down)")
	}
	if !isClosed(wg.Closer()) {
		t.Fatalf("Should be closing state")
	}

	// sum is 1 now

	if isClosed(done) {
		t.Fatalf("NB Chan should not be closed yet, since sum is 1")
	}

	if wg.AddSession(0) {
		t.Fatalf("Add 0 should fail (considered positive number and we are closing)")
	}

	if wg.AddSession(100) {
		t.Fatalf("Add 100 should fail (we are shutting down)")
	}

	if !isClosed(wg.Closer()) {
		t.Fatalf("Should be closing state")
	}

	if !wg.AddSession(-1) {
		t.Fatalf("Add -1 should not fail")
	}

	// sum is 0 now
	<-done

	if !isClosed(done) {
		t.Fatalf("NB Chan should be closed, since sum is 0")
	}

	if !isClosed(wg.Closer()) {
		t.Fatalf("Should be closing state")
	}

}