1 files changed, 158 insertions, 0 deletions
diff --git a/main/civisibility/integrations/manual_api_ddtest.go b/main/civisibility/integrations/manual_api_ddtest.go
new file mode 100644
index 0000000..7104e70
--- /dev/null
+++ b/main/civisibility/integrations/manual_api_ddtest.go
@@ -0,0 +1,158 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2024 Datadog, Inc.
+
+package integrations
+
+import (
+	"context"
+	"fmt"
+	"runtime"
+	"strings"
+	"time"
+
+	"ci-visibility-test-github/main/civisibility/constants"
+	"ci-visibility-test-github/main/civisibility/utils"
+
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/ext"
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
+)
+
+// Test
+
+// Ensures that tslvTest implements the DdTest interface.
+var _ DdTest = (*tslvTest)(nil)
+
+// tslvTest implements the DdTest interface and represents an individual test within a suite.
+type tslvTest struct {
+	ciVisibilityCommon
+	suite *tslvTestSuite
+	name  string
+}
+
+// createTest initializes a new test within a given suite.
+func createTest(suite *tslvTestSuite, name string, startTime time.Time) DdTest {
+	if suite == nil {
+		return nil
+	}
+
+	operationName := "test"
+	if suite.module.framework != "" {
+		operationName = fmt.Sprintf("%s.%s", strings.ToLower(suite.module.framework), operationName)
+	}
+
+	resourceName := fmt.Sprintf("%s.%s", suite.name, name)
+
+	// Test tags should include suite, module, and session tags so the backend can calculate the suite, module, and session fingerprint from the test.
+	testTags := append(suite.tags, tracer.Tag(constants.TestName, name))
+	testOpts := append(fillCommonTags([]tracer.StartSpanOption{
+		tracer.ResourceName(resourceName),
+		tracer.SpanType(constants.SpanTypeTest),
+		tracer.StartTime(startTime),
+	}), testTags...)
+
+	span, ctx := tracer.StartSpanFromContext(context.Background(), operationName, testOpts...)
+	if suite.module.session != nil {
+		span.SetTag(constants.TestSessionIDTag, fmt.Sprint(suite.module.session.sessionID))
+	}
+	span.SetTag(constants.TestModuleIDTag, fmt.Sprint(suite.module.moduleID))
+	span.SetTag(constants.TestSuiteIDTag, fmt.Sprint(suite.suiteID))
+
+	t := &tslvTest{
+		suite: suite,
+		name:  name,
+		ciVisibilityCommon: ciVisibilityCommon{
+			startTime: startTime,
+			tags:      testTags,
+			span:      span,
+			ctx:       ctx,
+		},
+	}
+
+	// Ensure to close everything before CI visibility exits. In CI visibility mode, we try to never lose data.
+	PushCiVisibilityCloseAction(func() { t.Close(ResultStatusFail) })
+
+	return t
+}
+
+// Name returns the name of the test.
+func (t *tslvTest) Name() string { return t.name }
+
+// Suite returns the suite to which the test belongs.
+func (t *tslvTest) Suite() DdTestSuite { return t.suite }
+
+// Close closes the test with the given status and sets the finish time to the current time.
+func (t *tslvTest) Close(status TestResultStatus) { t.CloseWithFinishTime(status, time.Now()) }
+
+// CloseWithFinishTime closes the test with the given status and finish time.
+func (t *tslvTest) CloseWithFinishTime(status TestResultStatus, finishTime time.Time) {
+	t.CloseWithFinishTimeAndSkipReason(status, finishTime, "")
+}
+
+// CloseWithFinishTimeAndSkipReason closes the test with the given status, finish time, and skip reason.
+func (t *tslvTest) CloseWithFinishTimeAndSkipReason(status TestResultStatus, finishTime time.Time, skipReason string) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	if t.closed {
+		return
+	}
+
+	switch status {
+	case ResultStatusPass:
+		t.span.SetTag(constants.TestStatus, constants.TestStatusPass)
+	case ResultStatusFail:
+		t.span.SetTag(constants.TestStatus, constants.TestStatusFail)
+	case ResultStatusSkip:
+		t.span.SetTag(constants.TestStatus, constants.TestStatusSkip)
+	}
+
+	if skipReason != "" {
+		t.span.SetTag(constants.TestSkipReason, skipReason)
+	}
+
+	t.span.Finish(tracer.FinishTime(finishTime))
+	t.closed = true
+}
+
+// SetError sets an error on the test and marks the suite and module as having an error.
+func (t *tslvTest) SetError(err error) {
+	t.ciVisibilityCommon.SetError(err)
+	t.Suite().SetTag(ext.Error, true)
+	t.Suite().Module().SetTag(ext.Error, true)
+}
+
+// SetErrorInfo sets detailed error information on the test and marks the suite and module as having an error.
+func (t *tslvTest) SetErrorInfo(errType string, message string, callstack string) {
+	t.ciVisibilityCommon.SetErrorInfo(errType, message, callstack)
+	t.Suite().SetTag(ext.Error, true)
+	t.Suite().Module().SetTag(ext.Error, true)
+}
+
+// SetTestFunc sets the function to be tested and records its source location.
+func (t *tslvTest) SetTestFunc(fn *runtime.Func) {
+	if fn == nil {
+		return
+	}
+
+	file, line := fn.FileLine(fn.Entry())
+	file = utils.GetRelativePathFromCITagsSourceRoot(file)
+	t.SetTag(constants.TestSourceFile, file)
+	t.SetTag(constants.TestSourceStartLine, line)
+
+	codeOwners := utils.GetCodeOwners()
+	if codeOwners != nil {
+		match, found := codeOwners.Match("/" + file)
+		if found {
+			t.SetTag(constants.TestCodeOwners, match.GetOwnersString())
+		}
+	}
+}
+
+// SetBenchmarkData sets benchmark data for the test.
+func (t *tslvTest) SetBenchmarkData(measureType string, data map[string]any) {
+	t.span.SetTag(constants.TestType, constants.TestTypeBenchmark)
+	for k, v := range data {
+		t.span.SetTag(fmt.Sprintf("benchmark.%s.%s", measureType, k), v)
+	}
+}