// Copyright 2023 The Chromium Authors // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "chrome/browser/enterprise/profile_management/saml_response_parser.h" #include #include #include "base/base64.h" #include "base/containers/flat_map.h" #include "base/run_loop.h" #include "base/strings/stringprintf.h" #include "base/test/bind.h" #include "base/test/task_environment.h" #include "services/data_decoder/public/cpp/test_support/in_process_data_decoder.h" #include "testing/gmock/include/gmock/gmock.h" #include "testing/gtest/include/gtest/gtest.h" namespace profile_management { using testing::_; namespace { const char kValidSAMLResponse[] = R"( https://issuer.com/ https://issuer.com/ user@domain.com attributevalue anotherattributevalue uselessattributevalue )"; const char kHTMLTemplate[] = R"( )"; } // namespace class SAMLResponseParserTest : public testing::Test { public: SAMLResponseParserTest() = default; ~SAMLResponseParserTest() override = default; private: base::test::SingleThreadTaskEnvironment task_environment_; data_decoder::test::InProcessDataDecoder in_process_data_decoder_; }; TEST_F(SAMLResponseParserTest, RetrievesNoAttributesWithEmptyResponse) { std::string response = ""; base::flat_map attributes; base::RunLoop loop; SAMLResponseParser body_reader( std::vector{"attribute", "notattribute", "anotherattribute"}, response, base::BindLambdaForTesting( [&attributes, &loop](const base::flat_map& result) { attributes = std::move(result); loop.Quit(); })); loop.Run(); EXPECT_TRUE(attributes.empty()); } TEST_F(SAMLResponseParserTest, RetrievesNoAttributesWithEmptySAMLResponse) { std::string response = base::StringPrintf(kHTMLTemplate, ""); base::flat_map attributes; base::RunLoop loop; SAMLResponseParser body_reader( std::vector{"attribute", "notattribute", "anotherattribute"}, response, base::BindLambdaForTesting( [&attributes, &loop](const base::flat_map& result) { attributes = std::move(result); loop.Quit(); })); loop.Run(); EXPECT_TRUE(attributes.empty()); } TEST_F(SAMLResponseParserTest, RetrievesSpecifiedAttributesWithValidResponse) { std::string encoded_saml_response = base::Base64Encode(kValidSAMLResponse); std::string response = base::StringPrintf(kHTMLTemplate, encoded_saml_response.c_str()); base::flat_map attributes; base::RunLoop loop; SAMLResponseParser body_reader( std::vector{"attribute", "notattribute", "anotherattribute"}, response, base::BindLambdaForTesting( [&attributes, &loop](const base::flat_map& result) { attributes = std::move(result); loop.Quit(); })); loop.Run(); EXPECT_EQ(attributes["attribute"], "attributevalue"); EXPECT_EQ(attributes["anotherattribute"], "anotherattributevalue"); EXPECT_EQ(attributes.find("notattribute"), attributes.end()); EXPECT_EQ(attributes.find("uselessattribute"), attributes.end()); EXPECT_EQ(attributes[SAMLResponseParser::kDestinationUrl], "https://dest.test"); } } // namespace profile_management