From b1d63b351483ba0c54f6cce9bced2aa7883c0772 Mon Sep 17 00:00:00 2001 From: Athish Pranav D Date: Mon, 9 Sep 2024 23:30:48 +0530 Subject: [PATCH] Added UTs Signed-off-by: Athish Pranav D --- .../filters/http/common/jwks_fetcher_test.cc | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/test/extensions/filters/http/common/jwks_fetcher_test.cc b/test/extensions/filters/http/common/jwks_fetcher_test.cc index 59184dda5b0e..571b61e23208 100644 --- a/test/extensions/filters/http/common/jwks_fetcher_test.cc +++ b/test/extensions/filters/http/common/jwks_fetcher_test.cc @@ -81,6 +81,28 @@ TEST_F(JwksFetcherTest, TestGetSuccess) { fetcher_->fetch(parent_span_, receiver); } +TEST_F(JwksFetcherTest, TestMessageHeader) { + // Setup + setupFetcher(config); + MockUpstream mock_pubkey(mock_factory_ctx_.server_factory_context_.cluster_manager_, "200", + publicKey); + MockJwksReceiver receiver; + + // Expectations for message + EXPECT_CALL(mock_factory_ctx_.server_factory_context_.cluster_manager_.thread_local_cluster_ + .async_client_, + send_(_, _, _)) + .WillOnce(Invoke([](Http::RequestMessagePtr& message, Http::AsyncClient::Callbacks&, + const Http::AsyncClient::RequestOptions&) -> Http::AsyncClient::Request* { + EXPECT_EQ(message->headers().getUserAgentValue(), + Http::Headers::get().UserAgentValues.GoBrowser); + return nullptr; + })); + + // Act + fetcher_->fetch(parent_span_, receiver); +} + TEST_F(JwksFetcherTest, TestGet400) { // Setup setupFetcher(config);