webview_bridge_nonce_test.go RawHistoryFolder

1package cef
2
3import (
4	"errors"
5	"testing"
6
7	"github.com/stretchr/testify/require"
8)
9
10func TestNewBridgeNonce_ReturnsEmptyOnRandomFailure(t *testing.T) {
11	oldBridgeNonceRandom := bridgeNonceRandom
12	bridgeNonceRandom = func([]byte) (int, error) {
13		return 0, errors.New("entropy unavailable")
14	}
15	defer func() {
16		bridgeNonceRandom = oldBridgeNonceRandom
17	}()
18
19	require.Empty(t, newBridgeNonce())
20}
21
22func TestWebViewRotateBridgeNonce_PreservesExistingNonceOnRandomFailure(t *testing.T) {
23	oldBridgeNonceRandom := bridgeNonceRandom
24	bridgeNonceRandom = func([]byte) (int, error) {
25		return 0, errors.New("entropy unavailable")
26	}
27	defer func() {
28		bridgeNonceRandom = oldBridgeNonceRandom
29	}()
30
31	wv := &WebView{bridgeNonce: "existing-nonce"}
32	require.Empty(t, wv.rotateBridgeNonce())
33
34	wv.mu.RLock()
35	defer wv.mu.RUnlock()
36	require.Equal(t, "existing-nonce", wv.bridgeNonce)
37}
38
39func TestWebViewRotateBridgeNonce_ReplacesNonceOnSuccess(t *testing.T) {
40	oldBridgeNonceRandom := bridgeNonceRandom
41	bridgeNonceRandom = func(buf []byte) (int, error) {
42		for i := range buf {
43			buf[i] = byte(i + 1)
44		}
45		return len(buf), nil
46	}
47	defer func() {
48		bridgeNonceRandom = oldBridgeNonceRandom
49	}()
50
51	wv := &WebView{bridgeNonce: "old-nonce"}
52	require.Equal(t, "0102030405060708090a0b0c0d0e0f10", wv.rotateBridgeNonce())
53
54	wv.mu.RLock()
55	defer wv.mu.RUnlock()
56	require.Equal(t, "0102030405060708090a0b0c0d0e0f10", wv.bridgeNonce)
57}
58
59func TestWebViewEnsureBridgeNonce_ReusesExistingNonce(t *testing.T) {
60	wv := &WebView{bridgeNonce: "existing-nonce"}
61	require.Equal(t, "existing-nonce", wv.ensureBridgeNonce())
62}