1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
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)
}
|
