scenario_test: add test cases for distribute policy

1 files changed, 99 insertions, 21 deletions

diff --git a/test/scenario_test/policy/policy_generator.go b/test/scenario_test/policy/policy_generator.go
index f72eb5c8..a4f70d0e 100644
--- a/test/scenario_test/policy/policy_generator.go
+++ b/test/scenario_test/policy/policy_generator.go
@@ -12,8 +12,7 @@ import (
 	"os"
 )
 
-
-func bindPolicy(outputDir, peer, target, policyName string, isReplace bool) {
+func bindPolicy(outputDir, peer, target, policyName string, isReplace bool, defaultReject bool) {
 
 	newConf := config.Bgp{}
 	_, d_err := toml.DecodeFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), &newConf)
@@ -33,28 +32,37 @@ func bindPolicy(outputDir, peer, target, policyName string, isReplace bool) {
 				} else {
 					ap.ImportPolicies = append(ap.ImportPolicies, policyName)
 				}
+				if defaultReject {
+					ap.DefaultImportPolicy = 1
+				}
 			case "export":
 				if isReplace {
 					ap.ExportPolicies = []string{policyName}
 				} else {
 					ap.ExportPolicies = append(ap.ExportPolicies, policyName)
 				}
+				if defaultReject {
+					ap.DefaultExportPolicy = 1
+				}
 			case "distribute":
 				if isReplace {
 					ap.DistributePolicies = []string{policyName}
 				} else {
 					ap.DistributePolicies = append(ap.DistributePolicies, policyName)
 				}
+				if defaultReject {
+					ap.DefaultDistributePolicy = 1
+				}
 			}
 			newConf.NeighborList[idx] = neighbor
 		}
 	}
 
 	policyConf := createPolicyConfig()
-    var buffer bytes.Buffer
-    encoder := toml.NewEncoder(&buffer)
-    encoder.Encode(newConf)
-    encoder.Encode(policyConf)
+	var buffer bytes.Buffer
+	encoder := toml.NewEncoder(&buffer)
+	encoder.Encode(newConf)
+	encoder.Encode(policyConf)
 
 	e_err := ioutil.WriteFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), buffer.Bytes(), 0644)
 	if e_err != nil {
@@ -126,6 +134,15 @@ func createPolicyConfig() *config.RoutingPolicy {
 			}},
 	}
 
