Update unit tests for rendered list feature

commit-id:1f55083a
ejoffe · Dec 6, 2021 · b7d34c8 · b7d34c8
1 parent 9a0e69a
commit b7d34c8
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/spr/spr_test.go b/spr/spr_test.go
@@ -77,6 +77,7 @@ func TestSPRBasicFlowFourCommits(t *testing.T) {
 	gitmock.ExpectLogAndRespond([]*git.Commit{&c1})
 	gitmock.ExpectPushCommits([]*git.Commit{&c1})
 	githubmock.ExpectCreatePullRequest(c1, nil)
+	githubmock.ExpectUpdatePullRequest(c1, nil)
 	githubmock.ExpectGetInfo()
 	s.UpdatePullRequests(ctx)
 	fmt.Printf("OUT: %s\n", output.String())
@@ -89,6 +90,8 @@ func TestSPRBasicFlowFourCommits(t *testing.T) {
 	gitmock.ExpectLogAndRespond([]*git.Commit{&c2, &c1})
 	gitmock.ExpectPushCommits([]*git.Commit{&c2})
 	githubmock.ExpectCreatePullRequest(c2, &c1)
+	githubmock.ExpectUpdatePullRequest(c1, nil)
+	githubmock.ExpectUpdatePullRequest(c2, &c1)
 	githubmock.ExpectGetInfo()
 	s.UpdatePullRequests(ctx)
 	lines := strings.Split(output.String(), "\n")
@@ -104,6 +107,10 @@ func TestSPRBasicFlowFourCommits(t *testing.T) {
 	gitmock.ExpectPushCommits([]*git.Commit{&c3, &c4})
 	githubmock.ExpectCreatePullRequest(c3, &c2)
 	githubmock.ExpectCreatePullRequest(c4, &c3)
+	githubmock.ExpectUpdatePullRequest(c1, nil)
+	githubmock.ExpectUpdatePullRequest(c2, &c1)
+	githubmock.ExpectUpdatePullRequest(c3, &c2)
+	githubmock.ExpectUpdatePullRequest(c4, &c3)
 	githubmock.ExpectGetInfo()
 	s.UpdatePullRequests(ctx)
 	lines = strings.Split(output.String(), "\n")
@@ -163,6 +170,8 @@ func TestSPRAmendCommit(t *testing.T) {
 	gitmock.ExpectPushCommits([]*git.Commit{&c1, &c2})
 	githubmock.ExpectCreatePullRequest(c1, nil)
 	githubmock.ExpectCreatePullRequest(c2, &c1)
+	githubmock.ExpectUpdatePullRequest(c1, nil)
+	githubmock.ExpectUpdatePullRequest(c2, &c1)
 	githubmock.ExpectGetInfo()
 	s.UpdatePullRequests(ctx)
 	fmt.Printf("OUT: %s\n", output.String())
@@ -178,6 +187,7 @@ func TestSPRAmendCommit(t *testing.T) {
 	gitmock.ExpectFetch()
 	gitmock.ExpectLogAndRespond([]*git.Commit{&c2, &c1})
 	gitmock.ExpectPushCommits([]*git.Commit{&c2})
+	githubmock.ExpectUpdatePullRequest(c1, nil)
 	githubmock.ExpectUpdatePullRequest(c2, &c1)
 	githubmock.ExpectGetInfo()
 	s.UpdatePullRequests(ctx)
@@ -262,6 +272,10 @@ func TestSPRReorderCommit(t *testing.T) {
 	githubmock.ExpectCreatePullRequest(c2, &c1)
 	githubmock.ExpectCreatePullRequest(c3, &c2)
 	githubmock.ExpectCreatePullRequest(c4, &c3)
+	githubmock.ExpectUpdatePullRequest(c1, nil)
+	githubmock.ExpectUpdatePullRequest(c2, &c1)
+	githubmock.ExpectUpdatePullRequest(c3, &c2)
+	githubmock.ExpectUpdatePullRequest(c4, &c3)
 	githubmock.ExpectGetInfo()
 	s.UpdatePullRequests(ctx)
 	fmt.Printf("OUT: %s\n", output.String())
@@ -345,6 +359,10 @@ func TestSPRDeleteCommit(t *testing.T) {
 	githubmock.ExpectCreatePullRequest(c2, &c1)
 	githubmock.ExpectCreatePullRequest(c3, &c2)
 	githubmock.ExpectCreatePullRequest(c4, &c3)
+	githubmock.ExpectUpdatePullRequest(c1, nil)
+	githubmock.ExpectUpdatePullRequest(c2, &c1)
+	githubmock.ExpectUpdatePullRequest(c3, &c2)
+	githubmock.ExpectUpdatePullRequest(c4, &c3)
 	githubmock.ExpectGetInfo()
 
 	s.UpdatePullRequests(ctx)