From 0c707e204d9716f0f5fb5cd55cfc7050ab6190a6 Mon Sep 17 00:00:00 2001
From: Kamil Smigielski <kamil.smigielski@allegro.pl>
Date: Mon, 18 Nov 2024 10:56:50 +0100
Subject: [PATCH] fix test

---
 .../allegro/tech/servicemesh/envoycontrol/CustomRouteTest.kt  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/envoy-control-tests/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/CustomRouteTest.kt b/envoy-control-tests/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/CustomRouteTest.kt
index 917644b32..2b9065eb4 100644
--- a/envoy-control-tests/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/CustomRouteTest.kt
+++ b/envoy-control-tests/src/main/kotlin/pl/allegro/tech/servicemesh/envoycontrol/CustomRouteTest.kt
@@ -3,6 +3,8 @@ package pl.allegro.tech.servicemesh.envoycontrol
 import org.assertj.core.api.Assertions.assertThat
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.extension.RegisterExtension
+import pl.allegro.tech.servicemesh.envoycontrol.assertions.isFrom
+import pl.allegro.tech.servicemesh.envoycontrol.assertions.isOk
 import pl.allegro.tech.servicemesh.envoycontrol.config.consul.ConsulExtension
 import pl.allegro.tech.servicemesh.envoycontrol.config.envoy.EnvoyExtension
 import pl.allegro.tech.servicemesh.envoycontrol.config.envoycontrol.EnvoyControlExtension
@@ -52,6 +54,6 @@ internal class CustomRouteTest {
             endpoint = "/status/wrapper/prometheus"
         )
         // then
-        assertThat(response.isSuccessful).isTrue()
+        assertThat(response).isOk().isFrom(wrapper)
     }
 }