| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "blimp/engine/app/blimp_engine_config.h" | |
| 6 | |
| 7 #include <memory> | |
| 8 #include <string> | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "base/command_line.h" | |
| 12 #include "base/files/file_util.h" | |
| 13 #include "base/files/scoped_temp_dir.h" | |
| 14 #include "base/strings/stringprintf.h" | |
| 15 #include "blimp/common/switches.h" | |
| 16 #include "testing/gtest/include/gtest/gtest.h" | |
| 17 | |
| 18 namespace blimp { | |
| 19 namespace engine { | |
| 20 namespace { | |
| 21 | |
| 22 // Reference client token. | |
| 23 static const char kTestClientAuthToken[] = "Reference client token"; | |
| 24 | |
| 25 class BlimpEngineConfigTest : public testing::Test { | |
| 26 protected: | |
| 27 void SetUp() override { | |
| 28 // Create a temporary directory and populate it with all of the switches | |
| 29 // If a test requires a switch's file to be missing, call | |
| 30 // RemoveFileForSwitch(). | |
| 31 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
| 32 CreateFileForSwitch(kClientAuthTokenPath, kTestClientAuthToken); | |
| 33 } | |
| 34 | |
| 35 // Creates a file in the temp directory for a given filepath switch. | |
| 36 void CreateFileForSwitch(const std::string& filepath_switch, | |
| 37 const std::string& data) const { | |
| 38 ASSERT_TRUE(base::WriteFile(GetFilepathForSwitch(filepath_switch), | |
| 39 data.c_str(), data.size())); | |
| 40 } | |
| 41 | |
| 42 // Removes the associated file for a given filepath switch. | |
| 43 void RemoveFileForSwitch(const std::string& filepath_switch) const { | |
| 44 base::DeleteFile(GetFilepathForSwitch(filepath_switch), false); | |
| 45 } | |
| 46 | |
| 47 // Creates and returns a CommandLine object with specified filepath switches. | |
| 48 base::CommandLine CreateCommandLine( | |
| 49 const std::vector<std::string>& filepath_switches) { | |
| 50 base::CommandLine::StringVector cmd_vec = {"dummy_program"}; | |
| 51 for (const std::string& filepath_switch : filepath_switches) { | |
| 52 cmd_vec.push_back(base::StringPrintf( | |
| 53 "--%s=%s", filepath_switch.c_str(), | |
| 54 GetFilepathForSwitch(filepath_switch).AsUTF8Unsafe().c_str())); | |
| 55 } | |
| 56 return base::CommandLine(cmd_vec); | |
| 57 } | |
| 58 | |
| 59 base::FilePath GetFilepathForSwitch( | |
| 60 const std::string& filepath_switch) const { | |
| 61 return temp_dir_.GetPath().Append(filepath_switch); | |
| 62 } | |
| 63 | |
| 64 const std::vector<std::string> all_filepath_switches_ = { | |
| 65 kClientAuthTokenPath}; | |
| 66 | |
| 67 base::ScopedTempDir temp_dir_; | |
| 68 }; | |
| 69 | |
| 70 TEST_F(BlimpEngineConfigTest, ClientAuthTokenCorrect) { | |
| 71 auto cmd_line = CreateCommandLine(all_filepath_switches_); | |
| 72 auto engine_config = BlimpEngineConfig::Create(cmd_line); | |
| 73 EXPECT_NE(nullptr, engine_config); | |
| 74 EXPECT_EQ(kTestClientAuthToken, engine_config->client_auth_token()); | |
| 75 } | |
| 76 | |
| 77 TEST_F(BlimpEngineConfigTest, ClientAuthTokenEmpty) { | |
| 78 RemoveFileForSwitch(kClientAuthTokenPath); | |
| 79 CreateFileForSwitch(kClientAuthTokenPath, " "); | |
| 80 auto cmd_line = CreateCommandLine(all_filepath_switches_); | |
| 81 auto engine_config = BlimpEngineConfig::Create(cmd_line); | |
| 82 EXPECT_EQ(nullptr, engine_config); | |
| 83 } | |
| 84 | |
| 85 } // namespace | |
| 86 } // namespace engine | |
| 87 } // namespace blimp | |
| OLD | NEW |