1 files changed, 139 insertions, 0 deletions
diff --git a/dashboard/app/api_test.go b/dashboard/app/api_test.go
index 8d63ce7a7..4a872231e 100644
--- a/dashboard/app/api_test.go
+++ b/dashboard/app/api_test.go
@@ -5,6 +5,9 @@ package main
 
 import (
 	"testing"
+	"time"
+
+	"github.com/google/syzkaller/dashboard/dashapi"
 )
 
 func TestClientSecretOK(t *testing.T) {
@@ -63,3 +66,139 @@ func TestClientNamespaceOK(t *testing.T) {
 		t.Errorf("Unexpected error %v %v", got, err)
 	}
 }
+
+func TestEmergentlyStoppedEmail(t *testing.T) {
+	c := NewCtx(t)
+	defer c.Close()
+
+	client := c.publicClient
+	build := testBuild(1)
+	client.UploadBuild(build)
+
+	crash := testCrash(build, 1)
+	client.ReportCrash(crash)
+
+	c.advanceTime(time.Hour)
+	_, err := c.AuthGET(AccessAdmin, "/admin?action=emergency_stop")
+	c.expectOK(err)
+
+	// There should be no email.
+	c.advanceTime(time.Hour)
+	c.expectNoEmail()
+}
+
+func TestEmergentlyStoppedReproEmail(t *testing.T) {
+	c := NewCtx(t)
+	defer c.Close()
+
+	client := c.publicClient
+	build := testBuild(1)
+	client.UploadBuild(build)
+
+	crash := testCrash(build, 1)
+	client.ReportCrash(crash)
+	c.pollEmailBug()
+
+	crash2 := testCrash(build, 1)
+	crash2.ReproOpts = []byte("repro opts")
+	crash2.ReproSyz = []byte("getpid()")
+	client.ReportCrash(crash2)
+
+	c.advanceTime(time.Hour)
+	_, err := c.AuthGET(AccessAdmin, "/admin?action=emergency_stop")
+	c.expectOK(err)
+
+	// There should be no email.
+	c.advanceTime(time.Hour)
+	c.expectNoEmail()
+}
+
+func TestEmergentlyStoppedExternalReport(t *testing.T) {
+	c := NewCtx(t)
+	defer c.Close()
+
+	client := c.client
+	build := testBuild(1)
+	client.UploadBuild(build)
+
+	crash := testCrash(build, 1)
+	client.ReportCrash(crash)
+
+	c.advanceTime(time.Hour)
+	_, err := c.AuthGET(AccessAdmin, "/admin?action=emergency_stop")
+	c.expectOK(err)
+
+	// There should be no email.
+	c.advanceTime(time.Hour)
+	client.pollBugs(0)
+}
+
+func TestEmergentlyStoppedEmailJob(t *testing.T) {
+	c := NewCtx(t)
+	defer c.Close()
+
+	client := c.publicClient
+	build := testBuild(1)
+	client.UploadBuild(build)
+
+	crash := testCrash(build, 1)
+	crash.ReproOpts = []byte("repro opts")
+	crash.ReproSyz = []byte("getpid()")
+	client.ReportCrash(crash)
+	sender := c.pollEmailBug().Sender
+	c.incomingEmail(sender, "#syz upstream\n")
+	sender = c.pollEmailBug().Sender
+
+	// Send a patch testing request.
+	c.advanceTime(time.Hour)
+	c.incomingEmail(sender, syzTestGitBranchSamplePatch,
+		EmailOptMessageID(1), EmailOptFrom("test@requester.com"),
+		EmailOptCC([]string{"somebody@else.com", "test@syzkaller.com"}))
+	c.expectNoEmail()
+
+	// Emulate a finished job.
+	pollResp := client.pollJobs(build.Manager)
+	c.expectEQ(pollResp.Type, dashapi.JobTestPatch)
+
+	c.advanceTime(time.Hour)
+	jobDoneReq := &dashapi.JobDoneReq{
+		ID:          pollResp.ID,
+		Build:       *build,
+		CrashTitle:  "test crash title",
+		CrashLog:    []byte("test crash log"),
+		CrashReport: []byte("test crash report"),
+	}
+	client.JobDone(jobDoneReq)
+
+	// Now we emergently stop syzbot.
+	c.advanceTime(time.Hour)
+	_, err := c.AuthGET(AccessAdmin, "/admin?action=emergency_stop")
+	c.expectOK(err)
+
+	// There should be no email.
+	c.advanceTime(time.Hour)
+	c.expectNoEmail()
+}
+
+func TestEmergentlyStoppedCrashReport(t *testing.T) {
+	c := NewCtx(t)
+	defer c.Close()
+
+	client := c.publicClient
+	build := testBuild(1)
+	client.UploadBuild(build)
+
+	// Now we emergently stop syzbot.
+	c.advanceTime(time.Hour)
+	_, err := c.AuthGET(AccessAdmin, "/admin?action=emergency_stop")
+	c.expectOK(err)
+
+	crash := testCrash(build, 1)
+	crash.ReproOpts = []byte("repro opts")
+	crash.ReproSyz = []byte("getpid()")
+	client.ReportCrash(crash)
+
+	listResp, err := client.BugList()
+	c.expectOK(err)
+	c.expectEQ(len(listResp.List), 0)
+}