+	ps6 := config.PrefixSet{
+		PrefixSetName: "ps6",
+		PrefixList: []config.Prefix{
+			config.Prefix{
+				Address:    net.ParseIP("192.168.10.0"),
+				Masklength: 24,
+			}},
+	}
+
 	nsPeer2 := config.NeighborSet{
 		NeighborSetName: "nsPeer2",
 		NeighborInfoList: []config.NeighborInfo{
@@ -456,6 +473,49 @@ func createPolicyConfig() *config.RoutingPolicy {
 		},
 	}
 
+	st_distribute_reject := config.Statement{
+		Name: "st_community_distriibute",
+		Conditions: config.Conditions{
+			BgpConditions: config.BgpConditions{
+				MatchCommunitySet: "comStr",
+			},
+			MatchSetOptions: config.MATCH_SET_OPTIONS_TYPE_ALL,
+		},
+		Actions: config.Actions{
+			AcceptRoute: false,
+		},
+	}
+
+	st_distribute_accept := config.Statement{
+		Name: "st_distriibute_accept",
+		Conditions: config.Conditions{
+			MatchPrefixSet:   "ps6",
+			MatchSetOptions: config.MATCH_SET_OPTIONS_TYPE_ALL,
+		},
+		Actions: config.Actions{
+			AcceptRoute: true,
+		},
+	}
+
+
+	st_distribute_comm_add := config.Statement{
+		Name: "st_distribute_comm_add",
+		Conditions: config.Conditions{
+			BgpConditions: config.BgpConditions{
+				MatchCommunitySet: "comStr",
+			},
+			MatchSetOptions: config.MATCH_SET_OPTIONS_TYPE_ALL,
+		},
+		Actions: config.Actions{
+			AcceptRoute: true,
+			BgpActions: config.BgpActions{
+				SetCommunity: config.SetCommunity{
+					Communities: []string{"65100:20"},
+					Options:     "ADD",
+				},
+			},
+		},
+	}
 
 	test_01_import_policy_initial := config.PolicyDefinition{
 		Name:          "test_01_import_policy_initial",
@@ -597,8 +657,23 @@ func createPolicyConfig() *config.RoutingPolicy {
 		StatementList: []config.Statement{st_comNull},
 	}
 
+	test_25_distribute_reject := config.PolicyDefinition{
+		Name:          "test_25_distribute_reject",
+		StatementList: []config.Statement{st_distribute_reject},
+	}
+
+	test_26_distribute_accept := config.PolicyDefinition{
+		Name:          "test_26_distribute_accept",
+		StatementList: []config.Statement{st_distribute_accept},
+	}
+
+	test_27_distribute_set_community_action := config.PolicyDefinition{
+		Name:          "test_27_distribute_set_community_action",
+		StatementList: []config.Statement{st_distribute_comm_add},
+	}
+
 	ds := config.DefinedSets{
-		PrefixSetList:   []config.PrefixSet{ps0, ps1, ps2, ps3, ps4, ps5, psExabgp},
+		PrefixSetList:   []config.PrefixSet{ps0, ps1, ps2, ps3, ps4, ps5, ps6, psExabgp},
 		NeighborSetList: []config.NeighborSet{nsPeer2, nsPeer2V6, nsExabgp},
 		BgpDefinedSets: config.BgpDefinedSets{
 			AsPathSetList:    []config.AsPathSet{aspathFrom, aspathAny, aspathOrigin, aspathOnly},
@@ -615,20 +690,20 @@ func createPolicyConfig() *config.RoutingPolicy {
 			test_03_import_policy_update_softreset,
 			test_04_export_policy_update,
 			test_04_export_policy_update_softreset,
-            test_05_import_policy_initial_ipv6,
+			test_05_import_policy_initial_ipv6,
 			test_06_export_policy_initial_ipv6,
-            test_07_import_policy_update,
+			test_07_import_policy_update,
 			test_07_import_policy_update_softreset,
-            test_08_export_policy_update,
+			test_08_export_policy_update,
 			test_08_export_policy_update_softreset,
-            test_09_aspath_length_condition_import,
-            test_10_aspath_from_condition_import,
+			test_09_aspath_length_condition_import,
+			test_10_aspath_from_condition_import,
 			test_11_aspath_any_condition_import,
-            test_12_aspath_origin_condition_import,
-            test_13_aspath_only_condition_import,
-            test_14_aspath_only_condition_import,
-            test_15_community_condition_import,
-            test_16_community_condition_regexp_import,
+			test_12_aspath_origin_condition_import,
+			test_13_aspath_only_condition_import,
+			test_14_aspath_only_condition_import,
+			test_15_community_condition_import,
+			test_16_community_condition_regexp_import,
 			test_17_community_add_action_import,
 			test_18_community_replace_action_import,
 			test_19_community_remove_action_import,
@@ -637,19 +712,22 @@ func createPolicyConfig() *config.RoutingPolicy {
 			test_22_community_replace_action_export,
 			test_23_community_remove_action_export,
 			test_24_community_null_action_export,
-        },
+			test_25_distribute_reject,
+			test_26_distribute_accept,
+			test_27_distribute_set_community_action,
+		},
 	}
 	return p
 }
 
-
 func main() {
 	var opts struct {
 		OutputDir  string `short:"d" long:"output" description:"specifing the output directory"`
-		Neighbor       string `short:"n" long:"neighbor" description:"neighbor ip adress to which add policy config"`
+		Neighbor   string `short:"n" long:"neighbor" description:"neighbor ip adress to which add policy config"`
 		Target     string `short:"t" long:"target" description:"target such as export or import to which add policy"`
 		PolicyName string `short:"p" long:"policy" description:"policy name bound to peer"`
 		Replace    bool   `short:"r" long:"replace" description:"Replace existing policy with new one" default:"false"`
+		Reject    bool   `short:"j" long:"reject" description:"Set default policy reject" default:"false"`
 	}
 
 	parser := flags.NewParser(&opts, flags.Default)
@@ -663,5 +741,5 @@ func main() {
 		log.Fatal(err)
 	}
 
-	bindPolicy(opts.OutputDir, opts.Neighbor, opts.Target, opts.PolicyName, opts.Replace)
+	bindPolicy(opts.OutputDir, opts.Neighbor, opts.Target, opts.PolicyName, opts.Replace, opts.Reject)
 }