Add standalone test

Author: black-adder

.gitignore

@@ -14,7 +14,7 @@ Gemfile.lock
 vendor/
 jaeger-ui-build/
 examples/hotrod/hotrod*
-cmd/standalone/standalone*
+cmd/standalone/standalone-linux
 cmd/agent/agent*
 cmd/collector/collector*
 cmd/query/query*

cmd/standalone/main.go

@@ -29,13 +29,14 @@ import (
 	"strconv"
 
 	"github.com/gorilla/mux"
+	jaegerClientConfig "github.com/uber/jaeger-client-go/config"
+	"github.com/uber/jaeger-lib/metrics"
+	"github.com/uber/jaeger-lib/metrics/go-kit"
+	"github.com/uber/jaeger-lib/metrics/go-kit/expvar"
 	"github.com/uber/tchannel-go"
 	"github.com/uber/tchannel-go/thrift"
 	"go.uber.org/zap"
 
-	"github.com/uber/jaeger-lib/metrics"
-	"github.com/uber/jaeger-lib/metrics/go-kit"
-	"github.com/uber/jaeger-lib/metrics/go-kit/expvar"
 	agentApp "github.com/uber/jaeger/cmd/agent/app"
 	basic "github.com/uber/jaeger/cmd/builder"
 	collector "github.com/uber/jaeger/cmd/collector/app/builder"
@@ -118,6 +119,17 @@ func startCollector(logger *zap.Logger, baseFactory metrics.Factory, memoryStore
 func startQuery(logger *zap.Logger, baseFactory metrics.Factory, memoryStore *memory.Store) {
 	metricsFactory := baseFactory.Namespace("jaeger-query", nil)
 
+	jaegerClientConfig.Configuration{
+		Sampler: &jaegerClientConfig.SamplerConfig{
+			Type:  "const",
+			Param: 1.0,
+		},
+		RPCMetrics: true,
+	}.InitGlobalTracer(
+		"jaeger-query",
+		jaegerClientConfig.Metrics(baseFactory),
+	)
+
 	storageBuild, err := query.NewStorageBuilder(
 		basic.Options.LoggerOption(logger),
 		basic.Options.MetricsFactoryOption(metricsFactory),

cmd/standalone/standalone_test.go

@@ -0,0 +1,130 @@
+// +build integration
+
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"testing"
+	"time"
+
+	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/api/types/container"
+	"github.com/docker/docker/client"
+	"github.com/docker/go-connections/nat"
+	"github.com/stretchr/testify/require"
+
+	"code.uber.internal/infra/jaeger/model/ui"
+
+	"golang.org/x/net/context"
+)
+
+const (
+	imageName = "standalone"
+
+	host          = "0.0.0.0"
+	queryPort     = "16686"
+	queryHostPort = host + ":" + queryPort
+	queryURL      = "http://" + queryHostPort
+
+	getServicesURL = queryURL + "/api/services"
+	getTraceURL    = queryURL + "/api/traces?service=jaeger-query&tag=jaeger-debug-id:debug"
+)
+
+var (
+	httpClient = &http.Client{
+		Timeout: time.Second,
+	}
+)
+
+func TestStandalone(t *testing.T) {
+	ctx := context.Background()
+	cli, err := client.NewEnvClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	exposedPorts, portBindings, _ := nat.ParsePortSpecs([]string{
+		fmt.Sprintf("%s:%s", queryHostPort, queryPort),
+	})
+
+	resp, err := cli.ContainerCreate(
+		ctx,
+		&container.Config{Image: imageName, ExposedPorts: exposedPorts},
+		&container.HostConfig{PortBindings: portBindings},
+		nil,
+		"",
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if err := cli.ContainerStart(ctx, resp.ID, types.ContainerStartOptions{}); err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		if err := cli.ContainerRemove(ctx, resp.ID, types.ContainerRemoveOptions{Force: true}); err != nil {
+			t.Fatalf("cannot remove container: %s", err)
+		}
+	}()
+
+	// Check if the query service is available
+	if err := healthCheck(); err != nil {
+		t.Fatal(err)
+	}
+
+	createTrace(t)
+	getAPITrace(t)
+}
+
+func createTrace(t *testing.T) {
+	req, err := http.NewRequest("GET", getServicesURL, nil)
+	require.NoError(t, err)
+	req.Header.Add("jaeger-debug-id", "debug")
+
+	resp, err := httpClient.Do(req)
+	require.NoError(t, err)
+	resp.Body.Close()
+}
+
+type response struct {
+	Data []*ui.Trace `json:"data"`
+}
+
+func getAPITrace(t *testing.T) {
+	req, err := http.NewRequest("GET", getTraceURL, nil)
+	require.NoError(t, err)
+
+	var queryResponse response
+
+	for i := 0; i < 20; i++ {
+		resp, err := httpClient.Do(req)
+		require.NoError(t, err)
+
+		body, _ := ioutil.ReadAll(resp.Body)
+
+		err = json.Unmarshal(body, &queryResponse)
+		require.NoError(t, err)
+		resp.Body.Close()
+
+		if len(queryResponse.Data) == 1 {
+			break
+		}
+		time.Sleep(time.Second)
+	}
+	require.Len(t, queryResponse.Data, 1)
+	require.Len(t, queryResponse.Data[0].Spans, 1)
+}
+
+func healthCheck() error {
+	for i := 0; i < 100; i++ {
+		_, err := http.Get(queryURL)
+		if err == nil {
+			return nil
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+	return fmt.Errorf("query service is not ready")
+}