v2ray-core/app/policy/manager_test.go

package policy_test

import (
	"context"
	"testing"
	"time"

	"v2ray.com/core"
	. "v2ray.com/core/app/policy"
	. "v2ray.com/ext/assert"
)

func TestPolicy(t *testing.T) {
	assert := With(t)

	manager, err := New(context.Background(), &Config{
		Level: map[uint32]*Policy{
			0: &Policy{
				Timeout: &Policy_Timeout{
					Handshake: &Second{
						Value: 2,
					},
				},
			},
		},
	})
	assert(err, IsNil)

	pDefault := core.DefaultPolicy()

	p0 := manager.ForLevel(0)
	assert(p0.Timeouts.Handshake, Equals, 2*time.Second)
	assert(p0.Timeouts.ConnectionIdle, Equals, pDefault.Timeouts.ConnectionIdle)

	p1 := manager.ForLevel(1)
	assert(p1.Timeouts.Handshake, Equals, pDefault.Timeouts.Handshake)
}