15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/switch_utils.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(SwitchUtilsTest, RemoveSwitches) { 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CommandLine::CharType* argv[] = { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("program"), 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("--app=http://www.google.com/"), 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) FILE_PATH_LITERAL("--force-first-run"), 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("--make-default-browser"), 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("--foo"), 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FILE_PATH_LITERAL("--bar")}; 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CommandLine cmd_line(arraysize(argv), argv); 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(cmd_line.GetCommandLineString().empty()); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::map<std::string, CommandLine::StringType> switches = 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmd_line.GetSwitches(); 2590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_EQ(5U, switches.size()); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switches::RemoveSwitchesForAutostart(&switches); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2U, switches.size()); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cmd_line.HasSwitch("foo")); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cmd_line.HasSwitch("bar")); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(SwitchUtilsTest, RemoveSwitchesFromString) { 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // All these command line args (except foo and bar) will 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // be removed after RemoveSwitchesForAutostart: 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CommandLine cmd_line = CommandLine::FromString( 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"program" 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L" --app=http://www.google.com/" 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L" --force-first-run" 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L" --make-default-browser" 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L" --foo" 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L" --bar"); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(cmd_line.GetCommandLineString().empty()); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::map<std::string, CommandLine::StringType> switches = 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cmd_line.GetSwitches(); 4890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_EQ(5U, switches.size()); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switches::RemoveSwitchesForAutostart(&switches); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2U, switches.size()); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cmd_line.HasSwitch("foo")); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(cmd_line.HasSwitch("bar")); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 56