messaging_test.go RawHistoryFolder

1package webkit
2
3import (
4	"context"
5	"sync"
6	"testing"
7
8	"github.com/stretchr/testify/require"
9)
10
11type messageRouterContextKey struct{}
12
13func TestMessageRouterBaseContextConcurrentUpdate(t *testing.T) {
14	t.Parallel()
15
16	router := NewMessageRouter(context.Background())
17	var wg sync.WaitGroup
18	wg.Add(2)
19
20	go func() {
21		defer wg.Done()
22		for i := range 1000 {
23			router.SetBaseContext(context.WithValue(context.Background(), messageRouterContextKey{}, i))
24		}
25	}()
26	go func() {
27		defer wg.Done()
28		for range 1000 {
29			if router.baseContext() == nil {
30				t.Error("base context must never be nil")
31			}
32		}
33	}()
34
35	wg.Wait()
36}
37
38func TestIsTrustedBridgeURI(t *testing.T) {
39	t.Parallel()
40
41	trusted := []string{
42		"dumb://history",
43		"dumb://favorites",
44		"dumb://config",
45		"dumb://error",
46		"dumb://crash",
47		"dumb://history/path?cursor=1",
48	}
49	for _, raw := range trusted {
50		t.Run(raw, func(t *testing.T) {
51			t.Parallel()
52			require.True(t, isTrustedBridgeURI(raw))
53		})
54	}
55
56	untrusted := []string{
57		"",
58		"https://example.com",
59		"http://localhost",
60		"file:///tmp/history.html",
61		"dumb://evil/history",
62		"dumb://example.com",
63		"javascript:alert(1)",
64	}
65	for _, raw := range untrusted {
66		t.Run(raw, func(t *testing.T) {
67			t.Parallel()
68			require.False(t, isTrustedBridgeURI(raw))
69		})
70	}
71}