Add a unit test for CommandLine that makes sure that GetProgram will not retu...

base/command_line_unittest.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/file_path.h"
@@ skipping 161 matching lines @@
   c1.AppendSwitch("switch1");
   CommandLine c2(FilePath(FILE_PATH_LITERAL("Program2")));
   c2.AppendSwitch("switch2");
 
   c1.AppendArguments(c2, true);
   EXPECT_EQ(c1.GetProgram().value(), c2.GetProgram().value());
   EXPECT_TRUE(c1.HasSwitch("switch1"));
   EXPECT_TRUE(c1.HasSwitch("switch2"));
 }
 
+#if defined(OS_WIN)
+// Make sure that the program part of a command line is always quoted.
+// Only makes sense on Windows.
+TEST(CommandLineTest, ProgramQuotes) {
+  const FilePath kProgram(FILE_PATH_LITERAL("Program"));
+  const FilePath kProgramQuoted(FILE_PATH_LITERAL("\"Program\""));
+
+  // Check that quotes are not returned from GetProgram().
+  CommandLine cl1(kProgram);
+  EXPECT_EQ(kProgram.value(), cl1.GetProgram().value());
+
+  CommandLine cl2(kProgramQuoted);
+  EXPECT_EQ(kProgram.value(), cl2.GetProgram().value());
+
+  CommandLine cl3(CommandLine::FromString(kProgramQuoted.value()));
+  EXPECT_EQ(kProgram.value(), cl3.GetProgram().value());
+
+  // Verify that in the command line string, the program part is always quoted.
+  CommandLine* check[] = {&cl1, &cl2, &cl3};
+  for (int i = 0; i < arraysize(check); ++i) {
+    CommandLine::StringType c(check[i]->command_line_string());
+    CommandLine::StringType p(check[i]->GetProgram().value());
+    EXPECT_EQ('"', c[0]);
+    EXPECT_EQ(p, c.substr(1, p.length()));
+    EXPECT_EQ('"', c[p.length() + 1]);
+  }
+}
+#endif