| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 egress_test |
| |
| import ( |
| "fmt" |
| "os" |
| "testing" |
| |
| "github.com/apache/servicecomb-mesher/proxy/cmd" |
| mesherconfig "github.com/apache/servicecomb-mesher/proxy/config" |
| _ "github.com/apache/servicecomb-mesher/proxy/control/istio" |
| "github.com/apache/servicecomb-mesher/proxy/pkg/egress" |
| "github.com/apache/servicecomb-mesher/proxy/pkg/egress/archaius" |
| "github.com/go-chassis/go-chassis/control" |
| "github.com/go-chassis/go-chassis/core/config" |
| "github.com/go-chassis/go-chassis/core/lager" |
| "gopkg.in/yaml.v2" |
| ) |
| |
| func BenchmarkMatch(b *testing.B) { |
| lager.Initialize("", "DEBUG", "", |
| "size", true, 1, 10, 7) |
| |
| gopath := os.Getenv("GOPATH") |
| os.Setenv("CHASSIS_HOME", gopath+"/src/github.com/apache/servicecomb-mesher") |
| cmd.Init() |
| config.Init() |
| mesherconfig.Init() |
| egress.Init() |
| opts := control.Options{ |
| Infra: config.GlobalDefinition.Panel.Infra, |
| Address: config.GlobalDefinition.Panel.Settings["address"], |
| } |
| control.Init(opts) |
| var yamlContent = `--- |
| egress: |
| infra: cse # pilot or cse |
| address: http://istio-pilot.istio-system:15010 |
| egressRule: |
| google-ext: |
| - hosts: |
| - "www.google.com" |
| - "*.yahoo.com" |
| ports: |
| - port: 80 |
| protocol: HTTP |
| facebook-ext: |
| - hosts: |
| - "www.facebook.com" |
| ports: |
| - port: 80 |
| protocol: HTTP` |
| |
| ss := mesherconfig.EgressConfig{} |
| err := yaml.Unmarshal([]byte(yamlContent), &ss) |
| if err != nil { |
| fmt.Println("unmarshal failed") |
| } |
| archaius.SetEgressRule(ss.Destinations) |
| |
| myString := "www.google.com" |
| for i := 0; i < b.N; i++ { |
| egress.Match(myString) |
| } |
| } |
| |
| func TestMatch(t *testing.T) { |
| lager.Initialize("", "DEBUG", "", |
| "size", true, 1, 10, 7) |
| |
| gopath := os.Getenv("GOPATH") |
| os.Setenv("CHASSIS_HOME", gopath+"/src/github.com/apache/servicecomb-mesher") |
| cmd.Init() |
| config.Init() |
| mesherconfig.Init() |
| egress.Init() |
| //control.Init() |
| var yamlContent = `--- |
| egress: |
| infra: cse # pilot or cse |
| address: http://istio-pilot.istio-system:15010 |
| egressRule: |
| google-ext: |
| - hosts: |
| - "www.google.com" |
| - "*.yahoo.com" |
| ports: |
| - port: 80 |
| protocol: HTTP |
| facebook-ext: |
| - hosts: |
| - "www.facebook.com" |
| ports: |
| - port: 80 |
| protocol: HTTP` |
| |
| ss := mesherconfig.EgressConfig{} |
| err := yaml.Unmarshal([]byte(yamlContent), &ss) |
| if err != nil { |
| fmt.Println("unmarshal failed") |
| } |
| archaius.SetEgressRule(ss.Destinations) |
| |
| myString := "www.google.com" |
| b, _ := egress.Match(myString) |
| if b == false { |
| t.Errorf("Expected true but got false") |
| } |
| myString = "*.yahoo.com" |
| b, _ = egress.Match(myString) |
| if b == false { |
| t.Errorf("Expected true but got false") |
| } |
| } |