1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Copyright 2017 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <string>
#include "src/tools/launcher/util/launcher_util.h"
#include "gtest/gtest.h"
namespace bazel {
namespace launcher {
using std::getenv;
using std::ios;
using std::ofstream;
using std::string;
class LaunchUtilTest : public ::testing::Test {
protected:
LaunchUtilTest() {}
virtual ~LaunchUtilTest() {}
void SetUp() override {
char* tmpdir = getenv("TEST_TMPDIR");
if (tmpdir != NULL) {
test_tmpdir = string(tmpdir);
} else {
tmpdir = getenv("TEMP");
ASSERT_FALSE(tmpdir == NULL);
test_tmpdir = string(tmpdir);
}
}
void TearDown() override {}
string GetTmpDir() { return this->test_tmpdir; }
// Create an empty file at path
static void CreateEmptyFile(const string& path) {
ofstream file_stream(path.c_str(), ios::out | ios::binary);
file_stream.put('\0');
}
private:
string test_tmpdir;
};
TEST_F(LaunchUtilTest, GetBinaryPathWithoutExtensionTest) {
ASSERT_EQ("foo", GetBinaryPathWithoutExtension("foo.exe"));
ASSERT_EQ("foo.sh", GetBinaryPathWithoutExtension("foo.sh.exe"));
ASSERT_EQ("foo.sh", GetBinaryPathWithoutExtension("foo.sh"));
}
TEST_F(LaunchUtilTest, GetBinaryPathWithExtensionTest) {
ASSERT_EQ("foo.exe", GetBinaryPathWithExtension("foo"));
ASSERT_EQ("foo.sh.exe", GetBinaryPathWithExtension("foo.sh.exe"));
ASSERT_EQ("foo.sh.exe", GetBinaryPathWithExtension("foo.sh"));
}
TEST_F(LaunchUtilTest, GetEscapedArgumentTest) {
ASSERT_EQ("foo", GetEscapedArgument("foo"));
ASSERT_EQ("\"foo bar\"", GetEscapedArgument("foo bar"));
ASSERT_EQ("\"\\\"foo bar\\\"\"", GetEscapedArgument("\"foo bar\""));
ASSERT_EQ("foo\\\\bar", GetEscapedArgument("foo\\bar"));
ASSERT_EQ("foo\\\"bar", GetEscapedArgument("foo\"bar"));
ASSERT_EQ("C:\\\\foo\\\\bar\\\\", GetEscapedArgument("C:\\foo\\bar\\"));
ASSERT_EQ("\"C:\\\\foo foo\\\\bar\\\\\"",
GetEscapedArgument("C:\\foo foo\\bar\\"));
}
TEST_F(LaunchUtilTest, DoesFilePathExistTest) {
string file1 = GetTmpDir() + "/foo";
string file2 = GetTmpDir() + "/bar";
CreateEmptyFile(file1);
ASSERT_TRUE(DoesFilePathExist(file1));
ASSERT_FALSE(DoesFilePathExist(file2));
}
} // namespace launcher
} // namespace bazel
|
