| //go:build integ |
| // +build integ |
| |
| // Copyright Istio Authors |
| // |
| // 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 vm |
| |
| import ( |
| "fmt" |
| "reflect" |
| "testing" |
| "time" |
| ) |
| |
| import ( |
| "google.golang.org/genproto/googleapis/devtools/cloudtrace/v1" |
| loggingpb "google.golang.org/genproto/googleapis/logging/v2" |
| monitoring "google.golang.org/genproto/googleapis/monitoring/v3" |
| "google.golang.org/protobuf/proto" |
| "istio.io/pkg/log" |
| ) |
| |
| import ( |
| "github.com/apache/dubbo-go-pixiu/pkg/test" |
| "github.com/apache/dubbo-go-pixiu/pkg/test/framework" |
| "github.com/apache/dubbo-go-pixiu/pkg/test/framework/components/echo" |
| "github.com/apache/dubbo-go-pixiu/pkg/test/framework/components/stackdriver" |
| "github.com/apache/dubbo-go-pixiu/pkg/test/util/retry" |
| sdtest "github.com/apache/dubbo-go-pixiu/tests/integration/telemetry/stackdriver" |
| ) |
| |
| func TestVMTelemetry(t *testing.T) { |
| framework. |
| NewTest(t). |
| Features("observability.telemetry.stackdriver"). |
| Run(func(t framework.TestContext) { |
| // Set up strict mTLS. This gives a bit more assurance the calls are actually going through envoy, |
| // and certs are set up correctly. |
| t.ConfigIstio().YAML(ns.Name(), enforceMTLS).ApplyOrFail(t) |
| |
| clientBuilder.BuildOrFail(t) |
| serverBuilder.BuildOrFail(t) |
| |
| retry.UntilSuccessOrFail(t, func() error { |
| // send single request from client -> server |
| if _, err := client.Call(echo.CallOptions{ |
| To: server, |
| Port: echo.Port{ |
| Name: "http", |
| }, |
| Count: 1, |
| Retry: echo.Retry{ |
| NoRetry: true, |
| }, |
| }); err != nil { |
| return err |
| } |
| |
| // Verify stackdriver metrics |
| gotMetrics := gotRequestCountMetrics(t, wantClientReqs, wantServerReqs) |
| |
| // Verify log entry |
| gotLogs := gotLogEntry(t, wantLogEntry) |
| |
| // verify traces |
| gotTraces := gotTrace(t, wantTrace) |
| |
| if !(gotMetrics && gotLogs && gotTraces) { |
| return fmt.Errorf("did not receive all expected telemetry; status: metrics=%t, logs=%t, traces=%t", gotMetrics, gotLogs, gotTraces) |
| } |
| |
| return nil |
| }, retry.Delay(3*time.Second), retry.Timeout(40*time.Second)) |
| }) |
| } |
| |
| func traceEqual(got, want *cloudtrace.Trace) bool { |
| if len(got.Spans) != len(want.Spans) { |
| log.Infof("incorrect number of spans: got %d, want: %d", len(got.Spans), len(want.Spans)) |
| return false |
| } |
| if got.ProjectId != want.ProjectId { |
| log.Errorf("mismatched project ids: got %q, want %q", got.ProjectId, want.ProjectId) |
| return false |
| } |
| |
| for _, wantSpan := range want.Spans { |
| foundSpan := false |
| for _, gotSpan := range got.Spans { |
| delete(gotSpan.Labels, "guid:x-request-id") |
| delete(gotSpan.Labels, "node_id") |
| delete(gotSpan.Labels, "peer.address") |
| delete(gotSpan.Labels, "zone") |
| delete(gotSpan.Labels, "g.co/agent") // ignore OpenCensus lib versions |
| delete(gotSpan.Labels, "response_size") // this could be slightly off, just ignore |
| if foundSpan = reflect.DeepEqual(gotSpan.Labels, wantSpan.Labels); foundSpan { |
| break |
| } |
| } |
| if !foundSpan { |
| log.Errorf("missing span from trace: got %v\nwant %v", got, want) |
| return false |
| } |
| } |
| |
| return true |
| } |
| |
| func gotRequestCountMetrics(t test.Failer, wantClient, wantServer *monitoring.TimeSeries) bool { |
| ts, err := sdInst.ListTimeSeries(ns.Name()) |
| if err != nil { |
| log.Errorf("could not get list of time-series from stackdriver: %v", err) |
| return false |
| } |
| |
| var gotServer, gotClient bool |
| for _, series := range ts { |
| // Making resource nil, as test can run on various platforms. |
| series.Resource = nil |
| if proto.Equal(series, wantServer) { |
| gotServer = true |
| } |
| if proto.Equal(series, wantClient) { |
| gotClient = true |
| } |
| } |
| |
| if !gotServer { |
| sdtest.LogMetricsDiff(t, wantServer, ts) |
| } |
| if !gotClient { |
| sdtest.LogMetricsDiff(t, wantClient, ts) |
| } |
| return gotServer && gotClient |
| } |
| |
| func gotLogEntry(t test.Failer, want *loggingpb.LogEntry) bool { |
| return sdtest.ValidateLogEntry(t, want, stackdriver.ServerAccessLog) == nil |
| } |
| |
| func gotTrace(t test.Failer, want *cloudtrace.Trace) bool { |
| traces, err := sdInst.ListTraces(ns.Name()) |
| if err != nil { |
| log.Errorf("failed to retrieve list of tracespans from stackdriver: %v", err) |
| return false |
| } |
| |
| for _, trace := range traces { |
| if found := traceEqual(trace, want); found { |
| return true |
| } |
| } |
| sdtest.LogTraceDiff(t, want, traces) |
| return false |
| } |