Add tests for Visit*

Change-Id: I52505ee4e23ec8b1c5763b3216788e999de9c8d3

Blueprints

@@ -26,6 +26,7 @@ bootstrap_go_package(
         "ninja_writer_test.go",
         "splice_modules_test.go",
         "unpack_test.go",
+	"visit_test.go",
     ],
 )
 

visit_test.go

@@ -0,0 +1,148 @@
+// Copyright 2016 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 blueprint
+
+import "testing"
+
+type visitModule struct {
+	properties struct {
+		Visit                 []string
+		VisitDepsDepthFirst   string `blueprint:"mutated"`
+		VisitDepsDepthFirstIf string `blueprint:"mutated"`
+		VisitDirectDeps       string `blueprint:"mutated"`
+		VisitDirectDepsIf     string `blueprint:"mutated"`
+	}
+}
+
+func newVisitModule() (Module, []interface{}) {
+	m := &visitModule{}
+	return m, []interface{}{&m.properties}
+}
+
+func (f *visitModule) GenerateBuildActions(ModuleContext) {
+}
+
+type visitTag struct {
+	BaseDependencyTag
+}
+
+var visitTagDep visitTag
+
+func visitDepsMutator(ctx BottomUpMutatorContext) {
+	if m, ok := ctx.Module().(*visitModule); ok {
+		ctx.AddDependency(ctx.Module(), visitTagDep, m.properties.Visit...)
+	}
+}
+
+func visitMutator(ctx TopDownMutatorContext) {
+	if m, ok := ctx.Module().(*visitModule); ok {
+		ctx.VisitDepsDepthFirst(func(dep Module) {
+			m.properties.VisitDepsDepthFirst = m.properties.VisitDepsDepthFirst + ctx.OtherModuleName(dep)
+		})
+		ctx.VisitDepsDepthFirstIf(func(dep Module) bool {
+			return ctx.OtherModuleName(dep) != "B"
+		}, func(dep Module) {
+			m.properties.VisitDepsDepthFirstIf = m.properties.VisitDepsDepthFirstIf + ctx.OtherModuleName(dep)
+		})
+		ctx.VisitDirectDeps(func(dep Module) {
+			m.properties.VisitDirectDeps = m.properties.VisitDirectDeps + ctx.OtherModuleName(dep)
+		})
+		ctx.VisitDirectDepsIf(func(dep Module) bool {
+			return ctx.OtherModuleName(dep) != "B"
+		}, func(dep Module) {
+			m.properties.VisitDirectDepsIf = m.properties.VisitDirectDepsIf + ctx.OtherModuleName(dep)
+		})
+	}
+}
+
+// A
+// |
+// B
+// |\
+// C \
+//  \|
+//   D
+//   |
+//   E
+func setupVisitTest(t *testing.T) *Context {
+	ctx := NewContext()
+	ctx.RegisterModuleType("visit_module", newVisitModule)
+	ctx.RegisterBottomUpMutator("visit_deps", visitDepsMutator)
+	ctx.RegisterTopDownMutator("visit", visitMutator)
+
+	ctx.MockFileSystem(map[string][]byte{
+		"Blueprints": []byte(`
+			visit_module {
+				name: "A",
+				visit: ["B"],
+			}
+	
+			visit_module {
+				name: "B",
+				visit: ["C", "D"],
+			}
+	
+			visit_module {
+				name: "C",
+				visit: ["D"],
+			}
+	
+			visit_module {
+				name: "D",
+				visit: ["E"],
+			}
+	
+			visit_module {
+				name: "E",
+			}
+		`),
+	})
+
+	_, errs := ctx.ParseBlueprintsFiles("Blueprints")
+	if len(errs) > 0 {
+		t.Errorf("unexpected parse errors:")
+		for _, err := range errs {
+			t.Errorf("  %s", err)
+		}
+		t.FailNow()
+	}
+
+	errs = ctx.ResolveDependencies(nil)
+	if len(errs) > 0 {
+		t.Errorf("unexpected dep errors:")
+		for _, err := range errs {
+			t.Errorf("  %s", err)
+		}
+		t.FailNow()
+	}
+
+	return ctx
+}
+
+func TestVisit(t *testing.T) {
+	ctx := setupVisitTest(t)
+
+	topModule := ctx.moduleGroups["A"].modules[0].logicModule.(*visitModule)
+	assertString(t, topModule.properties.VisitDepsDepthFirst, "EDCB")
+	assertString(t, topModule.properties.VisitDepsDepthFirstIf, "EDC")
+	assertString(t, topModule.properties.VisitDirectDeps, "B")
+	assertString(t, topModule.properties.VisitDirectDepsIf, "")
+}
+
+func assertString(t *testing.T, got, expected string) {
+	if got != expected {
+		t.Errorf("expected %q got %q", expected, got)
+	}
+}