pl.allegro.tech.servicemesh.envoycontrol.OutlierDetectionTest.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of envoy-control-tests Show documentation
Production-ready Control Plane for Service Mesh based on Envoy Proxy.
The newest version!
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.assertions.untilAsserted
import pl.allegro.tech.servicemesh.envoycontrol.config.RandomConfigFile
import pl.allegro.tech.servicemesh.envoycontrol.config.consul.ConsulExtension
import pl.allegro.tech.servicemesh.envoycontrol.config.service.EchoServiceExtension
import pl.allegro.tech.servicemesh.envoycontrol.config.envoy.EnvoyExtension
import pl.allegro.tech.servicemesh.envoycontrol.config.envoycontrol.EnvoyControlExtension

class OutlierDetectionTest {

    companion object {

        @JvmField
        @RegisterExtension
        val consul = ConsulExtension()

        @JvmField
        @RegisterExtension
        val envoyControl = EnvoyControlExtension(consul, mapOf(
            "envoy-control.envoy.snapshot.cluster-outlier-detection.enabled" to true
        ))

        @JvmField
        @RegisterExtension
        val healthyService = EchoServiceExtension()

        @JvmField
        @RegisterExtension
        val unhealthyService = EchoServiceExtension()

        // necessary since Envoy 1.28.0 to keep the same behaviour (https://www.envoyproxy.io/docs/envoy/latest/version_history/v1.28/v1.28.0)
        private val outlierEjectionConfig = """
            layered_runtime:
              layers:
              - name: static_layer
                static_layer:
                  envoy.reloadable_features.check_mep_on_first_eject: false
        """.trimIndent()

        @JvmField
        @RegisterExtension
        val envoy = EnvoyExtension(envoyControl, config = RandomConfigFile.copy(configOverride = outlierEjectionConfig))
    }

    @Test
    fun `should not send requests to instance when outlier check failed`() {
        // given
        val unhealthyIp = unhealthyService.container().ipAddress()
        consul.server.operations.registerService(healthyService, name = "echo")
        consul.server.operations.registerService(unhealthyService, name = "echo")
        unhealthyService.container().stop()

        untilAsserted {
            // when
            envoy.egressOperations.callService("echo")

            // then
            assertThat(hasOutlierCheckFailed(cluster = "echo", unhealthyIp = unhealthyIp)).isTrue()
        }

        // when
        repeat(times = 10) {
            // when
            val response = envoy.egressOperations.callService("echo")

            // then
            assertThat(response).isOk().isFrom(healthyService)
        }
    }

    fun hasOutlierCheckFailed(cluster: String, unhealthyIp: String): Boolean {
        return envoy.container.admin()
            .hostStatus(cluster, unhealthyIp)
            ?.healthStatus
            ?.failedOutlierCheck
            ?: false
    }
}