Add Test Optimization

.github/workflows/build.yml

@@ -18,6 +18,16 @@ jobs:
 
       - name: Build
         run: go build ./...
-
+
+      - name: Install orchestrion
+        run: |
+          go install github.com/DataDog/orchestrion@latest
+          orchestrion pin
+
       - name: Test
-        run: go test -v $(go list ./... | grep -v /integration-tests) # We do not run integration-tests here because they require more tooling (Pulumi, invoke, ..). They will be run in a dedicated job
+        run: orchestrion go test -v $(go list ./... | grep -v /integration-tests) # We do not run integration-tests here because they require more tooling (Pulumi, invoke, ..). They will be run in a dedicated job
+        env:
+          DD_CIVISIBILITY_ENABLED: true
+          DD_CIVISIBILITY_AGENTLESS_ENABLED: true
+          DD_ENV: ci
+          DD_API_KEY: ${{ secrets.DATADOG_API_KEY }}

.gitlab-ci.yml

@@ -77,14 +77,20 @@ integration-testing:
     - chmod 400 $E2E_PRIVATE_KEY_PATH
     - ssh-add $E2E_PRIVATE_KEY_PATH
     - pip install -r requirements.txt
+    - go install github.com/DataDog/orchestrion@latest
+    - orchestrion pin
+    - export DD_CIVISIBILITY_ENABLED=true
+    - export DD_CIVISIBILITY_AGENTLESS_ENABLED=true
+    - export DD_ENV=ci
+    - export DD_API_KEY=$(vault kv get -field=api_key kv/k8s/gitlab-runner/test-infra-definitions/dd-token)
   script:
     - |
       if [ ! -f ./dist/main ]; then
         echo "no main binary found, please run 'build-pulumi-go-main' job"
         exit 1
       fi
     # execute test from dist directory to use the generated binary
-    - go test ./integration-tests -v -timeout 0s -workingDir=dist
+    - orchestrion go test ./integration-tests -v -timeout 0s -workingDir=dist
   variables:
     E2E_PUBLIC_KEY_PATH: /tmp/agent-integration-test-ssh-key.pub
     E2E_PRIVATE_KEY_PATH: /tmp/agent-integration-test-ssh-key

go.mod

@@ -1,6 +1,6 @@
 module github.com/DataDog/test-infra-definitions
 
-go 1.22.5
+go 1.22.8
 
 require (
 	dario.cat/mergo v1.0.1