blob: 9ca7eb4e7ae94bdc218cf54c40df078bb9d79559 [file] [log] [blame] [edit]
// Copyright 2015 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 <fcntl.h>
#include <stddef.h>
#include <stdlib.h>
#include <unistd.h>
#include <string>
#include <utility>
#include <vector>
#include "src/main/cpp/blaze_util.h"
#include "src/main/cpp/blaze_util_platform.h"
#include "src/main/cpp/util/file.h"
#include "googlemock/include/gmock/gmock.h"
#include "googletest/include/gtest/gtest.h"
namespace blaze {
using std::string;
using ::testing::HasSubstr;
class BlazeUtilTest : public ::testing::Test {
protected:
BlazeUtilTest() {
}
virtual ~BlazeUtilTest() {
}
static void ForkAndWrite(int fds[], const string& input1,
const string& input2) {
int r = fork();
if (r == 0) {
close(fds[0]);
write(fds[1], input1.c_str(), input1.size());
usleep(500); // sleep for 50ms
write(fds[1], input2.c_str(), input2.size());
close(fds[1]);
exit(0);
} else if (r < 0) {
perror("fork()");
FAIL();
} else {
close(fds[1]);
}
}
static int WriteFileDescriptor2(const string& input1, const string& input2) {
// create a fd for the input string
int fds[2];
if (pipe(fds) == -1) {
return -1;
}
if (fcntl(fds[0], F_SETFL, O_NONBLOCK) == -1
|| fcntl(fds[1], F_SETFL, O_NONBLOCK) == -1) {
return -1;
}
if (!input2.empty()) {
ForkAndWrite(fds, input1, input2);
} else {
write(fds[1], input1.c_str(), input1.size());
close(fds[1]);
}
return fds[0];
}
static void AssertReadFrom2(const string& input1, const string& input2) {
int fd = WriteFileDescriptor2(input1, input2);
if (fd < 0) {
FAIL() << "Unable to create a pipe!";
} else {
string result;
bool success = blaze_util::ReadFrom(fd, &result);
close(fd);
if (!success) {
perror("ReadFrom");
FAIL() << "Unable to read file descriptor!";
} else {
ASSERT_EQ(input1 + input2, result);
}
}
}
static void AssertReadFrom(string input) { AssertReadFrom2(input, ""); }
void ReadFromTest() const {
AssertReadFrom(
"DummyJDK Blabla\n"
"More DummyJDK Blabla\n");
AssertReadFrom(
"dummyjdk version \"1.42.qual\"\n"
"DummyJDK Blabla\n"
"More DummyJDK Blabla\n");
AssertReadFrom2("first_line\n", "second line version \"1.4.2_0\"\n");
}
};
TEST_F(BlazeUtilTest, ReadFrom) { ReadFromTest(); }
TEST_F(BlazeUtilTest, TestSearchNullaryEmptyCase) {
ASSERT_FALSE(SearchNullaryOption({}, "flag", false));
ASSERT_TRUE(SearchNullaryOption({}, "flag", true));
}
TEST_F(BlazeUtilTest, TestSearchUnaryEmptyCase) {
ASSERT_STREQ(nullptr, SearchUnaryOption({}, "--flag", false));
}
TEST_F(BlazeUtilTest, TestSearchNullaryForEmpty) {
ASSERT_TRUE(SearchNullaryOption({"bazel", "build", ":target"}, "", true));
ASSERT_FALSE(SearchNullaryOption({"bazel", "build", ":target"}, "", false));
}
TEST_F(BlazeUtilTest, TestSearchNullaryForFlagNotPresent) {
ASSERT_FALSE(SearchNullaryOption({"bazel", "build", ":target"},
"flag", false));
ASSERT_TRUE(SearchNullaryOption({"bazel", "build", ":target"},
"flag", true));
}
TEST_F(BlazeUtilTest, TestSearchNullaryStartupOption) {
ASSERT_TRUE(SearchNullaryOption({"bazel", "--flag", "build", ":target"},
"flag", false));
ASSERT_TRUE(SearchNullaryOption({"bazel", "--flag", "build", ":target"},
"flag", true));
}
TEST_F(BlazeUtilTest, TestSearchNullaryStartupOptionWithEquals) {
ASSERT_DEATH(SearchNullaryOption(
{"bazel", "--flag=value", "build", ":target"}, "flag", false),
"In argument '--flag=value': option "
"'--flag' does not take a value");
}
TEST_F(BlazeUtilTest, TestSearchNullaryCommandOption) {
ASSERT_TRUE(SearchNullaryOption({"bazel", "build", ":target", "--flag"},
"flag", false));
}
TEST_F(BlazeUtilTest, TestSearchNullarySkipsAfterDashDash) {
ASSERT_FALSE(SearchNullaryOption(
{"bazel", "build", ":target", "--", "--flag"}, "flag", false));
}
TEST_F(BlazeUtilTest, TestSearchNullarySucceedsWithEqualsAndDashDash) {
ASSERT_FALSE(SearchNullaryOption(
{"bazel", "build", ":target", "--", "--flag=value"}, "flag", false));
}
TEST_F(BlazeUtilTest, TestSearchNullaryLastFlagWins) {
ASSERT_FALSE(SearchNullaryOption(
{"bazel", "--flag", "--noflag", "build"}, "flag", false));
ASSERT_FALSE(SearchNullaryOption(
{"bazel", "--flag", "--noflag", "build"}, "flag", true));
}
TEST_F(BlazeUtilTest, TestSearchUnaryForEmpty) {
ASSERT_STREQ(nullptr, SearchUnaryOption({"bazel", "build", ":target"}, "",
false));
}
TEST_F(BlazeUtilTest, TestSearchUnaryFlagNotPresent) {
ASSERT_STREQ(nullptr,
SearchUnaryOption({"bazel", "build", ":target"}, "--flag",
false));
}
TEST_F(BlazeUtilTest, TestSearchUnaryStartupOptionWithEquals) {
ASSERT_STREQ("value",
SearchUnaryOption({"bazel", "--flag=value", "build", ":target"},
"--flag", false));
}
TEST_F(BlazeUtilTest, TestSearchUnaryStartupOptionWithoutEquals) {
ASSERT_STREQ("value",
SearchUnaryOption(
{"bazel", "--flag", "value", "build", ":target"}, "--flag",
false));
}
TEST_F(BlazeUtilTest, TestSearchUnaryCommandOptionWithEquals) {
ASSERT_STREQ("value",
SearchUnaryOption(
{"bazel", "build", ":target", "--flag", "value"}, "--flag",
false));
}
TEST_F(BlazeUtilTest, TestSearchUnaryCommandOptionWithoutEquals) {
ASSERT_STREQ("value",
SearchUnaryOption(
{"bazel", "build", ":target", "--flag=value"}, "--flag",
false));
}
TEST_F(BlazeUtilTest, TestSearchUnarySkipsAfterDashDashWithEquals) {
ASSERT_STREQ(nullptr,
SearchUnaryOption(
{"bazel", "build", ":target", "--", "--flag", "value"},
"--flag", false));
}
TEST_F(BlazeUtilTest, TestSearchUnarySkipsAfterDashDashWithoutEquals) {
ASSERT_STREQ(nullptr,
SearchUnaryOption(
{"bazel", "build", ":target", "--", "--flag=value"},
"--flag", false));
}
TEST_F(BlazeUtilTest, TestSearchUnaryCommandOptionWarnsAboutDuplicates) {
testing::internal::CaptureStderr();
for (int i = 0; i < 2; ++i) {
bool warn_if_dupe = (i == 0);
ASSERT_STREQ("v1",
SearchUnaryOption(
{"foo", "--flag", "v1", "bar", "--flag=v2"}, "--flag",
warn_if_dupe));
if (warn_if_dupe) {
std::string stderr_output = testing::internal::GetCapturedStderr();
EXPECT_THAT(stderr_output, HasSubstr("--flag is given more than once"));
}
}
}
void assert_equal_vector_char_pointer(std::vector<std::string> expected,
std::vector<std::string> actual) {
ASSERT_THAT(actual, testing::ContainerEq(expected));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryForEmpty) {
assert_equal_vector_char_pointer(
{}, GetAllUnaryOptionValues({"bazel", "build", ":target"}, ""));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryFlagNotPresent) {
assert_equal_vector_char_pointer(
{}, GetAllUnaryOptionValues({"bazel", "build", ":target"}, "--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithEquals) {
assert_equal_vector_char_pointer(
{"value"}, GetAllUnaryOptionValues(
{"bazel", "--flag=value", "build", ":target"}, "--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithEquals2) {
assert_equal_vector_char_pointer(
{"value1", "value2"},
GetAllUnaryOptionValues(
{"bazel", "--flag=value1", "--flag=value2", "build", ":target"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithRepeatingFlag) {
assert_equal_vector_char_pointer(
{"--flag"}, GetAllUnaryOptionValues({"bazel", "--flag", "--flag",
"value1", "build", ":target"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithRepeatingFlagOptions) {
assert_equal_vector_char_pointer(
{"--flag"}, GetAllUnaryOptionValues(
{"bazel", "--flag", "--flag", "value1"}, "--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionValuesWithEquals) {
assert_equal_vector_char_pointer(
{"--flag", "value1"},
GetAllUnaryOptionValues({"bazel", "--flag=--flag", "--flag", "value1"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithEquals3) {
assert_equal_vector_char_pointer(
{"value1", "value2", "value3"},
GetAllUnaryOptionValues({"bazel", "--flag=value1", "--flag=value2",
"--flag=value3", "build", ":target"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithoutEquals) {
assert_equal_vector_char_pointer(
{"value"},
GetAllUnaryOptionValues({"bazel", "--flag", "value", "build", ":target"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryOptionWithoutEquals2) {
assert_equal_vector_char_pointer(
{"value1", "value2"},
GetAllUnaryOptionValues(
{"bazel", "--flag", "value1", "--flag", "value2", "build", ":target"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryCommandOptionWithEquals) {
assert_equal_vector_char_pointer(
{"value"},
GetAllUnaryOptionValues({"bazel", "build", ":target", "--flag", "value"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryCommandOptionWithEquals2) {
assert_equal_vector_char_pointer(
{"value1", "value2"},
GetAllUnaryOptionValues(
{"bazel", "build", ":target", "--flag", "value1", "--flag", "value2"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryCommandOptionWithoutEquals) {
assert_equal_vector_char_pointer(
{"value"}, GetAllUnaryOptionValues(
{"bazel", "build", ":target", "--flag=value"}, "--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryCommandOptionWithoutEquals2) {
assert_equal_vector_char_pointer(
{"value1", "value2"},
GetAllUnaryOptionValues(
{"bazel", "build", ":target", "--flag=value1", "--flag=value2"},
"--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnarySkipsAfterDashDashWithEquals) {
assert_equal_vector_char_pointer(
{},
GetAllUnaryOptionValues(
{"bazel", "build", ":target", "--", "--flag", "value"}, "--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnarySkipsAfterDashDashWithoutEquals) {
assert_equal_vector_char_pointer(
{}, GetAllUnaryOptionValues(
{"bazel", "build", ":target", "--", "--flag=value"}, "--flag"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryCommandOptionWithIgnoreAfter) {
assert_equal_vector_char_pointer(
{"value1", "/dev/null"},
GetAllUnaryOptionValues({"bazel", "build", ":target", "--flag", "value1",
"--flag", "/dev/null", "--flag", "value3"},
"--flag", "/dev/null"));
}
TEST_F(BlazeUtilTest, TestSearchAllUnaryCommandOptionWithIgnoreAfterDevNull) {
assert_equal_vector_char_pointer(
{"/dev/null"}, GetAllUnaryOptionValues(
{"bazel", "build", ":target", "--flag", "/dev/null",
"--flag", "value2", "--flag", "value3"},
"--flag", "/dev/null"));
}
} // namespace blaze