1 // Copyright (c) 2010 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 <string>
6 #include <vector>
7
8 #include "base/basictypes.h"
9 #include "base/command_line.h"
10 #include "base/file_path.h"
11 #include "base/utf_string_conversions.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13
14 // To test Windows quoting behavior, we use a string that has some backslashes
15 // and quotes.
16 // Consider the command-line argument: q\"bs1\bs2\\bs3q\\\"
17 // Here it is with C-style escapes.
18 #define TRICKY_QUOTED L"q\\\"bs1\\bs2\\\\bs3q\\\\\\\""
19 // It should be parsed by Windows as: q"bs1\bs2\\bs3q\"
20 // Here that is with C-style escapes.
21 #define TRICKY L"q\"bs1\\bs2\\\\bs3q\\\""
22
TEST(CommandLineTest,CommandLineConstructor)23 TEST(CommandLineTest, CommandLineConstructor) {
24 #if defined(OS_WIN)
25 CommandLine cl = CommandLine::FromString(
26 L"program --foo= -bAr /Spaetzel=pierogi /Baz flim "
27 L"--other-switches=\"--dog=canine --cat=feline\" "
28 L"-spaetzle=Crepe -=loosevalue flan "
29 L"--input-translation=\"45\"--output-rotation "
30 L"--quotes=" TRICKY_QUOTED L" "
31 L"-- -- --not-a-switch "
32 L"\"in the time of submarines...\"");
33 EXPECT_FALSE(cl.command_line_string().empty());
34 #elif defined(OS_POSIX)
35 const char* argv[] = {"program", "--foo=", "-bar",
36 "-spaetzel=pierogi", "-baz", "flim",
37 "--other-switches=--dog=canine --cat=feline",
38 "-spaetzle=Crepe", "-=loosevalue", "flan",
39 "--input-translation=45--output-rotation",
40 "--", "--", "--not-a-switch",
41 "in the time of submarines..."};
42 CommandLine cl(arraysize(argv), argv);
43 #endif
44 EXPECT_FALSE(cl.HasSwitch("cruller"));
45 EXPECT_FALSE(cl.HasSwitch("flim"));
46 EXPECT_FALSE(cl.HasSwitch("program"));
47 EXPECT_FALSE(cl.HasSwitch("dog"));
48 EXPECT_FALSE(cl.HasSwitch("cat"));
49 EXPECT_FALSE(cl.HasSwitch("output-rotation"));
50 EXPECT_FALSE(cl.HasSwitch("not-a-switch"));
51 EXPECT_FALSE(cl.HasSwitch("--"));
52
53 EXPECT_EQ(FilePath(FILE_PATH_LITERAL("program")).value(),
54 cl.GetProgram().value());
55
56 EXPECT_TRUE(cl.HasSwitch("foo"));
57 EXPECT_TRUE(cl.HasSwitch("bar"));
58 EXPECT_TRUE(cl.HasSwitch("baz"));
59 EXPECT_TRUE(cl.HasSwitch("spaetzle"));
60 #if defined(OS_WIN)
61 EXPECT_TRUE(cl.HasSwitch("SPAETZLE"));
62 #endif
63 EXPECT_TRUE(cl.HasSwitch("other-switches"));
64 EXPECT_TRUE(cl.HasSwitch("input-translation"));
65 #if defined(OS_WIN)
66 EXPECT_TRUE(cl.HasSwitch("quotes"));
67 #endif
68
69 EXPECT_EQ("Crepe", cl.GetSwitchValueASCII("spaetzle"));
70 EXPECT_EQ("", cl.GetSwitchValueASCII("Foo"));
71 EXPECT_EQ("", cl.GetSwitchValueASCII("bar"));
72 EXPECT_EQ("", cl.GetSwitchValueASCII("cruller"));
73 EXPECT_EQ("--dog=canine --cat=feline", cl.GetSwitchValueASCII(
74 "other-switches"));
75 EXPECT_EQ("45--output-rotation", cl.GetSwitchValueASCII("input-translation"));
76 #if defined(OS_WIN)
77 EXPECT_EQ(TRICKY, cl.GetSwitchValueNative("quotes"));
78 #endif
79
80 const std::vector<CommandLine::StringType>& args = cl.args();
81 ASSERT_EQ(5U, args.size());
82
83 std::vector<CommandLine::StringType>::const_iterator iter = args.begin();
84 EXPECT_EQ(FILE_PATH_LITERAL("flim"), *iter);
85 ++iter;
86 EXPECT_EQ(FILE_PATH_LITERAL("flan"), *iter);
87 ++iter;
88 EXPECT_EQ(FILE_PATH_LITERAL("--"), *iter);
89 ++iter;
90 EXPECT_EQ(FILE_PATH_LITERAL("--not-a-switch"), *iter);
91 ++iter;
92 EXPECT_EQ(FILE_PATH_LITERAL("in the time of submarines..."), *iter);
93 ++iter;
94 EXPECT_TRUE(iter == args.end());
95 #if defined(OS_POSIX)
96 const std::vector<std::string>& argvec = cl.argv();
97
98 for (size_t i = 0; i < argvec.size(); i++) {
99 EXPECT_EQ(0, argvec[i].compare(argv[i]));
100 }
101 #endif
102 }
103
104 // Tests behavior with an empty input string.
TEST(CommandLineTest,EmptyString)105 TEST(CommandLineTest, EmptyString) {
106 #if defined(OS_WIN)
107 CommandLine cl = CommandLine::FromString(L"");
108 EXPECT_TRUE(cl.command_line_string().empty());
109 EXPECT_TRUE(cl.GetProgram().empty());
110 #elif defined(OS_POSIX)
111 CommandLine cl(0, NULL);
112 EXPECT_TRUE(cl.argv().empty());
113 #endif
114 EXPECT_TRUE(cl.args().empty());
115 }
116
117 // Test methods for appending switches to a command line.
TEST(CommandLineTest,AppendSwitches)118 TEST(CommandLineTest, AppendSwitches) {
119 std::string switch1 = "switch1";
120 std::string switch2 = "switch2";
121 std::string value = "value";
122 std::string switch3 = "switch3";
123 std::string value3 = "a value with spaces";
124 std::string switch4 = "switch4";
125 std::string value4 = "\"a value with quotes\"";
126 std::string switch5 = "quotes";
127 std::string value5 = WideToUTF8(TRICKY);
128
129 CommandLine cl(FilePath(FILE_PATH_LITERAL("Program")));
130
131 cl.AppendSwitch(switch1);
132 cl.AppendSwitchASCII(switch2, value);
133 cl.AppendSwitchASCII(switch3, value3);
134 cl.AppendSwitchASCII(switch4, value4);
135 cl.AppendSwitchASCII(switch5, value5);
136
137 EXPECT_TRUE(cl.HasSwitch(switch1));
138 EXPECT_TRUE(cl.HasSwitch(switch2));
139 EXPECT_EQ(value, cl.GetSwitchValueASCII(switch2));
140 EXPECT_TRUE(cl.HasSwitch(switch3));
141 EXPECT_EQ(value3, cl.GetSwitchValueASCII(switch3));
142 EXPECT_TRUE(cl.HasSwitch(switch4));
143 EXPECT_EQ(value4, cl.GetSwitchValueASCII(switch4));
144 EXPECT_TRUE(cl.HasSwitch(switch5));
145 EXPECT_EQ(value5, cl.GetSwitchValueASCII(switch5));
146
147 #if defined(OS_WIN)
148 EXPECT_EQ(L"\"Program\" "
149 L"--switch1 "
150 L"--switch2=value "
151 L"--switch3=\"a value with spaces\" "
152 L"--switch4=\"\\\"a value with quotes\\\"\" "
153 L"--quotes=\"" TRICKY_QUOTED L"\"",
154 cl.command_line_string());
155 #endif
156 }
157
158 // Tests that when AppendArguments is called that the program is set correctly
159 // on the target CommandLine object and the switches from the source
160 // CommandLine are added to the target.
TEST(CommandLineTest,AppendArguments)161 TEST(CommandLineTest, AppendArguments) {
162 CommandLine cl1(FilePath(FILE_PATH_LITERAL("Program")));
163 cl1.AppendSwitch("switch1");
164 cl1.AppendSwitchASCII("switch2", "foo");
165
166 CommandLine cl2(CommandLine::NO_PROGRAM);
167 cl2.AppendArguments(cl1, true);
168 EXPECT_EQ(cl1.GetProgram().value(), cl2.GetProgram().value());
169 EXPECT_EQ(cl1.command_line_string(), cl2.command_line_string());
170
171 CommandLine c1(FilePath(FILE_PATH_LITERAL("Program1")));
172 c1.AppendSwitch("switch1");
173 CommandLine c2(FilePath(FILE_PATH_LITERAL("Program2")));
174 c2.AppendSwitch("switch2");
175
176 c1.AppendArguments(c2, true);
177 EXPECT_EQ(c1.GetProgram().value(), c2.GetProgram().value());
178 EXPECT_TRUE(c1.HasSwitch("switch1"));
179 EXPECT_TRUE(c1.HasSwitch("switch2"));
180 }
181
182 #if defined(OS_WIN)
183 // Make sure that the program part of a command line is always quoted.
184 // This only makes sense on Windows and the test is basically here to guard
185 // against regressions.
TEST(CommandLineTest,ProgramQuotes)186 TEST(CommandLineTest, ProgramQuotes) {
187 const FilePath kProgram(L"Program");
188
189 // Check that quotes are not returned from GetProgram().
190 CommandLine cl(kProgram);
191 EXPECT_EQ(kProgram.value(), cl.GetProgram().value());
192
193 // Verify that in the command line string, the program part is always quoted.
194 CommandLine::StringType cmd(cl.command_line_string());
195 CommandLine::StringType program(cl.GetProgram().value());
196 EXPECT_EQ('"', cmd[0]);
197 EXPECT_EQ(program, cmd.substr(1, program.length()));
198 EXPECT_EQ('"', cmd[program.length() + 1]);
199 }
200 #endif
201