From 4b123747b302479e063b09d590d22f785d668502 Mon Sep 17 00:00:00 2001
From: Marcin Grzejszczak <marcin.grzejszczak@gmail.com>
Date: Fri, 26 Feb 2021 10:20:53 +0100
Subject: [PATCH] Making tests less brittle

---
 .../sleuth/instrument/web/TraceNoWebEnvironmentTests.java      | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/TraceNoWebEnvironmentTests.java b/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/TraceNoWebEnvironmentTests.java
index 8f6c18ee8..5def5459f 100644
--- a/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/TraceNoWebEnvironmentTests.java
+++ b/spring-cloud-sleuth-core/src/test/java/org/springframework/cloud/sleuth/instrument/web/TraceNoWebEnvironmentTests.java
@@ -48,7 +48,8 @@ public class TraceNoWebEnvironmentTests {
 			client.createSomeTestRequest();
 		}
 		catch (Exception e) {
-			then(e.getCause().getClass())
+			Throwable cause = e.getCause() != null ? e.getCause() : e;
+			then(cause.getClass())
 					.isNotEqualTo(NoSuchBeanDefinitionException.class);
 		}
 	}