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
17TEST(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
47TEST(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