cloudpods/pkg/appsrv/radix_test.go

// Copyright 2019 Yunion
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package appsrv

import (
	"strings"
	"testing"
)

func TestRadixNode(t *testing.T) {
	r := NewRadix()
	r.Add([]string{"layer1"}, "layer1")
	r.Add([]string{"layer1", "layer1.1", "layer1.1.1", "layer1.1.1.1"}, "layer1.1.1.1")
	r.Add([]string{"layer1", "layer1.2", "layer1.2.1"}, "layer1.2.1")
	r.Add([]string{"layer1", "<layer1.x>"}, "layer1.*")
	r.Add([]string{"layer1", "layer1.0"}, "layer1.0")
	r.Add([]string{"layer1", "<layer1.x>", "layer1.*.1"}, "layer1.*.1")

	r.Add([]string{"layer1", "<phone_number:^1[0-9-]{10}$>", "layer1.2.1"}, "layer1.*.1_CHINA_MOBILE_REG")
	r.Add([]string{"layer1", "layer1.0", "<phone_number:^1[0-9-]{10}$>"}, "layer1.1.*_CHINA_MOBILE_REG")

	var ret interface{}
	f := func(path string, data interface{}) {
		t.Logf("%s %s", path, data)
	}
	r.Walk(f)
	params := make(map[string]string)

	ret = r.Match([]string{"layer1", "layer1.0", "layer2.0"}, params)
	if ret.(string) != "layer1.0" {
		t.Error("0 Unexpect result:", ret, "!= layer1.0")
	}
	ret = r.Match([]string{"layer1", "layer1.0"}, params)
	if ret.(string) != "layer1.0" {
		t.Error("0 Unexpect result:", ret, "!= layer1.0")
	}
	ret = r.Match([]string{"layer1", "layer1.1"}, params)
	if ret.(string) != "layer1.*" {
		t.Error("1 Unexpect result:", ret, "!= layer1.*")
	}
	ret = r.Match([]string{"layer1", "layer1.4"}, params)
	if ret.(string) != "layer1.*" {
		t.Error("2 Unexpect result:", ret, "!= layer1.*")
	}
	ret = r.Match([]string{"layer1", "layer1.1", "layer1.1.1", "layer1.1.1.1", "layer1.1.1.1.1", "layer1.1.1.1.1.1"}, params)
	if ret.(string) != "layer1.1.1.1" {
		t.Error("3 Unexpect result:", ret, "!= layer1.1.1.1")
	}
	ret = r.Match([]string{"layer1", "layer1.2", "layer1.2.1"}, params)
	if ret.(string) != "layer1.2.1" {
		t.Error("4 Unexpect result:", ret, "!= layer1.2.1")
	}
	ret = r.Match([]string{"layer1", "layer1.2"}, params)
	if ret.(string) != "layer1.*" {
		t.Error("5 Unexpect result:", ret, "!= layer1.*")
	}
	ret = r.Match([]string{"layer1", "layer1.3"}, params)
	if ret.(string) != "layer1.*" {
		t.Error("6 Unexpect result:", ret, "!= layer1.*")
	}
	ret = r.Match([]string{"layer1", "layer1.3", "layer1.*.1"}, params)
	if ret.(string) != "layer1.*.1" {
		t.Error("7 Unexpect result:", ret, "!= layer1.*.1")
	}
	ret = r.Match([]string{"layer1", "layer1.3", "layer1.*.1", "layer1.*.1.1"}, params)
	if ret.(string) != "layer1.*.1" {
		t.Error("8 Unexpect result:", ret, "!= layer1.*.1")
	}
	ret = r.Match([]string{"layer1", "12345678901", "layer1.2.1"}, params)
	if ret.(string) != "layer1.*.1_CHINA_MOBILE_REG" {
		t.Error("11 Unexpect result:", ret, "!= layer1.*.1_CHINA_MOBILE_REG")
	}
	ret = r.Match([]string{"layer1", "layer1.0", "12345678901"}, params)
	if ret.(string) != "layer1.1.*_CHINA_MOBILE_REG" {
		t.Error("12 Unexpect result:", ret, "!= layer1.1.*_CHINA_MOBILE_REG")
	}

	r.Add([]string{}, "root")
	ret = r.Match([]string{"layer2"}, params)
	if ret.(string) != "root" {
		t.Error("9 Unexpect result:", ret, "!= root")
	}

	r.Add([]string{"<phone_number:^1[0-9-]{10}$>"}, "phonelayer1")
	ret = r.Match([]string{"12345678900"}, params)
	if ret.(string) != "phonelayer1" {
		t.Error("13 Unexpect result:", ret, "!= phonelayer1")
	}

	if r.Add([]string{"layer1"}, "layer1") == nil {
		t.Error("10 Add duplicate data should fail")
	}
}

func TestRadixMatchParams(t *testing.T) {
	r := NewRadix()
	r.Add([]string{"POST", "clouds", "<cls_action>"}, "classAction")
	r.Add([]string{"POST", "clouds", "<resid>", "sync"}, "objectSyncAction")
	r.Add([]string{"POST", "clouds", "<resid>", "<obj_action>"}, "objectAction")
	r.Add([]string{"POST", "clouds", "<resid2:.*>", "<obj_action2:.*>", "over"}, "objectAction2")
	f := func(path string, data interface{}) {
		t.Logf("%s %s", path, data)
	}
	r.Walk(f)

	cases := []struct {
		in        []string
		out       interface{}
		outParams map[string]string
	}{
		{
			in:  []string{"POST", "clouds", "myid", "sync"},
			out: "objectSyncAction",
			outParams: map[string]string{
				"<resid>": "myid",
			},
		},
		{
			in:  []string{"POST", "clouds", "myid", "start"},
			out: "objectAction",
			outParams: map[string]string{
				"<resid>":      "myid",
				"<obj_action>": "start",
			},
		},
		{
			in:  []string{"POST", "clouds", "start"},
			out: "classAction",
			outParams: map[string]string{
				"<cls_action>": "start",
			},
		},
		{
			in:  []string{"POST", "clouds", "start", "test", "over"},
			out: "objectAction2",
			outParams: map[string]string{
				"<resid2>":      "start",
				"<obj_action2>": "test",
			},
		},
	}
	for _, c := range cases {
		t.Run(strings.Join(c.in, "_"), func(t *testing.T) {
			gotParams := map[string]string{}
			got := r.Match(c.in, gotParams)
			if got != c.out {
				t.Fatalf("want %s, got %s", c.out, c.out)
			}
			if len(gotParams) != len(c.outParams) {
				t.Fatalf("params length mismatch\nwant %#v\ngot %#v",
					c.outParams, gotParams,
				)
			}
			for k, v := range gotParams {
				v1 := c.outParams[k]
				if v == v1 {
					continue
				}
				t.Fatalf("params key %s has mismatch value\nwant %#v\ngot %#v",
					k, c.outParams, gotParams,
				)
			}
		})
	}
}