From fe5986af0ceecec3a7c4cae1997ef0891817f3b0 Mon Sep 17 00:00:00 2001
From: Daniel Redondo <daniel.redondo@undefinedlabs.com>
Date: Thu, 4 Jun 2020 15:00:23 +0200
Subject: [PATCH] Fix test span counter

---
 agent/recorder.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/agent/recorder.go b/agent/recorder.go
index 5d994aad..2c528f05 100644
--- a/agent/recorder.go
+++ b/agent/recorder.go
@@ -14,6 +14,7 @@ import (
 	"time"
 
 	"github.com/google/uuid"
+	"github.com/opentracing/opentracing-go"
 	"gopkg.in/tomb.v2"
 
 	"go.undefinedlabs.com/scopeagent/tags"
@@ -168,8 +169,10 @@ func (r *SpanRecorder) sendSpans() (error, bool) {
 
 		var testSpans int64
 		for _, span := range spans {
-			if isTestSpan(span) {
-				testSpans++
+			if tags, ok := span["tags"].(opentracing.Tags); ok {
+				if isTestSpan(tags) {
+					testSpans++
+				}
 			}
 		}