android/rule_builder_test.go - SHIFTPHONES/android_build_soong - Gitiles

 // Copyright 2019 Google Inc. 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.

 package android

 import (
 	"fmt"
 	"path/filepath"
 	"reflect"
 	"strings"
 	"testing"

 	"github.com/google/blueprint"

 	"android/soong/shared"
 )

 func pathContext() PathContext {
 	return PathContextForTesting(TestConfig("out", nil, "", map[string][]byte{
 		"ld":      nil,
 		"a.o":     nil,
 		"b.o":     nil,
 		"cp":      nil,
 		"a":       nil,
 		"b":       nil,
 		"ls":      nil,
 		"turbine": nil,
 		"java":    nil,
 		"javac":   nil,
 	}))
 }

 func ExampleRuleBuilder() {
 	rule := NewRuleBuilder()

 	ctx := pathContext()

 	rule.Command().
 		Tool(PathForSource(ctx, "ld")).
 		Inputs(PathsForTesting("a.o", "b.o")).
 		FlagWithOutput("-o ", PathForOutput(ctx, "linked"))
 	rule.Command().Text("echo success")

 	// To add the command to the build graph:
 	// rule.Build(pctx, ctx, "link", "link")

 	fmt.Printf("commands: %q\n", strings.Join(rule.Commands(), " && "))
 	fmt.Printf("tools: %q\n", rule.Tools())
 	fmt.Printf("inputs: %q\n", rule.Inputs())
 	fmt.Printf("outputs: %q\n", rule.Outputs())

 	// Output:
 	// commands: "ld a.o b.o -o out/linked && echo success"
 	// tools: ["ld"]
 	// inputs: ["a.o" "b.o"]
 	// outputs: ["out/linked"]
 }

 func ExampleRuleBuilder_Temporary() {
 	rule := NewRuleBuilder()

 	ctx := pathContext()

 	rule.Command().
 		Tool(PathForSource(ctx, "cp")).
 		Input(PathForSource(ctx, "a")).
 		Output(PathForOutput(ctx, "b"))
 	rule.Command().
 		Tool(PathForSource(ctx, "cp")).
 		Input(PathForOutput(ctx, "b")).
 		Output(PathForOutput(ctx, "c"))
 	rule.Temporary(PathForOutput(ctx, "b"))

 	fmt.Printf("commands: %q\n", strings.Join(rule.Commands(), " && "))
 	fmt.Printf("tools: %q\n", rule.Tools())
 	fmt.Printf("inputs: %q\n", rule.Inputs())
 	fmt.Printf("outputs: %q\n", rule.Outputs())

 	// Output:
 	// commands: "cp a out/b && cp out/b out/c"
 	// tools: ["cp"]
 	// inputs: ["a"]
 	// outputs: ["out/c"]
 }

 func ExampleRuleBuilder_DeleteTemporaryFiles() {
 	rule := NewRuleBuilder()

 	ctx := pathContext()

 	rule.Command().
 		Tool(PathForSource(ctx, "cp")).
 		Input(PathForSource(ctx, "a")).
 		Output(PathForOutput(ctx, "b"))
 	rule.Command().
 		Tool(PathForSource(ctx, "cp")).
 		Input(PathForOutput(ctx, "b")).
 		Output(PathForOutput(ctx, "c"))
 	rule.Temporary(PathForOutput(ctx, "b"))
 	rule.DeleteTemporaryFiles()

 	fmt.Printf("commands: %q\n", strings.Join(rule.Commands(), " && "))
 	fmt.Printf("tools: %q\n", rule.Tools())
 	fmt.Printf("inputs: %q\n", rule.Inputs())
 	fmt.Printf("outputs: %q\n", rule.Outputs())

 	// Output:
 	// commands: "cp a out/b && cp out/b out/c && rm -f out/b"
 	// tools: ["cp"]
 	// inputs: ["a"]
 	// outputs: ["out/c"]
 }

 func ExampleRuleBuilder_Installs() {
 	rule := NewRuleBuilder()

 	ctx := pathContext()

 	out := PathForOutput(ctx, "linked")

 	rule.Command().
 		Tool(PathForSource(ctx, "ld")).
 		Inputs(PathsForTesting("a.o", "b.o")).
 		FlagWithOutput("-o ", out)
 	rule.Install(out, "/bin/linked")
 	rule.Install(out, "/sbin/linked")

 	fmt.Printf("rule.Installs().String() = %q\n", rule.Installs().String())

 	// Output:
 	// rule.Installs().String() = "out/linked:/bin/linked out/linked:/sbin/linked"
 }

 func ExampleRuleBuilderCommand() {
 	rule := NewRuleBuilder()

 	ctx := pathContext()

 	// chained
 	rule.Command().
 		Tool(PathForSource(ctx, "ld")).
 		Inputs(PathsForTesting("a.o", "b.o")).
 		FlagWithOutput("-o ", PathForOutput(ctx, "linked"))

 	// unchained
 	cmd := rule.Command()
 	cmd.Tool(PathForSource(ctx, "ld"))
 	cmd.Inputs(PathsForTesting("a.o", "b.o"))
 	cmd.FlagWithOutput("-o ", PathForOutput(ctx, "linked"))

 	// mixed:
 	cmd = rule.Command().Tool(PathForSource(ctx, "ld"))
 	cmd.Inputs(PathsForTesting("a.o", "b.o"))
 	cmd.FlagWithOutput("-o ", PathForOutput(ctx, "linked"))
 }

 func ExampleRuleBuilderCommand_Flag() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "ls")).Flag("-l"))
 	// Output:
 	// ls -l
 }

 func ExampleRuleBuilderCommand_Flags() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "ls")).Flags([]string{"-l", "-a"}))
 	// Output:
 	// ls -l -a
 }

 func ExampleRuleBuilderCommand_FlagWithArg() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "ls")).
 		FlagWithArg("--sort=", "time"))
 	// Output:
 	// ls --sort=time
 }

 func ExampleRuleBuilderCommand_FlagForEachArg() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "ls")).
 		FlagForEachArg("--sort=", []string{"time", "size"}))
 	// Output:
 	// ls --sort=time --sort=size
 }

 func ExampleRuleBuilderCommand_FlagForEachInput() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "turbine")).
 		FlagForEachInput("--classpath ", PathsForTesting("a.jar", "b.jar")))
 	// Output:
 	// turbine --classpath a.jar --classpath b.jar
 }

 func ExampleRuleBuilderCommand_FlagWithInputList() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "java")).
 		FlagWithInputList("-classpath=", PathsForTesting("a.jar", "b.jar"), ":"))
 	// Output:
 	// java -classpath=a.jar:b.jar
 }

 func ExampleRuleBuilderCommand_FlagWithInput() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "java")).
 		FlagWithInput("-classpath=", PathForSource(ctx, "a")))
 	// Output:
 	// java -classpath=a
 }

 func ExampleRuleBuilderCommand_FlagWithList() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "ls")).
 		FlagWithList("--sort=", []string{"time", "size"}, ","))
 	// Output:
 	// ls --sort=time,size
 }

 func ExampleRuleBuilderCommand_FlagWithRspFileInputList() {
 	ctx := pathContext()
 	fmt.Println(NewRuleBuilder().Command().
 		Tool(PathForSource(ctx, "javac")).
 		FlagWithRspFileInputList("@", PathsForTesting("a.java", "b.java")).
 		NinjaEscapedString())
 	// Output:
 	// javac @$out.rsp
 }

 func ExampleRuleBuilderCommand_String() {
 	fmt.Println(NewRuleBuilder().Command().
 		Text("FOO=foo").
 		Text("echo $FOO").
 		String())
 	// Output:
 	// FOO=foo echo $FOO
 }

 func ExampleRuleBuilderCommand_NinjaEscapedString() {
 	fmt.Println(NewRuleBuilder().Command().
 		Text("FOO=foo").
 		Text("echo $FOO").
 		NinjaEscapedString())
 	// Output:
 	// FOO=foo echo $$FOO
 }

 func TestRuleBuilder(t *testing.T) {
 	fs := map[string][]byte{
 		"dep_fixer":  nil,
 		"input":      nil,
 		"Implicit":   nil,
 		"Input":      nil,
 		"OrderOnly":  nil,
 		"OrderOnlys": nil,
 		"Tool":       nil,
 		"input2":     nil,
 		"tool2":      nil,
 		"input3":     nil,
 	}

 	ctx := PathContextForTesting(TestConfig("out", nil, "", fs))

 	addCommands := func(rule *RuleBuilder) {
 		cmd := rule.Command().
 			DepFile(PathForOutput(ctx, "DepFile")).
 			Flag("Flag").
 			FlagWithArg("FlagWithArg=", "arg").
 			FlagWithDepFile("FlagWithDepFile=", PathForOutput(ctx, "depfile")).
 			FlagWithInput("FlagWithInput=", PathForSource(ctx, "input")).
 			FlagWithOutput("FlagWithOutput=", PathForOutput(ctx, "output")).
 			Implicit(PathForSource(ctx, "Implicit")).
 			ImplicitDepFile(PathForOutput(ctx, "ImplicitDepFile")).
 			ImplicitOutput(PathForOutput(ctx, "ImplicitOutput")).
 			Input(PathForSource(ctx, "Input")).
 			Output(PathForOutput(ctx, "Output")).
 			OrderOnly(PathForSource(ctx, "OrderOnly")).
 			Text("Text").
 			Tool(PathForSource(ctx, "Tool"))

 		rule.Command().
 			Text("command2").
 			DepFile(PathForOutput(ctx, "depfile2")).
 			Input(PathForSource(ctx, "input2")).
 			Output(PathForOutput(ctx, "output2")).
 			OrderOnlys(PathsForSource(ctx, []string{"OrderOnlys"})).
 			Tool(PathForSource(ctx, "tool2"))

 		// Test updates to the first command after the second command has been started
 		cmd.Text("after command2")
 		// Test updating a command when the previous update did not replace the cmd variable
 		cmd.Text("old cmd")

 		// Test a command that uses the output of a previous command as an input
 		rule.Command().
 			Text("command3").
 			Input(PathForSource(ctx, "input3")).
 			Input(PathForOutput(ctx, "output2")).
 			Output(PathForOutput(ctx, "output3"))
 	}

 	wantInputs := PathsForSource(ctx, []string{"Implicit", "Input", "input", "input2", "input3"})
 	wantOutputs := PathsForOutput(ctx, []string{"ImplicitOutput", "Output", "output", "output2", "output3"})
 	wantDepFiles := PathsForOutput(ctx, []string{"DepFile", "depfile", "ImplicitDepFile", "depfile2"})
 	wantTools := PathsForSource(ctx, []string{"Tool", "tool2"})
 	wantOrderOnlys := PathsForSource(ctx, []string{"OrderOnly", "OrderOnlys"})

 	t.Run("normal", func(t *testing.T) {
 		rule := NewRuleBuilder()
 		addCommands(rule)

 		wantCommands := []string{
 			"out/DepFile Flag FlagWithArg=arg FlagWithDepFile=out/depfile FlagWithInput=input FlagWithOutput=out/output Input out/Output Text Tool after command2 old cmd",
 			"command2 out/depfile2 input2 out/output2 tool2",
 			"command3 input3 out/output2 out/output3",
 		}

 		wantDepMergerCommand := "out/host/" + ctx.Config().PrebuiltOS() + "/bin/dep_fixer out/DepFile out/depfile out/ImplicitDepFile out/depfile2"

 		if g, w := rule.Commands(), wantCommands; !reflect.DeepEqual(g, w) {
 			t.Errorf("\nwant rule.Commands() = %#v\n                   got %#v", w, g)
 		}

 		if g, w := rule.Inputs(), wantInputs; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.Inputs() = %#v\n                 got %#v", w, g)
 		}
 		if g, w := rule.Outputs(), wantOutputs; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.Outputs() = %#v\n                  got %#v", w, g)
 		}
 		if g, w := rule.DepFiles(), wantDepFiles; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.DepFiles() = %#v\n                  got %#v", w, g)
 		}
 		if g, w := rule.Tools(), wantTools; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.Tools() = %#v\n                got %#v", w, g)
 		}
 		if g, w := rule.OrderOnlys(), wantOrderOnlys; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.OrderOnlys() = %#v\n                got %#v", w, g)
 		}

 		if g, w := rule.depFileMergerCmd(ctx, rule.DepFiles()).String(), wantDepMergerCommand; g != w {
 			t.Errorf("\nwant rule.depFileMergerCmd() = %#v\n                   got %#v", w, g)
 		}
 	})

 	t.Run("sbox", func(t *testing.T) {
 		rule := NewRuleBuilder().Sbox(PathForOutput(ctx))
 		addCommands(rule)

 		wantCommands := []string{
 			"__SBOX_OUT_DIR__/DepFile Flag FlagWithArg=arg FlagWithDepFile=__SBOX_OUT_DIR__/depfile FlagWithInput=input FlagWithOutput=__SBOX_OUT_DIR__/output Input __SBOX_OUT_DIR__/Output Text Tool after command2 old cmd",
 			"command2 __SBOX_OUT_DIR__/depfile2 input2 __SBOX_OUT_DIR__/output2 tool2",
 			"command3 input3 __SBOX_OUT_DIR__/output2 __SBOX_OUT_DIR__/output3",
 		}

 		wantDepMergerCommand := "out/host/" + ctx.Config().PrebuiltOS() + "/bin/dep_fixer __SBOX_OUT_DIR__/DepFile __SBOX_OUT_DIR__/depfile __SBOX_OUT_DIR__/ImplicitDepFile __SBOX_OUT_DIR__/depfile2"

 		if g, w := rule.Commands(), wantCommands; !reflect.DeepEqual(g, w) {
 			t.Errorf("\nwant rule.Commands() = %#v\n                   got %#v", w, g)
 		}

 		if g, w := rule.Inputs(), wantInputs; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.Inputs() = %#v\n                 got %#v", w, g)
 		}
 		if g, w := rule.Outputs(), wantOutputs; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.Outputs() = %#v\n                  got %#v", w, g)
 		}
 		if g, w := rule.DepFiles(), wantDepFiles; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.DepFiles() = %#v\n                  got %#v", w, g)
 		}
 		if g, w := rule.Tools(), wantTools; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.Tools() = %#v\n                got %#v", w, g)
 		}
 		if g, w := rule.OrderOnlys(), wantOrderOnlys; !reflect.DeepEqual(w, g) {
 			t.Errorf("\nwant rule.OrderOnlys() = %#v\n                got %#v", w, g)
 		}

 		if g, w := rule.depFileMergerCmd(ctx, rule.DepFiles()).String(), wantDepMergerCommand; g != w {
 			t.Errorf("\nwant rule.depFileMergerCmd() = %#v\n                   got %#v", w, g)
 		}
 	})
 }

 func testRuleBuilderFactory() Module {
 	module := &testRuleBuilderModule{}
 	module.AddProperties(&module.properties)
 	InitAndroidModule(module)
 	return module
 }

 type testRuleBuilderModule struct {
 	ModuleBase
 	properties struct {
 		Src string

 		Restat bool
 		Sbox   bool
 	}
 }

 func (t *testRuleBuilderModule) GenerateAndroidBuildActions(ctx ModuleContext) {
 	in := PathForSource(ctx, t.properties.Src)
 	out := PathForModuleOut(ctx, ctx.ModuleName())
 	outDep := PathForModuleOut(ctx, ctx.ModuleName()+".d")
 	outDir := PathForModuleOut(ctx)

 	testRuleBuilder_Build(ctx, in, out, outDep, outDir, t.properties.Restat, t.properties.Sbox)
 }

 type testRuleBuilderSingleton struct{}

 func testRuleBuilderSingletonFactory() Singleton {
 	return &testRuleBuilderSingleton{}
 }

 func (t *testRuleBuilderSingleton) GenerateBuildActions(ctx SingletonContext) {
 	in := PathForSource(ctx, "bar")
 	out := PathForOutput(ctx, "baz")
 	outDep := PathForOutput(ctx, "baz.d")
 	outDir := PathForOutput(ctx)
 	testRuleBuilder_Build(ctx, in, out, outDep, outDir, true, false)
 }

 func testRuleBuilder_Build(ctx BuilderContext, in Path, out, outDep, outDir WritablePath, restat, sbox bool) {
 	rule := NewRuleBuilder()

 	if sbox {
 		rule.Sbox(outDir)
 	}

 	rule.Command().Tool(PathForSource(ctx, "cp")).Input(in).Output(out).ImplicitDepFile(outDep)

 	if restat {
 		rule.Restat()
 	}

 	rule.Build(pctx, ctx, "rule", "desc")
 }

 func TestRuleBuilder_Build(t *testing.T) {
 	fs := map[string][]byte{
 		"bar": nil,
 		"cp":  nil,
 	}

 	bp := `
 		rule_builder_test {
 			name: "foo",
 			src: "bar",
 			restat: true,
 		}
 		rule_builder_test {
 			name: "foo_sbox",
 			src: "bar",
 			sbox: true,
 		}
 	`

 	config := TestConfig(buildDir, nil, bp, fs)
 	ctx := NewTestContext()
 	ctx.RegisterModuleType("rule_builder_test", testRuleBuilderFactory)
 	ctx.RegisterSingletonType("rule_builder_test", testRuleBuilderSingletonFactory)
 	ctx.Register(config)

 	_, errs := ctx.ParseFileList(".", []string{"Android.bp"})
 	FailIfErrored(t, errs)
 	_, errs = ctx.PrepareBuildActions(config)
 	FailIfErrored(t, errs)

 	check := func(t *testing.T, params TestingBuildParams, wantCommand, wantOutput, wantDepfile string, wantRestat bool, extraCmdDeps []string) {
 		t.Helper()
 		if params.RuleParams.Command != wantCommand {
 			t.Errorf("\nwant RuleParams.Command = %q\n                      got %q", wantCommand, params.RuleParams.Command)
 		}

 		wantDeps := append([]string{"cp"}, extraCmdDeps...)
 		if !reflect.DeepEqual(params.RuleParams.CommandDeps, wantDeps) {
 			t.Errorf("\nwant RuleParams.CommandDeps = %q\n                          got %q", wantDeps, params.RuleParams.CommandDeps)
 		}

 		if params.RuleParams.Restat != wantRestat {
 			t.Errorf("want RuleParams.Restat = %v, got %v", wantRestat, params.RuleParams.Restat)
 		}

 		if len(params.Implicits) != 1 || params.Implicits[0].String() != "bar" {
 			t.Errorf("want Implicits = [%q], got %q", "bar", params.Implicits.Strings())
 		}

 		if params.Output.String() != wantOutput {
 			t.Errorf("want Output = %q, got %q", wantOutput, params.Output)
 		}

 		if len(params.ImplicitOutputs) != 0 {
 			t.Errorf("want ImplicitOutputs = [], got %q", params.ImplicitOutputs.Strings())
 		}

 		if params.Depfile.String() != wantDepfile {
 			t.Errorf("want Depfile = %q, got %q", wantDepfile, params.Depfile)
 		}

 		if params.Deps != blueprint.DepsGCC {
 			t.Errorf("want Deps = %q, got %q", blueprint.DepsGCC, params.Deps)
 		}
 	}

 	t.Run("module", func(t *testing.T) {
 		outFile := filepath.Join(buildDir, ".intermediates", "foo", "foo")
 		check(t, ctx.ModuleForTests("foo", "").Rule("rule"),
 			"cp bar "+outFile,
 			outFile, outFile+".d", true, nil)
 	})
 	t.Run("sbox", func(t *testing.T) {
 		outDir := filepath.Join(buildDir, ".intermediates", "foo_sbox")
 		outFile := filepath.Join(outDir, "foo_sbox")
 		depFile := filepath.Join(outDir, "foo_sbox.d")
 		sbox := filepath.Join(buildDir, "host", config.PrebuiltOS(), "bin/sbox")
 		sandboxPath := shared.TempDirForOutDir(buildDir)

 		cmd := sbox + ` -c 'cp bar __SBOX_OUT_DIR__/foo_sbox' --sandbox-path ` + sandboxPath + " --output-root " + outDir + " --depfile-out " + depFile + " __SBOX_OUT_DIR__/foo_sbox"

 		check(t, ctx.ModuleForTests("foo_sbox", "").Rule("rule"),
 			cmd, outFile, depFile, false, []string{sbox})
 	})
 	t.Run("singleton", func(t *testing.T) {
 		outFile := filepath.Join(buildDir, "baz")
 		check(t, ctx.SingletonForTests("rule_builder_test").Rule("rule"),
 			"cp bar "+outFile, outFile, outFile+".d", true, nil)
 	})
 }

 func Test_ninjaEscapeExceptForSpans(t *testing.T) {
 	type args struct {
 		s     string
 		spans [][2]int
 	}
 	tests := []struct {
 		name string
 		args args
 		want string
 	}{
 		{
 			name: "empty",
 			args: args{
 				s: "",
 			},
 			want: "",
 		},
 		{
 			name: "unescape none",
 			args: args{
 				s: "$abc",
 			},
 			want: "$$abc",
 		},
 		{
 			name: "unescape all",
 			args: args{
 				s:     "$abc",
 				spans: [][2]int{{0, 4}},
 			},
 			want: "$abc",
 		},
 		{
 			name: "unescape first",
 			args: args{
 				s:     "$abc$",
 				spans: [][2]int{{0, 1}},
 			},
 			want: "$abc$$",
 		},
 		{
 			name: "unescape last",
 			args: args{
 				s:     "$abc$",
 				spans: [][2]int{{4, 5}},
 			},
 			want: "$$abc$",
 		},
 		{
 			name: "unescape middle",
 			args: args{
 				s:     "$a$b$c$",
 				spans: [][2]int{{2, 5}},
 			},
 			want: "$$a$b$c$$",
 		},
 		{
 			name: "unescape multiple",
 			args: args{
 				s:     "$a$b$c$",
 				spans: [][2]int{{2, 3}, {4, 5}},
 			},
 			want: "$$a$b$c$$",
 		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			if got := ninjaEscapeExceptForSpans(tt.args.s, tt.args.spans); got != tt.want {
 				t.Errorf("ninjaEscapeExceptForSpans() = %v, want %v", got, tt.want)
 			}
 		})
 	}
 }
	// Copyright 2019 Google Inc. 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.

	package android

	import (
	"fmt"
	"path/filepath"
	"reflect"
	"strings"
	"testing"

	"github.com/google/blueprint"

	"android/soong/shared"
	)

	func pathContext() PathContext {
	return PathContextForTesting(TestConfig("out", nil, "", map[string][]byte{
	"ld": nil,
	"a.o": nil,
	"b.o": nil,
	"cp": nil,
	"a": nil,
	"b": nil,
	"ls": nil,
	"turbine": nil,
	"java": nil,
	"javac": nil,
	}))
	}

	func ExampleRuleBuilder() {
	rule := NewRuleBuilder()

	ctx := pathContext()

	rule.Command().
	Tool(PathForSource(ctx, "ld")).
	Inputs(PathsForTesting("a.o", "b.o")).
	FlagWithOutput("-o ", PathForOutput(ctx, "linked"))
	rule.Command().Text("echo success")

	// To add the command to the build graph:
	// rule.Build(pctx, ctx, "link", "link")

	fmt.Printf("commands: %q\n", strings.Join(rule.Commands(), " && "))
	fmt.Printf("tools: %q\n", rule.Tools())
	fmt.Printf("inputs: %q\n", rule.Inputs())
	fmt.Printf("outputs: %q\n", rule.Outputs())

	// Output:
	// commands: "ld a.o b.o -o out/linked && echo success"
	// tools: ["ld"]
	// inputs: ["a.o" "b.o"]
	// outputs: ["out/linked"]
	}

	func ExampleRuleBuilder_Temporary() {
	rule := NewRuleBuilder()

	ctx := pathContext()

	rule.Command().
	Tool(PathForSource(ctx, "cp")).
	Input(PathForSource(ctx, "a")).
	Output(PathForOutput(ctx, "b"))
	rule.Command().
	Tool(PathForSource(ctx, "cp")).
	Input(PathForOutput(ctx, "b")).
	Output(PathForOutput(ctx, "c"))
	rule.Temporary(PathForOutput(ctx, "b"))

	fmt.Printf("commands: %q\n", strings.Join(rule.Commands(), " && "))
	fmt.Printf("tools: %q\n", rule.Tools())
	fmt.Printf("inputs: %q\n", rule.Inputs())
	fmt.Printf("outputs: %q\n", rule.Outputs())

	// Output:
	// commands: "cp a out/b && cp out/b out/c"
	// tools: ["cp"]
	// inputs: ["a"]
	// outputs: ["out/c"]
	}

	func ExampleRuleBuilder_DeleteTemporaryFiles() {
	rule := NewRuleBuilder()

	ctx := pathContext()

	rule.Command().
	Tool(PathForSource(ctx, "cp")).
	Input(PathForSource(ctx, "a")).
	Output(PathForOutput(ctx, "b"))
	rule.Command().
	Tool(PathForSource(ctx, "cp")).
	Input(PathForOutput(ctx, "b")).
	Output(PathForOutput(ctx, "c"))
	rule.Temporary(PathForOutput(ctx, "b"))
	rule.DeleteTemporaryFiles()

	fmt.Printf("commands: %q\n", strings.Join(rule.Commands(), " && "))
	fmt.Printf("tools: %q\n", rule.Tools())
	fmt.Printf("inputs: %q\n", rule.Inputs())
	fmt.Printf("outputs: %q\n", rule.Outputs())

	// Output:
	// commands: "cp a out/b && cp out/b out/c && rm -f out/b"
	// tools: ["cp"]
	// inputs: ["a"]
	// outputs: ["out/c"]
	}

	func ExampleRuleBuilder_Installs() {
	rule := NewRuleBuilder()

	ctx := pathContext()

	out := PathForOutput(ctx, "linked")

	rule.Command().
	Tool(PathForSource(ctx, "ld")).
	Inputs(PathsForTesting("a.o", "b.o")).
	FlagWithOutput("-o ", out)
	rule.Install(out, "/bin/linked")
	rule.Install(out, "/sbin/linked")

	fmt.Printf("rule.Installs().String() = %q\n", rule.Installs().String())

	// Output:
	// rule.Installs().String() = "out/linked:/bin/linked out/linked:/sbin/linked"
	}

	func ExampleRuleBuilderCommand() {
	rule := NewRuleBuilder()

	ctx := pathContext()

	// chained
	rule.Command().
	Tool(PathForSource(ctx, "ld")).
	Inputs(PathsForTesting("a.o", "b.o")).
	FlagWithOutput("-o ", PathForOutput(ctx, "linked"))

	// unchained
	cmd := rule.Command()
	cmd.Tool(PathForSource(ctx, "ld"))
	cmd.Inputs(PathsForTesting("a.o", "b.o"))
	cmd.FlagWithOutput("-o ", PathForOutput(ctx, "linked"))

	// mixed:
	cmd = rule.Command().Tool(PathForSource(ctx, "ld"))
	cmd.Inputs(PathsForTesting("a.o", "b.o"))
	cmd.FlagWithOutput("-o ", PathForOutput(ctx, "linked"))
	}

	func ExampleRuleBuilderCommand_Flag() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "ls")).Flag("-l"))
	// Output:
	// ls -l
	}

	func ExampleRuleBuilderCommand_Flags() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "ls")).Flags([]string{"-l", "-a"}))
	// Output:
	// ls -l -a
	}

	func ExampleRuleBuilderCommand_FlagWithArg() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "ls")).
	FlagWithArg("--sort=", "time"))
	// Output:
	// ls --sort=time
	}

	func ExampleRuleBuilderCommand_FlagForEachArg() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "ls")).
	FlagForEachArg("--sort=", []string{"time", "size"}))
	// Output:
	// ls --sort=time --sort=size
	}

	func ExampleRuleBuilderCommand_FlagForEachInput() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "turbine")).
	FlagForEachInput("--classpath ", PathsForTesting("a.jar", "b.jar")))
	// Output:
	// turbine --classpath a.jar --classpath b.jar
	}

	func ExampleRuleBuilderCommand_FlagWithInputList() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "java")).
	FlagWithInputList("-classpath=", PathsForTesting("a.jar", "b.jar"), ":"))
	// Output:
	// java -classpath=a.jar:b.jar
	}

	func ExampleRuleBuilderCommand_FlagWithInput() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "java")).
	FlagWithInput("-classpath=", PathForSource(ctx, "a")))
	// Output:
	// java -classpath=a
	}

	func ExampleRuleBuilderCommand_FlagWithList() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "ls")).
	FlagWithList("--sort=", []string{"time", "size"}, ","))
	// Output:
	// ls --sort=time,size
	}

	func ExampleRuleBuilderCommand_FlagWithRspFileInputList() {
	ctx := pathContext()
	fmt.Println(NewRuleBuilder().Command().
	Tool(PathForSource(ctx, "javac")).
	FlagWithRspFileInputList("@", PathsForTesting("a.java", "b.java")).
	NinjaEscapedString())
	// Output:
	// javac @$out.rsp
	}

	func ExampleRuleBuilderCommand_String() {
	fmt.Println(NewRuleBuilder().Command().
	Text("FOO=foo").
	Text("echo $FOO").
	String())
	// Output:
	// FOO=foo echo $FOO
	}

	func ExampleRuleBuilderCommand_NinjaEscapedString() {
	fmt.Println(NewRuleBuilder().Command().
	Text("FOO=foo").
	Text("echo $FOO").
	NinjaEscapedString())
	// Output:
	// FOO=foo echo $$FOO
	}

	func TestRuleBuilder(t *testing.T) {
	fs := map[string][]byte{
	"dep_fixer": nil,
	"input": nil,
	"Implicit": nil,
	"Input": nil,
	"OrderOnly": nil,
	"OrderOnlys": nil,
	"Tool": nil,
	"input2": nil,
	"tool2": nil,
	"input3": nil,
	}

	ctx := PathContextForTesting(TestConfig("out", nil, "", fs))

	addCommands := func(rule *RuleBuilder) {
	cmd := rule.Command().
	DepFile(PathForOutput(ctx, "DepFile")).
	Flag("Flag").
	FlagWithArg("FlagWithArg=", "arg").
	FlagWithDepFile("FlagWithDepFile=", PathForOutput(ctx, "depfile")).
	FlagWithInput("FlagWithInput=", PathForSource(ctx, "input")).
	FlagWithOutput("FlagWithOutput=", PathForOutput(ctx, "output")).
	Implicit(PathForSource(ctx, "Implicit")).
	ImplicitDepFile(PathForOutput(ctx, "ImplicitDepFile")).
	ImplicitOutput(PathForOutput(ctx, "ImplicitOutput")).
	Input(PathForSource(ctx, "Input")).
	Output(PathForOutput(ctx, "Output")).
	OrderOnly(PathForSource(ctx, "OrderOnly")).
	Text("Text").
	Tool(PathForSource(ctx, "Tool"))

	rule.Command().
	Text("command2").
	DepFile(PathForOutput(ctx, "depfile2")).
	Input(PathForSource(ctx, "input2")).
	Output(PathForOutput(ctx, "output2")).
	OrderOnlys(PathsForSource(ctx, []string{"OrderOnlys"})).
	Tool(PathForSource(ctx, "tool2"))

	// Test updates to the first command after the second command has been started
	cmd.Text("after command2")
	// Test updating a command when the previous update did not replace the cmd variable
	cmd.Text("old cmd")

	// Test a command that uses the output of a previous command as an input
	rule.Command().
	Text("command3").
	Input(PathForSource(ctx, "input3")).
	Input(PathForOutput(ctx, "output2")).
	Output(PathForOutput(ctx, "output3"))
	}

	wantInputs := PathsForSource(ctx, []string{"Implicit", "Input", "input", "input2", "input3"})
	wantOutputs := PathsForOutput(ctx, []string{"ImplicitOutput", "Output", "output", "output2", "output3"})
	wantDepFiles := PathsForOutput(ctx, []string{"DepFile", "depfile", "ImplicitDepFile", "depfile2"})
	wantTools := PathsForSource(ctx, []string{"Tool", "tool2"})
	wantOrderOnlys := PathsForSource(ctx, []string{"OrderOnly", "OrderOnlys"})

	t.Run("normal", func(t *testing.T) {
	rule := NewRuleBuilder()
	addCommands(rule)

	wantCommands := []string{
	"out/DepFile Flag FlagWithArg=arg FlagWithDepFile=out/depfile FlagWithInput=input FlagWithOutput=out/output Input out/Output Text Tool after command2 old cmd",
	"command2 out/depfile2 input2 out/output2 tool2",
	"command3 input3 out/output2 out/output3",
	}

	wantDepMergerCommand := "out/host/" + ctx.Config().PrebuiltOS() + "/bin/dep_fixer out/DepFile out/depfile out/ImplicitDepFile out/depfile2"

	if g, w := rule.Commands(), wantCommands; !reflect.DeepEqual(g, w) {
	t.Errorf("\nwant rule.Commands() = %#v\n got %#v", w, g)
	}

	if g, w := rule.Inputs(), wantInputs; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.Inputs() = %#v\n got %#v", w, g)
	}
	if g, w := rule.Outputs(), wantOutputs; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.Outputs() = %#v\n got %#v", w, g)
	}
	if g, w := rule.DepFiles(), wantDepFiles; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.DepFiles() = %#v\n got %#v", w, g)
	}
	if g, w := rule.Tools(), wantTools; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.Tools() = %#v\n got %#v", w, g)
	}
	if g, w := rule.OrderOnlys(), wantOrderOnlys; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.OrderOnlys() = %#v\n got %#v", w, g)
	}

	if g, w := rule.depFileMergerCmd(ctx, rule.DepFiles()).String(), wantDepMergerCommand; g != w {
	t.Errorf("\nwant rule.depFileMergerCmd() = %#v\n got %#v", w, g)
	}
	})

	t.Run("sbox", func(t *testing.T) {
	rule := NewRuleBuilder().Sbox(PathForOutput(ctx))
	addCommands(rule)

	wantCommands := []string{
	"__SBOX_OUT_DIR__/DepFile Flag FlagWithArg=arg FlagWithDepFile=__SBOX_OUT_DIR__/depfile FlagWithInput=input FlagWithOutput=__SBOX_OUT_DIR__/output Input __SBOX_OUT_DIR__/Output Text Tool after command2 old cmd",
	"command2 __SBOX_OUT_DIR__/depfile2 input2 __SBOX_OUT_DIR__/output2 tool2",
	"command3 input3 __SBOX_OUT_DIR__/output2 __SBOX_OUT_DIR__/output3",
	}

	wantDepMergerCommand := "out/host/" + ctx.Config().PrebuiltOS() + "/bin/dep_fixer __SBOX_OUT_DIR__/DepFile __SBOX_OUT_DIR__/depfile __SBOX_OUT_DIR__/ImplicitDepFile __SBOX_OUT_DIR__/depfile2"

	if g, w := rule.Commands(), wantCommands; !reflect.DeepEqual(g, w) {
	t.Errorf("\nwant rule.Commands() = %#v\n got %#v", w, g)
	}

	if g, w := rule.Inputs(), wantInputs; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.Inputs() = %#v\n got %#v", w, g)
	}
	if g, w := rule.Outputs(), wantOutputs; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.Outputs() = %#v\n got %#v", w, g)
	}
	if g, w := rule.DepFiles(), wantDepFiles; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.DepFiles() = %#v\n got %#v", w, g)
	}
	if g, w := rule.Tools(), wantTools; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.Tools() = %#v\n got %#v", w, g)
	}
	if g, w := rule.OrderOnlys(), wantOrderOnlys; !reflect.DeepEqual(w, g) {
	t.Errorf("\nwant rule.OrderOnlys() = %#v\n got %#v", w, g)
	}

	if g, w := rule.depFileMergerCmd(ctx, rule.DepFiles()).String(), wantDepMergerCommand; g != w {
	t.Errorf("\nwant rule.depFileMergerCmd() = %#v\n got %#v", w, g)
	}
	})
	}

	func testRuleBuilderFactory() Module {
	module := &testRuleBuilderModule{}
	module.AddProperties(&module.properties)
	InitAndroidModule(module)
	return module
	}

	type testRuleBuilderModule struct {
	ModuleBase
	properties struct {
	Src string

	Restat bool
	Sbox bool
	}
	}

	func (t *testRuleBuilderModule) GenerateAndroidBuildActions(ctx ModuleContext) {
	in := PathForSource(ctx, t.properties.Src)
	out := PathForModuleOut(ctx, ctx.ModuleName())
	outDep := PathForModuleOut(ctx, ctx.ModuleName()+".d")
	outDir := PathForModuleOut(ctx)

	testRuleBuilder_Build(ctx, in, out, outDep, outDir, t.properties.Restat, t.properties.Sbox)
	}

	type testRuleBuilderSingleton struct{}

	func testRuleBuilderSingletonFactory() Singleton {
	return &testRuleBuilderSingleton{}
	}

	func (t *testRuleBuilderSingleton) GenerateBuildActions(ctx SingletonContext) {
	in := PathForSource(ctx, "bar")
	out := PathForOutput(ctx, "baz")
	outDep := PathForOutput(ctx, "baz.d")
	outDir := PathForOutput(ctx)
	testRuleBuilder_Build(ctx, in, out, outDep, outDir, true, false)
	}

	func testRuleBuilder_Build(ctx BuilderContext, in Path, out, outDep, outDir WritablePath, restat, sbox bool) {
	rule := NewRuleBuilder()

	if sbox {
	rule.Sbox(outDir)
	}

	rule.Command().Tool(PathForSource(ctx, "cp")).Input(in).Output(out).ImplicitDepFile(outDep)

	if restat {
	rule.Restat()
	}

	rule.Build(pctx, ctx, "rule", "desc")
	}

	func TestRuleBuilder_Build(t *testing.T) {
	fs := map[string][]byte{
	"bar": nil,
	"cp": nil,
	}

	bp := `
	rule_builder_test {
	name: "foo",
	src: "bar",
	restat: true,
	}
	rule_builder_test {
	name: "foo_sbox",
	src: "bar",
	sbox: true,
	}
	`

	config := TestConfig(buildDir, nil, bp, fs)
	ctx := NewTestContext()
	ctx.RegisterModuleType("rule_builder_test", testRuleBuilderFactory)
	ctx.RegisterSingletonType("rule_builder_test", testRuleBuilderSingletonFactory)
	ctx.Register(config)

	_, errs := ctx.ParseFileList(".", []string{"Android.bp"})
	FailIfErrored(t, errs)
	_, errs = ctx.PrepareBuildActions(config)
	FailIfErrored(t, errs)

	check := func(t *testing.T, params TestingBuildParams, wantCommand, wantOutput, wantDepfile string, wantRestat bool, extraCmdDeps []string) {
	t.Helper()
	if params.RuleParams.Command != wantCommand {
	t.Errorf("\nwant RuleParams.Command = %q\n got %q", wantCommand, params.RuleParams.Command)
	}

	wantDeps := append([]string{"cp"}, extraCmdDeps...)
	if !reflect.DeepEqual(params.RuleParams.CommandDeps, wantDeps) {
	t.Errorf("\nwant RuleParams.CommandDeps = %q\n got %q", wantDeps, params.RuleParams.CommandDeps)
	}

	if params.RuleParams.Restat != wantRestat {
	t.Errorf("want RuleParams.Restat = %v, got %v", wantRestat, params.RuleParams.Restat)
	}

	if len(params.Implicits) != 1 \|\| params.Implicits[0].String() != "bar" {
	t.Errorf("want Implicits = [%q], got %q", "bar", params.Implicits.Strings())
	}

	if params.Output.String() != wantOutput {
	t.Errorf("want Output = %q, got %q", wantOutput, params.Output)
	}

	if len(params.ImplicitOutputs) != 0 {
	t.Errorf("want ImplicitOutputs = [], got %q", params.ImplicitOutputs.Strings())
	}

	if params.Depfile.String() != wantDepfile {
	t.Errorf("want Depfile = %q, got %q", wantDepfile, params.Depfile)
	}

	if params.Deps != blueprint.DepsGCC {
	t.Errorf("want Deps = %q, got %q", blueprint.DepsGCC, params.Deps)
	}
	}

	t.Run("module", func(t *testing.T) {
	outFile := filepath.Join(buildDir, ".intermediates", "foo", "foo")
	check(t, ctx.ModuleForTests("foo", "").Rule("rule"),
	"cp bar "+outFile,
	outFile, outFile+".d", true, nil)
	})
	t.Run("sbox", func(t *testing.T) {
	outDir := filepath.Join(buildDir, ".intermediates", "foo_sbox")
	outFile := filepath.Join(outDir, "foo_sbox")
	depFile := filepath.Join(outDir, "foo_sbox.d")
	sbox := filepath.Join(buildDir, "host", config.PrebuiltOS(), "bin/sbox")
	sandboxPath := shared.TempDirForOutDir(buildDir)

	cmd := sbox + ` -c 'cp bar __SBOX_OUT_DIR__/foo_sbox' --sandbox-path ` + sandboxPath + " --output-root " + outDir + " --depfile-out " + depFile + " __SBOX_OUT_DIR__/foo_sbox"

	check(t, ctx.ModuleForTests("foo_sbox", "").Rule("rule"),
	cmd, outFile, depFile, false, []string{sbox})
	})
	t.Run("singleton", func(t *testing.T) {
	outFile := filepath.Join(buildDir, "baz")
	check(t, ctx.SingletonForTests("rule_builder_test").Rule("rule"),
	"cp bar "+outFile, outFile, outFile+".d", true, nil)
	})
	}

	func Test_ninjaEscapeExceptForSpans(t *testing.T) {
	type args struct {
	s string
	spans [][2]int
	}
	tests := []struct {
	name string
	args args
	want string
	}{
	{
	name: "empty",
	args: args{
	s: "",
	},
	want: "",
	},
	{
	name: "unescape none",
	args: args{
	s: "$abc",
	},
	want: "$$abc",
	},
	{
	name: "unescape all",
	args: args{
	s: "$abc",
	spans: [][2]int{{0, 4}},
	},
	want: "$abc",
	},
	{
	name: "unescape first",
	args: args{
	s: "$abc$",
	spans: [][2]int{{0, 1}},
	},
	want: "$abc$$",
	},
	{
	name: "unescape last",
	args: args{
	s: "$abc$",
	spans: [][2]int{{4, 5}},
	},
	want: "$$abc$",
	},
	{
	name: "unescape middle",
	args: args{
	s: "$a$b$c$",
	spans: [][2]int{{2, 5}},
	},
	want: "$$a$b$c$$",
	},
	{
	name: "unescape multiple",
	args: args{
	s: "$a$b$c$",
	spans: [][2]int{{2, 3}, {4, 5}},
	},
	want: "$$a$b$c$$",
	},
	}
	for _, tt := range tests {
	t.Run(tt.name, func(t *testing.T) {
	if got := ninjaEscapeExceptForSpans(tt.args.s, tt.args.spans); got != tt.want {
	t.Errorf("ninjaEscapeExceptForSpans() = %v, want %v", got, tt.want)
	}
	})
	}
	}