改进：增加schedpolicy和dynamicschedtag支持

2026-09-24 16:03:43 +08:00 · 2018-10-17 08:12:21 +08:00
parent 219015d50c
commit fc777095c1
14 changed files with 1428 additions and 10 deletions
@@ -0,0 +1,147 @@
+package conditionparser
+
+import (
+	"testing"
+
+	"go/parser"
+	"yunion.io/x/jsonutils"
+)
+
+func TestAst(t *testing.T) {
+	input := jsonutils.NewDict()
+	input.Add(jsonutils.NewString("windows"), "server", "os_type")
+	disk := jsonutils.NewDict()
+	disk.Add(jsonutils.NewString("ssd"), "medium_type")
+	disks := jsonutils.NewArray(disk)
+	input.Add(disks, "server", "disks")
+	input.Add(disk, "server", "disk.0")
+
+	cases := []struct {
+		in   string
+		want bool
+	}{
+		{`server.os_type == "windows"`, true},
+		{`server.os_type.startswith("window")`, true},
+		{`server.disks[0].medium_type == "ssd"`, true},
+		{`server.disks[0].medium_type == "hdd"`, false},
+		{`server.os_type == "windows" && server.disks[0].medium_type == "ssd"`, true},
+	}
+
+	for _, c := range cases {
+		result, err := Eval(c.in, input)
+		if err != nil {
+			t.Errorf("eval expr %s error %s", c.in, err)
+			return
+		}
+
+		if result != c.want {
+			t.Errorf("expect %v got %v", c.want, result)
+			return
+		}
+	}
+
+}
+
+func TestIsValid(t *testing.T) {
+	cases := []struct {
+		in   string
+		want bool
+	}{
+		{`server.os_type == "windows"`, true},
+		{`server.os_type ==`, false},
+		{`dfadsfsdf ==`, false},
+		{`dfadsfsdf +`, false},
+	}
+	for _, c := range cases {
+		got := IsValid(c.in)
+		if got != c.want {
+			t.Errorf("%s is valid %v got %v", c.in, c.want, got)
+		}
+	}
+}
+
+func TestEval2(t *testing.T) {
+	inputStr := `{"server": {"disable_delete":false,
+"disk.0":{"backend":"local","format":"qcow2","image_id":"4b9fa54c-858c-4c2b-8719-27aee120b3cb","image_properties":{"os_arch":"x86_64","os_distribution":"CentOS","os_type":"Linux","os_version":"7.5.1804"},"medium":"hybrid","size":40960},
+"hypervisor":"kvm","keypair_id":"None","name":"testsched","os_type":"Linux","owner_tenant_id":"5d65667d112e47249ae66dbd7bc07030",
+"sched_tag.0":"ssd","secgrp_id":"default","vcpu_count":1,"vmem_size":1024}}`
+	input, err := jsonutils.ParseString(inputStr)
+	if err != nil {
+		t.Errorf("fail to parse server json")
+		return
+	}
+
+	cases := []struct {
+		in   string
+		want bool
+	}{
+		{`server.os_type == "Linux"`, true},
+		{`server.vmem_size > 2048`, false},
+		{`server.hypervisor.in("kvm", "aliyun")`, true},
+		{`server.disable_delete`, false},
+		{`server.disk[0].backend == "local"`, true},
+		{`server.sched_tag[0] != "ssd"`, false},
+		{`server.sched_tag[0] == "ssd"`, true},
+		// {`server.os_type == "windows" && server.disks[0].medium_type == "ssd"`, true},
+	}
+
+	for _, c := range cases {
+		result, err := Eval(c.in, input)
+		if err != nil {
+			t.Errorf("eval expr %s error %s", c.in, err)
+			return
+		}
+
+		if result != c.want {
+			t.Errorf("expect %v got %v", c.want, result)
+			return
+		}
+	}
+}
+
+func TestEval3(t *testing.T) {
+	exprStr := `server.disk.backend == "local"`
+	expr, err := parser.ParseExpr(exprStr)
+	if err != nil {
+		t.Errorf("parse exprStr fail %s %s", exprStr, err)
+		return
+	}
+
+	t.Logf("%s", jsonutils.Marshal(expr))
+
+	inputStr := `{"server":{
+	"disk.0":{"backend": "local", "medium": "hdd"},
+	"disk.1":{"backend": "local", "medium": "hdd"},
+	"disk.2":{"backend": "rbd", "medium": "ssd"},
+	"disk.3":{"backend": "rbd", "medium": "ssd"}}
+}`
+	input, err := jsonutils.ParseString(inputStr)
+	if err != nil {
+		t.Errorf("fail to parse server json %s", err)
+		return
+	}
+
+	cases := []struct {
+		in   string
+		want bool
+	}{
+		{`server.disk.backend == "local"`, true},
+		{`server.disk.medium == "ssd"`, true},
+		{`server.disk.medium == "hdd"`, true},
+		{`server.disk.medium == "hybrid"`, false},
+		{`server.disk.medium.contains("ssd")`, true},
+	}
+
+	for _, c := range cases {
+		result, err := Eval(c.in, input)
+		if err != nil {
+			t.Errorf("eval expr %s error %s", c.in, err)
+			return
+		}
+
+		if result != c.want {
+			t.Errorf("expect %v got %v", c.want, result)
+			return
+		}
+	}
+}