equinix-labs · tobert · Aug 25, 2021 · Aug 19, 2021 · Aug 19, 2021 · Aug 19, 2021
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -0,0 +1,29 @@
+name: CI
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v2
+      - name: Setup
+        uses: actions/setup-go@v2
+        with:
+          go-version: 1.16
+      - name: Build otelinit
+        run: go build -v ./...
+      - name: Build test stub
+        working-directory: ./cmd/test-otel-init-go
+        run: go build -v
+      - name: Install otel-cli for tests
+        uses: engineerd/[email protected]
+        with:
+          name: "otel-cli"
+          pathInArchive: /otel-cli
+          url: "https://github.com/equinix-labs/otel-cli/releases/download/v0.0.5/otel-cli-0.0.5-Linux-x86_64.tar.gz"
+      - name: Test
+        run: go test -v ./...
diff --git a/.gitignore b/.gitignore
@@ -11,5 +11,4 @@
 # Output of the go coverage tool, specifically when used with LiteIDE
 *.out
 
-# Dependency directories (remove the comment below to include it)
-# vendor/
+cmd/test-otel-init-go/test-otel-init-go
diff --git a/README.md b/README.md
@@ -22,7 +22,7 @@ import (
 
 func main() {
     ctx := context.Background()
-    otelShutdown := otelinit.InitOpenTelemetry(ctx, "my-amazing-application")
+    ctx, otelShutdown := otelinit.InitOpenTelemetry(ctx, "my-amazing-application")
     defer otelShutdown(ctx)
 }
 ```

diff --git a/cmd/test-otel-init-go/main.go b/cmd/test-otel-init-go/main.go
@@ -0,0 +1,73 @@
+package main
+
+// All this program does is load up otel-init-go, create one trace, dump state
+// in json for all these things, then exit. This data is intended to be consumed
+// in main_test.go, which is really about testing otel-cli-init itself.
+
+import (
+	"context"
+	"encoding/json"
+	"log"
+	"os"
+	"strconv"
+	"strings"
+
+	"github.com/equinix-labs/otel-init-go/otelinit"
+	"go.opentelemetry.io/otel"
+)
+
+func main() {
+	ctx := context.Background()
+	ctx, otelShutdown := otelinit.InitOpenTelemetry(ctx, "otel-init-go-test")
+	defer otelShutdown(ctx)
+
+	tracer := otel.Tracer("otel-init-go-test")
+	ctx, span := tracer.Start(ctx, "dump state")
+
+	env := make(map[string]string)
+	for _, e := range os.Environ() {
+		parts := strings.SplitN(e, "=", 2)
+		if len(parts) == 2 {
+			// it's not great to hard-code a specific envvar here but this is
+			// probably the most dangerous one we don't want to blithely print
+			// here. open to suggestions...
+			if !strings.HasPrefix(parts[0], "GITHUB") {
+				env[parts[0]] = parts[1]
+			}
+		} else {
+			log.Fatalf("BUG this shouldn't happen")
+		}
+	}
+
+	// public.go stuffs the config in the context just so we can do this
+	conf, ok := otelinit.ConfigFromContext(ctx)
+	if !ok {
+		log.Println("failed to retrieve otelinit.Config pointer from context, test results may be invalid")
+		conf = &otelinit.Config{}
+	}
+	sc := span.SpanContext()
+	outData := map[string]map[string]string{
+		"config": {
+			"endpoint":     conf.Endpoint,
+			"service_name": conf.Servicename,
+			"insecure":     strconv.FormatBool(conf.Insecure),
+		},
+		"otel": {
+			"trace_id":    sc.TraceID().String(),
+			"span_id":     sc.SpanID().String(),
+			"trace_flags": sc.TraceFlags().String(),
+			"is_sampled":  strconv.FormatBool(sc.IsSampled()),
+		},
+		"env": env,
+	}
+
+	js, err := json.MarshalIndent(outData, "", "    ")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	os.Stdout.Write(js)
+	os.Stdout.WriteString("\n")
+
+	span.End()
+}