E2E test for Channel Tracing

pkg/tracing/setup.go

@@ -40,6 +40,13 @@ var (
 		SampleRate:     0.01,
 		ZipkinEndpoint: "http://zipkin.istio-system.svc.cluster.local:9411/api/v2/spans",
 	}
+
+	AlwaysSample = &tracingconfig.Config{
+		Backend:        tracingconfig.Zipkin,
+		Debug:          true,
+		SampleRate:     1.0,
+		ZipkinEndpoint: "http://zipkin.istio-system.svc.cluster.local:9411/api/v2/spans",
+	}
 )
 
 // setupPublishing sets up trace publishing for the process. Note that other pieces
@@ -111,7 +118,7 @@ func enableZeroSamplingCM(ns string) corev1.ConfigMap {
 			Namespace: ns,
 		},
 		Data: map[string]string{
-			"enable":          "True",
+			"backend":         "zipkin",
 			"debug":           "False",
 			"sample-rate":     "0",
 			"zipkin-endpoint": "http://zipkin.istio-system.svc.cluster.local:9411/api/v2/spans",

test/base/resources/kube.go

@@ -98,7 +98,7 @@ func EventLoggerPod(name string) *corev1.Pod {
 	return eventLoggerPod("logevents", name)
 }
 
-// EventDetailsPod creates a Pod that vaalidates events received and log details about events.
+// EventDetailsPod creates a Pod that validates events received and log details about events.
 func EventDetailsPod(name string) *corev1.Pod {
 	return eventLoggerPod("eventdetails", name)
 }

test/common/validation.go

@@ -34,38 +34,41 @@ const (
 	timeout  = 4 * time.Minute
 )
 
+// GetLog gets the logs from the given Pod in the namespace of this client. It will get the logs
+// from the first container, whichever it is.
+func (client *Client) GetLog(podName string) (string, error) {
+	containerName, err := client.getContainerName(podName, client.Namespace)
+	if err != nil {
+		return "", err
+	}
+	logs, err := client.Kube.PodLogs(podName, containerName, client.Namespace)
+	if err != nil {
+		return "", err
+	}
+	return string(logs), nil
+}
+
 // CheckLog waits until logs for the logger Pod satisfy the checker.
 // If the checker does not pass within timeout it returns error.
 func (client *Client) CheckLog(podName string, checker func(string) bool) error {
-	namespace := client.Namespace
-	containerName, err := client.getContainerName(podName, namespace)
-	if err != nil {
-		return err
-	}
 	return wait.PollImmediate(interval, timeout, func() (bool, error) {
-		logs, err := client.Kube.PodLogs(podName, containerName, namespace)
+		logs, err := client.GetLog(podName)
 		if err != nil {
 			return true, err
 		}
-		return checker(string(logs)), nil
+		return checker(logs), nil
 	})
 }
 
 // CheckLogEmpty waits the given amount of time and check the log is empty
 func (client *Client) CheckLogEmpty(podName string, timeout time.Duration) error {
 	time.Sleep(timeout)
-
-	namespace := client.Namespace
-	containerName, err := client.getContainerName(podName, namespace)
+	logs, err := client.GetLog(podName)
 	if err != nil {
 		return err
 	}
-	logs, err := client.Kube.PodLogs(podName, containerName, namespace)
-	if err != nil {
-		return err
-	}
-	if string(logs) != "" {
-		return fmt.Errorf("expected empty log, got %s", string(logs))
+	if logs != "" {
+		return fmt.Errorf("expected empty log, got %s", logs)
 	}
 	return nil
 }
@@ -106,16 +109,11 @@ func CheckerContainsAtLeast(content string, count int) func(string) bool {
 // FindAnyLogContents attempts to find logs for given Pod/Container that has 'any' of the given contents.
 // It returns an error if it couldn't retrieve the logs. In case 'any' of the contents are there, it returns true.
 func (client *Client) FindAnyLogContents(podName string, contents []string) (bool, error) {
-	namespace := client.Namespace
-	containerName, err := client.getContainerName(podName, namespace)
-	if err != nil {
-		return false, err
-	}
-	logs, err := client.Kube.PodLogs(podName, containerName, namespace)
+	logs, err := client.GetLog(podName)
 	if err != nil {
 		return false, err
 	}
-	eventContentsSet, err := parseEventContentsFromPodLogs(string(logs))
+	eventContentsSet, err := parseEventContentsFromPodLogs(logs)
 	if err != nil {
 		return false, err
 	}

test/conformance/channel_tracing_test.go

@@ -0,0 +1,155 @@
+// +build e2e
+
+/*
+Copyright 2019 The Knative 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 conformance
+
+import (
+	"fmt"
+	"regexp"
+	"testing"
+	"time"
+
+	"k8s.io/apimachinery/pkg/util/uuid"
+	"knative.dev/eventing/test/base/resources"
+	"knative.dev/eventing/test/common"
+	"knative.dev/eventing/test/conformance/zipkin"
+)
+
+func TestChannelTracing(t *testing.T) {
+	testCases := map[string]struct {
+		incomingTraceId bool
+		istio           bool
+	}{
+		"no incoming trace id": {},
+		"includes incoming trace id": {
+			incomingTraceId: true,
+		}, /*
+			"caller has istio": {
+				istio: true,
+			},*/
+	}
+
+	for n, tc := range testCases {
+		loggerPodName := "logger"
+		t.Run(n, func(t *testing.T) {
+			channelTestRunner.RunTests(t, common.FeatureBasic, func(st *testing.T, channel string) {
+				// Don't accidentally use t, use st instead. To ensure this, shadow 't' to some a
+				// useless type.
+				t := struct{}{}
+				_ = fmt.Sprintf("%s", t)
+
+				client := common.Setup(st, true)
+				defer common.TearDown(client)
+
+				// Do NOT call zipkin.CleanupZipkinTracingSetup. That will be called exactly once in
+				// TestMain.
+				zipkin.SetupZipkinTracing(client.Kube.Kube, st.Logf)
+
+				mustContain := setupChannelTracing(st, channel, client, loggerPodName, tc.incomingTraceId)
+				assertLogContents(st, client, loggerPodName, mustContain)
+
+				traceID := getTraceID(st, client, loggerPodName)
+				expectedNumSpans := 3
+				if tc.incomingTraceId {
+					expectedNumSpans = 4
+				}
+				trace, err := zipkin.JSONTrace(traceID, expectedNumSpans, 15*time.Second)
+				if err != nil {
+					st.Fatalf("Unable to get trace %q: %v", traceID, err)
+				}
+				st.Logf("I got the trace, %q!\n%+v", traceID, trace)
+
+				// TODO: Assert information in the trace.
+			})
+		})
+	}
+}
+
+// setupChannelTracing is the general setup for TestChannelTracing. It creates the following:
+// SendEvents (Pod) -> Channel -> Subscription -> K8s Service -> LogEvents (Pod)
+// It returns a string that is expected to be sent by the SendEvents Pod and should be present in
+// the LogEvents Pod logs.
+func setupChannelTracing(t *testing.T, channel string, client *common.Client, loggerPodName string, incomingTraceId bool) string {
+	// Create the Channel.
+	channelName := "ch"
+	channelTypeMeta := common.GetChannelTypeMeta(channel)
+	client.CreateChannelOrFail(channelName, channelTypeMeta)
+
+	// Create the 'sink', a LogEvents Pod and a K8s Service that points to it.
+	pod := resources.EventDetailsPod(loggerPodName)
+	client.CreatePodOrFail(pod, common.WithService(loggerPodName))
+
+	// Create the Subscription linking the Channel to the LogEvents K8s Service.
+	client.CreateSubscriptionOrFail(
+		"sub",
+		channelName,
+		channelTypeMeta,
+		resources.WithSubscriberForSubscription(loggerPodName),
+	)
+
+	// Wait for all test resources to be ready, so that we can start sending events.
+	if err := client.WaitForAllTestResourcesReady(); err != nil {
+		t.Fatalf("Failed to get all test resources ready: %v", err)
+	}
+
+	// Everything is setup to receive an event. Generate a CloudEvent.
+	senderName := "sender"
+	eventID := fmt.Sprintf("%s", uuid.NewUUID())
+	body := fmt.Sprintf("TestChannelTracing %s", eventID)
+	event := &resources.CloudEvent{
+		ID:       eventID,
+		Source:   senderName,
+		Type:     resources.CloudEventDefaultType,
+		Data:     fmt.Sprintf(`{"msg":%q}`, body),
+		Encoding: resources.CloudEventEncodingBinary,
+	}
+
+	// Send the CloudEvent (either with or without tracing inside the SendEvents Pod).
+	sendEvent := client.SendFakeEventToAddressable
+	if incomingTraceId {
+		sendEvent = client.SendFakeEventWithTracingToAddressable
+	}
+	if err := sendEvent(senderName, channelName, channelTypeMeta, event); err != nil {
+		t.Fatalf("Failed to send fake CloudEvent to the channel %q", channelName)
+	}
+	return body
+}
+
+// assertLogContents verifies that loggerPodName's logs contains mustContain. It is used to show
+// that the expected event was sent to the logger Pod.
+func assertLogContents(t *testing.T, client *common.Client, loggerPodName string, mustContain string) {
+	if err := client.CheckLog(loggerPodName, common.CheckerContains(mustContain)); err != nil {
+		t.Fatalf("String %q not found in logs of logger pod %q: %v", mustContain, loggerPodName, err)
+	}
+}
+
+// getTraceID gets the TraceID from loggerPodName's logs. It will also assert that body is present.
+func getTraceID(t *testing.T, client *common.Client, loggerPodName string) string {
+	logs, err := client.GetLog(loggerPodName)
+	if err != nil {
+		t.Fatalf("Error getting logs: %v", err)
+	}
+	// This is the format that the eventdetails image prints headers.
+	re := regexp.MustCompile("\nGot Header X-B3-Traceid: ([a-zA-Z0-9]{32})\n")
+	submatches := re.FindStringSubmatch(logs)
+	if len(submatches) != 2 {
+		t.Fatalf("Unable to extract traceID: %q", logs)
+	}
+	traceID := submatches[1]
+	return traceID
+}

test/conformance/main_test.go

@@ -16,11 +16,13 @@ limitations under the License.
 package conformance
 
 import (
+	"log"
 	"os"
 	"testing"
 
 	"knative.dev/eventing/test"
 	"knative.dev/eventing/test/common"
+	"knative.dev/eventing/test/conformance/zipkin"
 )
 
 var setup = common.Setup
@@ -30,11 +32,21 @@ var channels test.Channels
 var channelTestRunner common.ChannelTestRunner
 
 func TestMain(m *testing.M) {
+	os.Exit(wrappedMain(m))
+}
+
+func wrappedMain(m *testing.M) int {
 	test.InitializeEventingFlags()
 	channels = test.EventingFlags.Channels
 	channelTestRunner = common.ChannelTestRunner{
 		ChannelFeatureMap: common.ChannelFeatureMap,
 		ChannelsToTest:    test.EventingFlags.Channels,
 	}
-	os.Exit(m.Run())
+
+	// Any tests may SetupZipkinTracing, it will only actually be done once. This should be the ONLY
+	// place that cleans it up. If an individual test calls this instead, then it will break other
+	// tests that need the tracing in place.
+	defer zipkin.CleanupZipkinTracingSetup(log.Printf)
+
+	return m.Run()
 }