1 files changed, 121 insertions, 0 deletions
diff --git a/main/civisibility/integrations/manual_api_ddtestsuite.go b/main/civisibility/integrations/manual_api_ddtestsuite.go
new file mode 100644
index 0000000..68d936d
--- /dev/null
+++ b/main/civisibility/integrations/manual_api_ddtestsuite.go
@@ -0,0 +1,121 @@
+// 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"
+	"strings"
+	"time"
+
+	"ci-visibility-test-github/main/civisibility/constants"
+
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/ext"
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
+)
+
+// Test Suite
+
+// Ensures that tslvTestSuite implements the DdTestSuite interface.
+var _ DdTestSuite = (*tslvTestSuite)(nil)
+
+// tslvTestSuite implements the DdTestSuite interface and represents a suite of tests within a module.
+type tslvTestSuite struct {
+	ciVisibilityCommon
+	module  *tslvTestModule
+	suiteID uint64
+	name    string
+}
+
+// createTestSuite initializes a new test suite within a given module.
+func createTestSuite(module *tslvTestModule, name string, startTime time.Time) DdTestSuite {
+	if module == nil {
+		return nil
+	}
+
+	operationName := "test_suite"
+	if module.framework != "" {
+		operationName = fmt.Sprintf("%s.%s", strings.ToLower(module.framework), operationName)
+	}
+
+	resourceName := name
+
+	// Suite tags should include module and session tags so the backend can calculate the module and session fingerprint from the suite.
+	suiteTags := append(module.tags, tracer.Tag(constants.TestSuite, name))
+	testOpts := append(fillCommonTags([]tracer.StartSpanOption{
+		tracer.ResourceName(resourceName),
+		tracer.SpanType(constants.SpanTypeTestSuite),
+		tracer.StartTime(startTime),
+	}), suiteTags...)
+
+	span, ctx := tracer.StartSpanFromContext(context.Background(), operationName, testOpts...)
+	suiteID := span.Context().SpanID()
+	if module.session != nil {
+		span.SetTag(constants.TestSessionIDTag, fmt.Sprint(module.session.sessionID))
+	}
+	span.SetTag(constants.TestModuleIDTag, fmt.Sprint(module.moduleID))
+	span.SetTag(constants.TestSuiteIDTag, fmt.Sprint(suiteID))
+
+	suite := &tslvTestSuite{
+		module:  module,
+		suiteID: suiteID,
+		name:    name,
+		ciVisibilityCommon: ciVisibilityCommon{
+			startTime: startTime,
+			tags:      suiteTags,
+			span:      span,
+			ctx:       ctx,
+		},
+	}
+
+	// Ensure to close everything before CI visibility exits. In CI visibility mode, we try to never lose data.
+	PushCiVisibilityCloseAction(func() { suite.Close() })
+
+	return suite
+}
+
+// Name returns the name of the test suite.
+func (t *tslvTestSuite) Name() string { return t.name }
+
+// Module returns the module to which the test suite belongs.
+func (t *tslvTestSuite) Module() DdTestModule { return t.module }
+
+// Close closes the test suite and sets the finish time to the current time.
+func (t *tslvTestSuite) Close() { t.CloseWithFinishTime(time.Now()) }
+
+// CloseWithFinishTime closes the test suite with the given finish time.
+func (t *tslvTestSuite) CloseWithFinishTime(finishTime time.Time) {
+	t.mutex.Lock()
+	defer t.mutex.Unlock()
+	if t.closed {
+		return
+	}
+
+	t.span.Finish(tracer.FinishTime(finishTime))
+	t.closed = true
+}
+
+// SetError sets an error on the test suite and marks the module as having an error.
+func (t *tslvTestSuite) SetError(err error) {
+	t.ciVisibilityCommon.SetError(err)
+	t.Module().SetTag(ext.Error, true)
+}
+
+// SetErrorInfo sets detailed error information on the test suite and marks the module as having an error.
+func (t *tslvTestSuite) SetErrorInfo(errType string, message string, callstack string) {
+	t.ciVisibilityCommon.SetErrorInfo(errType, message, callstack)
+	t.Module().SetTag(ext.Error, true)
+}
+
+// CreateTest creates a new test with the given name and sets the start time to the current time.
+func (t *tslvTestSuite) CreateTest(name string) DdTest {
+	return t.CreateTestWithStartTime(name, time.Now())
+}
+
+// CreateTestWithStartTime creates a new test with the given name and start time.
+func (t *tslvTestSuite) CreateTestWithStartTime(name string, startTime time.Time) DdTest {
+	return createTest(t, name, startTime)
+}