1 files changed, 88 insertions, 1 deletions
diff --git a/syz-hub/state/state_test.go b/syz-hub/state/state_test.go
index bfdf4b3a7..0972ad6bc 100644
--- a/syz-hub/state/state_test.go
+++ b/syz-hub/state/state_test.go
@@ -4,13 +4,16 @@
 package state
 
 import (
+	"fmt"
 	"io/ioutil"
 	"os"
+	"path/filepath"
+	"runtime"
 	"testing"
 )
 
 func TestState(t *testing.T) {
-	dir, err := ioutil.TempDir("", "syz-gce-state-test")
+	dir, err := ioutil.TempDir("", "syz-hub-state-test")
 	if err != nil {
 		t.Fatalf("failed to create temp dir: %v", err)
 	}
@@ -24,4 +27,88 @@ func TestState(t *testing.T) {
 	if err == nil {
 		t.Fatalf("synced with unconnected manager")
 	}
+	calls := []string{"read", "write"}
+	if err := st.Connect("foo", false, calls, nil); err != nil {
+		t.Fatalf("Connect failed: %v", err)
+	}
+	_, _, err = st.Sync("foo", nil, nil)
+	if err != nil {
+		t.Fatalf("Sync failed: %v", err)
+	}
+}
+
+func TestRepro(t *testing.T) {
+	dir, err := ioutil.TempDir("", "syz-hub-state-test")
+	if err != nil {
+		t.Fatalf("failed to create temp dir: %v", err)
+	}
+	defer os.RemoveAll(dir)
+
+	st, err := Make(dir)
+	if err != nil {
+		t.Fatalf("failed to make state: %v", err)
+	}
+
+	if err := st.Connect("foo", false, []string{"open", "read", "write"}, nil); err != nil {
+		t.Fatalf("Connect failed: %v", err)
+	}
+	if err := st.Connect("bar", false, []string{"open", "read", "close"}, nil); err != nil {
+		t.Fatalf("Connect failed: %v", err)
+	}
+	checkPendingRepro(t, st, "foo", "")
+	checkPendingRepro(t, st, "bar", "")
+
+	if err := st.AddRepro("foo", []byte("open()")); err != nil {
+		t.Fatalf("AddRepro failed: %v", err)
+	}
+	checkPendingRepro(t, st, "foo", "")
+	checkPendingRepro(t, st, "bar", "open()")
+	checkPendingRepro(t, st, "bar", "")
+
+	// This repro is already present.
+	if err := st.AddRepro("bar", []byte("open()")); err != nil {
+		t.Fatalf("AddRepro failed: %v", err)
+	}
+	if err := st.AddRepro("bar", []byte("read()")); err != nil {
+		t.Fatalf("AddRepro failed: %v", err)
+	}
+	if err := st.AddRepro("bar", []byte("open()\nread()")); err != nil {
+		t.Fatalf("AddRepro failed: %v", err)
+	}
+	// This does not satisfy foo's call set.
+	if err := st.AddRepro("bar", []byte("close()")); err != nil {
+		t.Fatalf("AddRepro failed: %v", err)
+	}
+	checkPendingRepro(t, st, "bar", "")
+
+	// Check how persistence works.
+	st, err = Make(dir)
+	if err != nil {
+		t.Fatalf("failed to make state: %v", err)
+	}
+	if err := st.Connect("foo", false, []string{"open", "read", "write"}, nil); err != nil {
+		t.Fatalf("Connect failed: %v", err)
+	}
+	if err := st.Connect("bar", false, []string{"open", "read", "close"}, nil); err != nil {
+		t.Fatalf("Connect failed: %v", err)
+	}
+	checkPendingRepro(t, st, "bar", "")
+	checkPendingRepro(t, st, "foo", "read()")
+	checkPendingRepro(t, st, "foo", "open()\nread()")
+	checkPendingRepro(t, st, "foo", "")
+}
+
+func checkPendingRepro(t *testing.T, st *State, name, result string) {
+	repro, err := st.PendingRepro(name)
+	if err != nil {
+		t.Fatalf("\n%v: PendingRepro failed: %v", caller(1), err)
+	}
+	if string(repro) != result {
+		t.Fatalf("\n%v: PendingRepro returned %q, want %q", caller(1), string(repro), result)
+	}
+}
+
+func caller(skip int) string {
+	_, file, line, _ := runtime.Caller(skip + 1)
+	return fmt.Sprintf("%v:%v", filepath.Base(file), line)
 }