shutdown_test.go RawHistoryFolder

1package webterm
2
3import (
4	"context"
5	"strings"
6	"sync"
7	"testing"
8
9	"github.com/stretchr/testify/require"
10)
11
12func TestShutdownAdmissionRace(t *testing.T) {
13	for range 100 {
14		server, err := NewServer(t.Context(), Settings{}, strings.Repeat("d", 43), func(context.Context, *Terminal) error { return nil })
15		require.NoError(t, err)
16		var racers sync.WaitGroup
17		start := make(chan struct{})
18		admitted := make(chan bool, 1)
19		release := make(chan struct{})
20		drained := make(chan struct{})
21		racers.Add(2)
22		go func() {
23			defer racers.Done()
24			<-start
25			ok := server.admit()
26			admitted <- ok
27			if ok {
28				<-release
29				server.workers.Done()
30			}
31		}()
32		go func() { defer racers.Done(); <-start; server.Wait(); close(drained) }()
33		close(start)
34		if <-admitted {
35			select {
36			case <-drained:
37				t.Fatal("shutdown missed an admitted worker")
38			default:
39			}
40		}
41		close(release)
42		racers.Wait()
43		require.False(t, server.admit(), "shutdown must permanently close admission")
44	}
45}
46
47func TestCanceledServerRejectsAdmission(t *testing.T) {
48	ctx, cancel := context.WithCancel(t.Context())
49	server, err := NewServer(ctx, Settings{}, strings.Repeat("e", 43), func(context.Context, *Terminal) error { return nil })
50	require.NoError(t, err)
51	cancel()
52	require.False(t, server.admit())
53	server.Wait()
54}