Support for source_target_relative expansion in GN

tools/gn/substitution_writer_unittest.cc

 // 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 <sstream>
 
 #include "build/build_config.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "tools/gn/err.h"
 #include "tools/gn/escape.h"
@@ skipping 25 matching lines @@
 
 TEST(SubstitutionWriter, ApplyPatternToSource) {
   TestWithScope setup;
 
   SubstitutionPattern pattern;
   Err err;
   ASSERT_TRUE(pattern.Parse("{{source_gen_dir}}/{{source_name_part}}.tmp",
                             nullptr, &err));
 
   SourceFile result = SubstitutionWriter::ApplyPatternToSource(
-      setup.settings(), pattern, SourceFile("//foo/bar/myfile.txt"));
+      nullptr, setup.settings(), pattern, SourceFile("//foo/bar/myfile.txt"));
   ASSERT_EQ("//out/Debug/gen/foo/bar/myfile.tmp", result.value());
 }
 
 TEST(SubstitutionWriter, ApplyPatternToSourceAsOutputFile) {
   TestWithScope setup;
 
   SubstitutionPattern pattern;
   Err err;
   ASSERT_TRUE(pattern.Parse("{{source_gen_dir}}/{{source_name_part}}.tmp",
                             nullptr, &err));
 
   OutputFile result = SubstitutionWriter::ApplyPatternToSourceAsOutputFile(
-      setup.settings(), pattern, SourceFile("//foo/bar/myfile.txt"));
+      nullptr, setup.settings(), pattern, SourceFile("//foo/bar/myfile.txt"));
   ASSERT_EQ("gen/foo/bar/myfile.tmp", result.value());
 }
 
 TEST(SubstitutionWriter, WriteNinjaVariablesForSource) {
   TestWithScope setup;
 
   std::vector<SubstitutionType> types;
   types.push_back(SUBSTITUTION_SOURCE);
   types.push_back(SUBSTITUTION_SOURCE_NAME_PART);
   types.push_back(SUBSTITUTION_SOURCE_DIR);
 
   EscapeOptions options;
   options.mode = ESCAPE_NONE;
 
   std::ostringstream out;
   SubstitutionWriter::WriteNinjaVariablesForSource(
-      setup.settings(), SourceFile("//foo/bar/baz.txt"), types, options, out);
+      nullptr, setup.settings(), SourceFile("//foo/bar/baz.txt"), types,
+      options, out);
 
   // The "source" should be skipped since that will expand to $in which is
   // implicit.
   EXPECT_EQ(
       "  source_name_part = baz\n"
       "  source_dir = ../../foo/bar\n",
       out.str());
 }
 
 TEST(SubstitutionWriter, WriteWithNinjaVariables) {
   Err err;
   SubstitutionPattern pattern;
   ASSERT_TRUE(pattern.Parse("-i {{source}} --out=bar\"{{source_name_part}}\".o",
                             nullptr, &err));
   EXPECT_FALSE(err.has_error());
 
   EscapeOptions options;
   options.mode = ESCAPE_NONE;
 
   std::ostringstream out;
   SubstitutionWriter::WriteWithNinjaVariables(pattern, options, out);
 
   EXPECT_EQ(
       "-i ${in} --out=bar\"${source_name_part}\".o",
       out.str());
 }
 
 TEST(SubstitutionWriter, SourceSubstitutions) {
   TestWithScope setup;
+  Err err;
+
+  Target target(setup.settings(), Label(SourceDir("//foo/bar/"), "baz"));
+  target.set_output_type(Target::STATIC_LIBRARY);
+  target.SetToolchain(setup.toolchain());
+  ASSERT_TRUE(target.OnResolved(&err));
 
   // Call to get substitutions relative to the build dir.
   #define GetRelSubst(str, what) \
       SubstitutionWriter::GetSourceSubstitution( \
+          &target, \
           setup.settings(), \
           SourceFile(str), \
           what, \
           SubstitutionWriter::OUTPUT_RELATIVE, \
           setup.settings()->build_settings()->build_dir())
 
   // Call to get absolute directory substitutions.
   #define GetAbsSubst(str, what) \
       SubstitutionWriter::GetSourceSubstitution( \
+          &target, \
           setup.settings(), \
           SourceFile(str), \
           what, \
           SubstitutionWriter::OUTPUT_ABSOLUTE, \
           SourceDir())
 
   // Try all possible templates with a normal looking string.
   EXPECT_EQ("../../foo/bar/baz.txt",
             GetRelSubst("//foo/bar/baz.txt", SUBSTITUTION_SOURCE));
   EXPECT_EQ("//foo/bar/baz.txt",
@@ skipping 39 matching lines @@
 #if defined(OS_WIN)
   EXPECT_EQ("gen/ABS_PATH/C",
             GetRelSubst("/C:/baz.txt", SUBSTITUTION_SOURCE_GEN_DIR));
   EXPECT_EQ("obj/ABS_PATH/C",
             GetRelSubst("/C:/baz.txt", SUBSTITUTION_SOURCE_OUT_DIR));
 #endif
 
   EXPECT_EQ(".",
             GetRelSubst("//baz.txt", SUBSTITUTION_SOURCE_ROOT_RELATIVE_DIR));
 
+  EXPECT_EQ("baz.txt",
+      GetRelSubst("//foo/bar/baz.txt", SUBSTITUTION_SOURCE_TARGET_RELATIVE));
+  EXPECT_EQ("baz.txt",
+      GetAbsSubst("//foo/bar/baz.txt", SUBSTITUTION_SOURCE_TARGET_RELATIVE));
+
   #undef GetAbsSubst
   #undef GetRelSubst
 }
 
 TEST(SubstitutionWriter, TargetSubstitutions) {
   TestWithScope setup;
   Err err;
 
   Target target(setup.settings(), Label(SourceDir("//foo/bar/"), "baz"));
   target.set_output_type(Target::STATIC_LIBRARY);
@@ skipping 126 matching lines @@
   EXPECT_EQ("./baz.exe",
             SubstitutionWriter::ApplyPatternToLinkerAsOutputFile(
                 &target, &tool, output_name).value());
 
   // Override the output name to a new subdirectory.
   target.set_output_dir(SourceDir("//out/Debug/foo/bar"));
   EXPECT_EQ("foo/bar/baz.exe",
             SubstitutionWriter::ApplyPatternToLinkerAsOutputFile(
                 &target, &tool, output_name).value());
 }