mirror of
https://github.com/php/frankenphp.git
synced 2026-03-24 00:52:11 +01:00
98573ed7c0
This PR: - moves state.go to its own module - moves the phpheaders test the phpheaders module - simplifies backoff.go - makes the backoff error instead of panic (so it can be tested) - removes some unused C structs
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package state
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test2GoroutinesYieldToEachOtherViaStates(t *testing.T) {
|
|
threadState := &ThreadState{currentState: Booting}
|
|
|
|
go func() {
|
|
threadState.WaitFor(Inactive)
|
|
assert.True(t, threadState.Is(Inactive))
|
|
threadState.Set(Ready)
|
|
}()
|
|
|
|
threadState.Set(Inactive)
|
|
threadState.WaitFor(Ready)
|
|
assert.True(t, threadState.Is(Ready))
|
|
}
|
|
|
|
func TestStateShouldHaveCorrectAmountOfSubscribers(t *testing.T) {
|
|
threadState := &ThreadState{currentState: Booting}
|
|
|
|
// 3 subscribers waiting for different states
|
|
go threadState.WaitFor(Inactive)
|
|
go threadState.WaitFor(Inactive, ShuttingDown)
|
|
go threadState.WaitFor(ShuttingDown)
|
|
|
|
assertNumberOfSubscribers(t, threadState, 3)
|
|
|
|
threadState.Set(Inactive)
|
|
assertNumberOfSubscribers(t, threadState, 1)
|
|
|
|
assert.True(t, threadState.CompareAndSwap(Inactive, ShuttingDown))
|
|
assertNumberOfSubscribers(t, threadState, 0)
|
|
}
|
|
|
|
func assertNumberOfSubscribers(t *testing.T, threadState *ThreadState, expected int) {
|
|
t.Helper()
|
|
for range 10_000 { // wait for 1 second max
|
|
time.Sleep(100 * time.Microsecond)
|
|
threadState.mu.RLock()
|
|
if len(threadState.subscribers) == expected {
|
|
threadState.mu.RUnlock()
|
|
break
|
|
}
|
|
threadState.mu.RUnlock()
|
|
}
|
|
threadState.mu.RLock()
|
|
assert.Len(t, threadState.subscribers, expected)
|
|
threadState.mu.RUnlock()
|
|
}
|