1// Copyright (c) 2011 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 "chrome/common/switch_utils.h" 6 7#include "base/basictypes.h" 8#include "base/command_line.h" 9#include "base/files/file_path.h" 10#include "testing/gtest/include/gtest/gtest.h" 11 12TEST(SwitchUtilsTest, RemoveSwitches) { 13 const CommandLine::CharType* argv[] = { 14 FILE_PATH_LITERAL("program"), 15 FILE_PATH_LITERAL("--app=http://www.google.com/"), 16 FILE_PATH_LITERAL("--force-first-run"), 17 FILE_PATH_LITERAL("--make-default-browser"), 18 FILE_PATH_LITERAL("--foo"), 19 FILE_PATH_LITERAL("--bar")}; 20 CommandLine cmd_line(arraysize(argv), argv); 21 EXPECT_FALSE(cmd_line.GetCommandLineString().empty()); 22 23 std::map<std::string, CommandLine::StringType> switches = 24 cmd_line.GetSwitches(); 25 EXPECT_EQ(5U, switches.size()); 26 27 switches::RemoveSwitchesForAutostart(&switches); 28 EXPECT_EQ(2U, switches.size()); 29 EXPECT_TRUE(cmd_line.HasSwitch("foo")); 30 EXPECT_TRUE(cmd_line.HasSwitch("bar")); 31} 32 33#if defined(OS_WIN) 34TEST(SwitchUtilsTest, RemoveSwitchesFromString) { 35 // All these command line args (except foo and bar) will 36 // be removed after RemoveSwitchesForAutostart: 37 CommandLine cmd_line = CommandLine::FromString( 38 L"program" 39 L" --app=http://www.google.com/" 40 L" --force-first-run" 41 L" --make-default-browser" 42 L" --foo" 43 L" --bar"); 44 EXPECT_FALSE(cmd_line.GetCommandLineString().empty()); 45 46 std::map<std::string, CommandLine::StringType> switches = 47 cmd_line.GetSwitches(); 48 EXPECT_EQ(5U, switches.size()); 49 50 switches::RemoveSwitchesForAutostart(&switches); 51 EXPECT_EQ(2U, switches.size()); 52 EXPECT_TRUE(cmd_line.HasSwitch("foo")); 53 EXPECT_TRUE(cmd_line.HasSwitch("bar")); 54} 55#endif 56