Refactor api

tun_test.go

@@ -0,0 +1,70 @@
+package tun
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/sagernet/sing/common/ranges"
+)
+
+func TestUIDRanges(t *testing.T) {
+	for _, testRange := range []struct {
+		include     []ranges.Range[uint32]
+		exclude     []ranges.Range[uint32]
+		androidUser []int
+		expected    []ranges.Range[uint32]
+	}{
+		{},
+		{
+			include: []ranges.Range[uint32]{
+				ranges.NewSingle[uint32](0),
+				ranges.NewSingle[uint32](1000),
+			},
+			expected: []ranges.Range[uint32]{
+				{Start: 1, End: 999},
+				{Start: 1001, End: userEnd},
+			},
+		},
+		{
+			androidUser: []int{0},
+			expected: []ranges.Range[uint32]{
+				{Start: androidUserRange, End: userEnd},
+			},
+		},
+		{
+			androidUser: []int{0},
+			expected: []ranges.Range[uint32]{
+				{Start: 100000, End: userEnd},
+			},
+		},
+		{
+			androidUser: []int{10},
+			include: []ranges.Range[uint32]{
+				ranges.NewSingle[uint32](0),
+			},
+			expected: []ranges.Range[uint32]{
+				{Start: 1, End: androidUserRange*10 - 1},
+				{Start: androidUserRange * 11, End: userEnd},
+			},
+		},
+		{
+			include: []ranges.Range[uint32]{
+				{Start: 123456, End: 123456},
+			},
+			exclude: []ranges.Range[uint32]{
+				{Start: 1000, End: 1000},
+			},
+			androidUser: []int{0},
+			expected: []ranges.Range[uint32]{
+				{Start: 1000, End: 1000},
+				{Start: 100000, End: 123455},
+				{Start: 123457, End: userEnd},
+			},
+		},
+	} {
+		result := buildExcludedRanges(testRange.include, testRange.exclude, testRange.androidUser)
+		if !reflect.DeepEqual(result, testRange.expected) {
+			t.Fatal("input", testRange.include, testRange.exclude, testRange.androidUser, "\nexpected", testRange.expected, "\ngot", result)
+		}
+	}
+}