1 files changed, 86 insertions, 0 deletions
diff --git a/src/util/fipstools/delocate/delocate_test.go b/src/util/fipstools/delocate/delocate_test.go
new file mode 100644
index 00000000..e0ecc177
--- /dev/null
+++ b/src/util/fipstools/delocate/delocate_test.go
@@ -0,0 +1,86 @@
+// Copyright (c) 2017, Google Inc.
+//
+// Permission to use, copy, modify, and/or distribute this software for any
+// purpose with or without fee is hereby granted, provided that the above
+// copyright notice and this permission notice appear in all copies.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+// SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
+// OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+// CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
+
+package main
+
+import (
+	"bytes"
+	"flag"
+	"io/ioutil"
+	"path/filepath"
+	"testing"
+)
+
+var (
+	testDataDir = flag.String("testdata", "testdata", "The path to the test data directory.")
+	update      = flag.Bool("update", false, "If true, update output files rather than compare them.")
+)
+
+type delocateTest struct {
+	name string
+	in   []string
+	out  string
+}
+
+func (test *delocateTest) Path(file string) string {
+	return filepath.Join(*testDataDir, test.name, file)
+}
+
+var delocateTests = []delocateTest{
+	{"ppc64le-GlobalEntry", []string{"in.s"}, "out.s"},
+	{"ppc64le-LoadToR0", []string{"in.s"}, "out.s"},
+	{"ppc64le-Sample2", []string{"in.s"}, "out.s"},
+	{"ppc64le-Sample", []string{"in.s"}, "out.s"},
+	{"ppc64le-TOCWithOffset", []string{"in.s"}, "out.s"},
+	{"x86_64-Basic", []string{"in.s"}, "out.s"},
+	{"x86_64-BSS", []string{"in.s"}, "out.s"},
+	{"x86_64-GOTRewrite", []string{"in.s"}, "out.s"},
+	{"x86_64-LabelRewrite", []string{"in1.s", "in2.s"}, "out.s"},
+	{"x86_64-Sections", []string{"in.s"}, "out.s"},
+}
+
+func TestDelocate(t *testing.T) {
+	for _, test := range delocateTests {
+		t.Run(test.name, func(t *testing.T) {
+			var inputs []inputFile
+			for i, in := range test.in {
+				inputs = append(inputs, inputFile{
+					index: i,
+					path:  test.Path(in),
+				})
+			}
+
+			if err := parseInputs(inputs); err != nil {
+				t.Fatalf("parseInputs failed: %s", err)
+			}
+
+			var buf bytes.Buffer
+			if err := transform(&buf, inputs); err != nil {
+				t.Fatalf("transform failed: %s", err)
+			}
+
+			if *update {
+				ioutil.WriteFile(test.Path(test.out), buf.Bytes(), 0666)
+			} else {
+				expected, err := ioutil.ReadFile(test.Path(test.out))
+				if err != nil {
+					t.Fatalf("could not read %q: %s", test.Path(test.out), err)
+				}
+				if !bytes.Equal(buf.Bytes(), expected) {
+					t.Errorf("delocated output differed. Wanted:\n%s\nGot:\n%s\n", expected, buf.Bytes())
+				}
+			}
+		})
+	}
+}