Auto register custom ITransportFactory in Spring integration. (#1194)

Author: maciejwalkowiak

CHANGELOG.md

@@ -12,6 +12,7 @@
 * Enhancement: Polish Performance API (#1165)
 * Enhancement: Set "debug" through external properties (#1186)
 * Enhancement: Simplify Spring integration (#1188)
+* Enhancement: Auto register custom ITransportFactory in Spring integration (#1194)
 * Enhancement: Improve Kotlin property access in Performance API (#1193)
 
 # 4.0.0-alpha.3

sentry-spring/src/main/java/io/sentry/spring/SentryInitBeanPostProcessor.java

@@ -1,6 +1,7 @@
 package io.sentry.spring;
 
 import com.jakewharton.nopen.annotation.Open;
+import io.sentry.ITransportFactory;
 import io.sentry.Sentry;
 import io.sentry.SentryOptions;
 import io.sentry.SentryOptions.TracesSamplerCallback;
@@ -33,6 +34,9 @@ public Object postProcessAfterInitialization(
         applicationContext
             .getBeanProvider(TracesSamplerCallback.class)
             .ifAvailable(options::setTracesSampler);
+        applicationContext
+            .getBeanProvider(ITransportFactory.class)
+            .ifAvailable(options::setTransportFactory);
       }
       Sentry.init(options);
     }

sentry-spring/src/test/kotlin/io/sentry/spring/EnableSentryTest.kt

@@ -1,6 +1,8 @@
 package io.sentry.spring
 
+import com.nhaarman.mockitokotlin2.mock
 import io.sentry.IHub
+import io.sentry.ITransportFactory
 import io.sentry.SentryOptions
 import kotlin.test.Test
 import org.assertj.core.api.Assertions.assertThat
@@ -92,6 +94,16 @@ class EnableSentryTest {
             }
     }
 
+    @Test
+    fun `configures custom TransportFactory`() {
+        ApplicationContextRunner().withConfiguration(UserConfigurations.of(AppConfigWithCustomTransportFactory::class.java))
+            .run {
+                val options = it.getBean(SentryOptions::class.java)
+                val transportFactory = it.getBean(ITransportFactory::class.java)
+                assertThat(options.transportFactory).isEqualTo(transportFactory)
+            }
+    }
+
     @EnableSentry(dsn = "http://key@localhost/proj")
     class AppConfig
 
@@ -114,4 +126,11 @@ class EnableSentryTest {
             }
         }
     }
+
+    @EnableSentry(dsn = "http://key@localhost/proj")
+    class AppConfigWithCustomTransportFactory {
+
+        @Bean
+        fun tracesSampler() = mock<ITransportFactory>()
+    }
 }