Function TestCallback

znet/callbacks_test.go:7–101

(t *testing.T)

)

func TestCallback(t *testing.T) {
	// Test empty list
	cb := &callbacks{}
	if cb.Len() != 0 {
		t.Errorf("Expected count for empty list is 0, but got %d", cb.Len())
	}

	// Ensure invoking on an empty registry is a no-op (no panic).
	cb.Invoke()

	// Test adding callback functions
	var count, expected, remove, totalCount int
	totalCount = 10
	remove = 5

	// Add multiple callback functions
	for i := 1; i < totalCount; i++ {
		expected = expected + i
		func(ii int) {
			cb.Add(ii, ii, func() { count = count + ii })
		}(i)
	}

	// Verify count after adding
	expectedCallbacks := totalCount - 1
	if cb.Len() != expectedCallbacks {
		t.Errorf("Expected callback count is %d, but got %d", expectedCallbacks, cb.Len())
	}

	// Test adding nil callback
	cb.Add(remove, remove, nil)
	if cb.Len() != expectedCallbacks {
		t.Errorf("Expected count after adding nil callback is %d, but got %d", expectedCallbacks, cb.Len())
	}

	// Replace an existing callback with a non-nil one; count should remain unchanged.
	cb.Add(remove, remove, func() { count += remove })
	if cb.Len() != expectedCallbacks {
		t.Errorf("Expected count after replacing existing callback is %d, but got %d", expectedCallbacks, cb.Len())
	}

	// Remove specified callback
	cb.Remove(remove, remove)

	// Try to remove non-existent callback
	cb.Remove(remove+1, remove+2)

	// Execute all callbacks
	cb.Invoke()

	// Verify execution result
	expectedSum := expected - remove
	if count != expectedSum {
		t.Errorf("Expected execution result is %d, but got %d", expectedSum, count)
	}

	// Test string type handler and key
	cb2 := &callbacks{}