blob: b162cedadbc8584aa2852080413bf3d1dfa48d56 [file] [log] [blame]
// Copyright 2014 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 <stddef.h>
#include "gn/c_include_iterator.h"
#include "gn/input_file.h"
#include "gn/location.h"
#include "util/test/test.h"
namespace {
bool RangeIs(const LocationRange& range,
int line,
int begin_char,
int end_char) {
return range.begin().line_number() == line &&
range.end().line_number() == line &&
range.begin().column_number() == begin_char &&
range.end().column_number() == end_char;
}
} // namespace
TEST(CIncludeIterator, Basic) {
std::string buffer;
buffer.append("// Some comment\n");
buffer.append("\n");
buffer.append("#include \"foo/bar.h\"\n");
buffer.append("\n");
buffer.append("#include <stdio.h>\n");
buffer.append("\n");
buffer.append(" #include \"foo/baz.h\"\n"); // Leading whitespace
buffer.append("#include \"la/deda.h\"\n");
// Line annotated with "// nogncheck"
buffer.append("#include \"should_be_skipped.h\" // nogncheck\n");
buffer.append("#import \"weird_mac_import.h\"\n");
buffer.append("\n");
buffer.append("void SomeCode() {\n");
InputFile file(SourceFile("//foo.cc"));
file.SetContents(buffer);
CIncludeIterator iter(&file);
IncludeStringWithLocation include;
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("foo/bar.h", include.contents);
EXPECT_TRUE(RangeIs(include.location, 3, 11, 20)) << include.location.begin().Describe(true);
EXPECT_FALSE(include.system_style_include);
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("stdio.h", include.contents);
EXPECT_TRUE(RangeIs(include.location, 5, 11, 18)) << include.location.begin().Describe(true);
EXPECT_TRUE(include.system_style_include);
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("foo/baz.h", include.contents);
EXPECT_TRUE(RangeIs(include.location, 7, 12, 21)) << include.location.begin().Describe(true);
EXPECT_FALSE(include.system_style_include);
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("la/deda.h", include.contents);
EXPECT_TRUE(RangeIs(include.location, 8, 11, 20)) << include.location.begin().Describe(true);
EXPECT_FALSE(include.system_style_include);
// The line annotated with "nogncheck" should be skipped.
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("weird_mac_import.h", include.contents);
EXPECT_TRUE(RangeIs(include.location, 10, 10, 28)) << include.location.begin().Describe(true);
EXPECT_FALSE(include.system_style_include);
EXPECT_FALSE(iter.GetNextIncludeString(&include));
}
// Tests that we don't search for includes indefinitely.
TEST(CIncludeIterator, GiveUp) {
std::string buffer;
for (size_t i = 0; i < 1000; i++)
buffer.append("x\n");
buffer.append("#include \"foo/bar.h\"\n");
InputFile file(SourceFile("//foo.cc"));
file.SetContents(buffer);
IncludeStringWithLocation include;
CIncludeIterator iter(&file);
EXPECT_FALSE(iter.GetNextIncludeString(&include));
EXPECT_TRUE(include.contents.empty());
}
// Don't count blank lines, comments, and preprocessor when giving up.
TEST(CIncludeIterator, DontGiveUp) {
std::string buffer;
for (size_t i = 0; i < 1000; i++)
buffer.push_back('\n');
for (size_t i = 0; i < 1000; i++)
buffer.append("// comment\n");
for (size_t i = 0; i < 1000; i++)
buffer.append("#preproc\n");
buffer.append("#include \"foo/bar.h\"\n");
InputFile file(SourceFile("//foo.cc"));
file.SetContents(buffer);
IncludeStringWithLocation include;
CIncludeIterator iter(&file);
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("foo/bar.h", include.contents);
}
// Tests that we'll tolerate some small numbers of non-includes interspersed
// with real includes.
TEST(CIncludeIterator, TolerateNonIncludes) {
const size_t kSkip = CIncludeIterator::kMaxNonIncludeLines - 2;
const size_t kGroupCount = 100;
std::string include_str("foo/bar.h");
// Allow a series of includes with blanks in between.
std::string buffer;
for (size_t group = 0; group < kGroupCount; group++) {
for (size_t i = 0; i < kSkip; i++)
buffer.append("foo\n");
buffer.append("#include \"" + include_str + "\"\n");
}
InputFile file(SourceFile("//foo.cc"));
file.SetContents(buffer);
IncludeStringWithLocation include;
CIncludeIterator iter(&file);
for (size_t group = 0; group < kGroupCount; group++) {
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ(include_str, std::string(include.contents));
}
EXPECT_FALSE(iter.GetNextIncludeString(&include));
}
// Tests that comments of the form
// /*
// *
// */
// are not counted toward the non-include line count.
TEST(CIncludeIterator, CStyleComments) {
std::string buffer("/*");
for (size_t i = 0; i < 1000; i++)
buffer.append(" *\n");
buffer.append(" */\n\n");
buffer.append("#include \"foo/bar.h\"\n");
InputFile file(SourceFile("//foo.cc"));
file.SetContents(buffer);
IncludeStringWithLocation include;
CIncludeIterator iter(&file);
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("foo/bar.h", include.contents);
}
// Tests that spaces between the hash and directive are ignored.
TEST(CIncludeIterator, SpacesAfterHash) {
std::string buffer("# include \"foo/bar.h\"\n");
InputFile file(SourceFile("//foo.cc"));
file.SetContents(buffer);
IncludeStringWithLocation include;
CIncludeIterator iter(&file);
EXPECT_TRUE(iter.GetNextIncludeString(&include));
EXPECT_EQ("foo/bar.h", include.contents);
EXPECT_FALSE(iter.GetNextIncludeString(&include));
}