diff --git a/airbyte-analytics/src/test/java/io/airbyte/analytics/SegmentTrackingClientTest.java b/airbyte-analytics/src/test/java/io/airbyte/analytics/SegmentTrackingClientTest.java index 16b87edab824..9169fc3f4b10 100644 --- a/airbyte-analytics/src/test/java/io/airbyte/analytics/SegmentTrackingClientTest.java +++ b/airbyte-analytics/src/test/java/io/airbyte/analytics/SegmentTrackingClientTest.java @@ -33,6 +33,7 @@ import com.segment.analytics.Analytics; import com.segment.analytics.messages.IdentifyMessage; import com.segment.analytics.messages.TrackMessage; +import io.airbyte.config.Configs.WorkerEnvironment; import java.util.Map; import java.util.UUID; import java.util.function.Supplier; @@ -55,7 +56,7 @@ class SegmentTrackingClientTest { void setup() { analytics = mock(Analytics.class); roleSupplier = mock(Supplier.class); - segmentTrackingClient = new SegmentTrackingClient(() -> identity, null, analytics); + segmentTrackingClient = new SegmentTrackingClient(() -> identity, WorkerEnvironment.DOCKER, null, analytics); } @SuppressWarnings("OptionalGetWithoutIsPresent") @@ -70,6 +71,7 @@ void testIdentify() { verify(analytics).enqueue(mockBuilder.capture()); final IdentifyMessage actual = mockBuilder.getValue().build(); final Map expectedTraits = ImmutableMap.builder() + .put("deployment_env", WorkerEnvironment.DOCKER) .put("airbyte_version", AIRBYTE_VERSION) .put("email", identity.getEmail().get()) .put("anonymized", identity.isAnonymousDataCollection()) @@ -82,7 +84,7 @@ void testIdentify() { @Test void testIdentifyWithRole() { - segmentTrackingClient = new SegmentTrackingClient(() -> identity, "role", analytics); + segmentTrackingClient = new SegmentTrackingClient(() -> identity, WorkerEnvironment.DOCKER, "role", analytics); // equals is not defined on MessageBuilder, so we need to use ArgumentCaptor to inspect each field // manually. ArgumentCaptor mockBuilder = ArgumentCaptor.forClass(IdentifyMessage.Builder.class); @@ -93,6 +95,7 @@ void testIdentifyWithRole() { verify(analytics).enqueue(mockBuilder.capture()); final IdentifyMessage actual = mockBuilder.getValue().build(); final Map expectedTraits = ImmutableMap.builder() + .put("deployment_env", WorkerEnvironment.DOCKER) .put("airbyte_version", AIRBYTE_VERSION) .put("email", identity.getEmail().get()) .put("anonymized", identity.isAnonymousDataCollection()) diff --git a/airbyte-server/src/main/java/io/airbyte/server/ServerApp.java b/airbyte-server/src/main/java/io/airbyte/server/ServerApp.java index 1dd118d48288..3d888e697efd 100644 --- a/airbyte-server/src/main/java/io/airbyte/server/ServerApp.java +++ b/airbyte-server/src/main/java/io/airbyte/server/ServerApp.java @@ -219,6 +219,7 @@ public static void runServer(final Set requestFilters, TrackingClientSingleton.initialize( configs.getTrackingStrategy(), + WorkerEnvironment.DOCKER, configs.getAirbyteRole(), configs.getAirbyteVersion(), configRepository);