• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 <string>
6 
7 #include "base/command_line.h"
8 #include "base/environment.h"
9 #include "base/file_path.h"
10 #include "base/memory/scoped_ptr.h"
11 #include "base/process_util.h"
12 #include "base/stringprintf.h"
13 #include "base/string_util.h"
14 #include "net/test/python_utils.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16 
TEST(PythonUtils,Append)17 TEST(PythonUtils, Append) {
18   const FilePath::CharType kAppendDir1[] =
19       FILE_PATH_LITERAL("test/path_append1");
20   const FilePath::CharType kAppendDir2[] =
21       FILE_PATH_LITERAL("test/path_append2");
22 
23   scoped_ptr<base::Environment> env(base::Environment::Create());
24 
25   std::string python_path;
26   FilePath append_path1(kAppendDir1);
27   FilePath append_path2(kAppendDir2);
28 
29   // Get a clean start
30   env->UnSetVar(kPythonPathEnv);
31 
32   // Append the path
33   AppendToPythonPath(append_path1);
34   env->GetVar(kPythonPathEnv, &python_path);
35   ASSERT_EQ(python_path, "test/path_append1");
36 
37   // Append the safe path again, nothing changes
38   AppendToPythonPath(append_path2);
39   env->GetVar(kPythonPathEnv, &python_path);
40 #if defined(OS_WIN)
41   ASSERT_EQ(std::string("test/path_append1;test/path_append2"), python_path);
42 #elif defined(OS_POSIX)
43   ASSERT_EQ(std::string("test/path_append1:test/path_append2"), python_path);
44 #endif
45 }
46 
TEST(PythonUtils,PythonRunTime)47 TEST(PythonUtils, PythonRunTime) {
48   FilePath dir;
49   EXPECT_TRUE(GetPythonRunTime(&dir));
50 
51   // Run a python command to print a string and make sure the output is what
52   // we want.
53   CommandLine cmd_line(dir);
54   cmd_line.AppendArg("-c");
55   std::string input("PythonUtilsTest");
56   std::string python_cmd = StringPrintf("print '%s';", input.c_str());
57   cmd_line.AppendArg(python_cmd);
58   std::string output;
59   EXPECT_TRUE(base::GetAppOutput(cmd_line, &output));
60   TrimWhitespace(output, TRIM_TRAILING, &output);
61   EXPECT_EQ(input, output);
62 }
63