1 files changed, 95 insertions, 0 deletions
diff --git a/prog/encoding_test.go b/prog/encoding_test.go
new file mode 100644
index 000000000..f23a92cc7
--- /dev/null
+++ b/prog/encoding_test.go
@@ -0,0 +1,95 @@
+// Copyright 2016 syzkaller project authors. All rights reserved.
+// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
+
+package prog
+
+import (
+	"fmt"
+	"reflect"
+	"sort"
+	"testing"
+)
+
+func setToArray(s map[string]struct{}) []string {
+	a := make([]string, 0, len(s))
+	for c := range s {
+		a = append(a, c)
+	}
+	sort.Strings(a)
+	return a
+}
+
+func TestCallSet(t *testing.T) {
+	tests := []struct {
+		prog  string
+		ok    bool
+		calls []string
+	}{
+		{
+			"",
+			false,
+			[]string{},
+		},
+		{
+			"r0 =  (foo)",
+			false,
+			[]string{},
+		},
+		{
+			"getpid()",
+			true,
+			[]string{"getpid"},
+		},
+		{
+			"r11 =  getpid()",
+			true,
+			[]string{"getpid"},
+		},
+		{
+			"getpid()\n" +
+				"open(0x1, something that this package may not understand)\n" +
+				"getpid()\n" +
+				"#read()\n" +
+				"\n" +
+				"close$foo(&(0x0000) = {})\n",
+			true,
+			[]string{"getpid", "open", "close$foo"},
+		},
+	}
+	for i, test := range tests {
+		t.Run(fmt.Sprint(i), func(t *testing.T) {
+			calls, err := CallSet([]byte(test.prog))
+			if err != nil && test.ok {
+				t.Fatalf("parsing failed: %v", err)
+			}
+			if err == nil && !test.ok {
+				t.Fatalf("parsing did not fail")
+			}
+			callArray := setToArray(calls)
+			sort.Strings(test.calls)
+			if !reflect.DeepEqual(callArray, test.calls) {
+				t.Fatalf("got call set %+v, expect %+v", callArray, test.calls)
+			}
+		})
+	}
+}
+
+func TestCallSetRandom(t *testing.T) {
+	rs, iters := initTest(t)
+	for i := 0; i < iters; i++ {
+		p := Generate(rs, 10, nil)
+		calls0 := make(map[string]struct{})
+		for _, c := range p.Calls {
+			calls0[c.Meta.Name] = struct{}{}
+		}
+		calls1, err := CallSet(p.Serialize())
+		if err != nil {
+			t.Fatalf("CallSet failed: %v", err)
+		}
+		callArray0 := setToArray(calls0)
+		callArray1 := setToArray(calls1)
+		if !reflect.DeepEqual(callArray0, callArray1) {
+			t.Fatalf("got call set:\n%+v\nexpect:\n%+v", callArray1, callArray0)
+		}
+	}
+}