concurrent_test.go

package xerror

import (
	"sync"
	"testing"

	"github.com/gomooth/xerror/xcode"
)

func TestConcurrent_Register(t *testing.T) {
	var wg sync.WaitGroup
	for i := 0; i < 100; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			xcode.Register(xcode.NewWithHTTPStatus(20000+i, 400, "业务错误"))
		}(i)
	}
	wg.Wait()

	for i := 0; i < 100; i++ {
		code := xcode.NewFromRegistry(20000+i, "test")
		if code.HttpStatus() != 400 {
			t.Fatalf("code %d: expected httpStatus 400, got %d", 20000+i, code.HttpStatus())
		}
	}
	xcode.Reset()
}

func TestConcurrent_Override(t *testing.T) {
	var wg sync.WaitGroup
	for i := 0; i < 100; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			xcode.Override(500, 400+i%10, "覆写500")
		}(i)
	}
	wg.Wait()

	code := xcode.NewFromRegistry(500, "test")
	if code.HttpStatus() < 400 || code.HttpStatus() > 409 {
		t.Fatalf("unexpected httpStatus %d", code.HttpStatus())
	}
	xcode.Reset()
}

func TestConcurrent_Reset(t *testing.T) {
	var wg sync.WaitGroup
	for i := 0; i < 50; i++ {
		wg.Add(2)
		go func(i int) {
			defer wg.Done()
			xcode.Register(xcode.NewWithHTTPStatus(30000+i, 400, "注册"))
		}(i)
		go func() {
			defer wg.Done()
			xcode.Reset()
		}()
	}
	wg.Wait()
	xcode.Reset()
}

func TestConcurrent_Wrap(t *testing.T) {
	err := New("base error")
	var wg sync.WaitGroup
	results := make([]XError, 100)
	for i := 0; i < 100; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			results[i] = Wrap(err, "wrapped")
		}(i)
	}
	wg.Wait()

	for i, r := range results {
		if r.Message() != "wrapped" {
			t.Fatalf("result[%d]: expected 'wrapped', got %q", i, r.Message())
		}
	}
}

func TestConcurrent_WithFields(t *testing.T) {
	err := New("base error")
	var wg sync.WaitGroup
	results := make([]XError, 100)
	for i := 0; i < 100; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			results[i] = err.WithFields(F("key", i))
		}(i)
	}
	wg.Wait()

	for i, r := range results {
		fields := r.GetFields()
		if len(fields) != 1 {
			t.Fatalf("result[%d]: expected 1 field, got %d", i, len(fields))
		}
		if fields[0].Key != "key" {
			t.Fatalf("result[%d]: expected key 'key', got %q", i, fields[0].Key)
		}
	}
}