From 17e4f4a2437b47784c9f7b65cba11db5b23631df Mon Sep 17 00:00:00 2001
From: Nazarii Denha <dengaaa2002@gmail.com>
Date: Thu, 26 Jan 2023 13:52:34 +0100
Subject: [PATCH] update test

---
 coordinator/manager_test.go | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/coordinator/manager_test.go b/coordinator/manager_test.go
index cfee25470..79eba42b3 100644
--- a/coordinator/manager_test.go
+++ b/coordinator/manager_test.go
@@ -504,14 +504,26 @@ func testVerifyProvedBatchesOnStartup(t *testing.T) {
 	<-time.After(5 * time.Second)
 	// the coordinator will delete the roller if the subscription is closed.
 	roller.close()
-
 	// Close rollerManager and ws handler.
 	handler.Shutdown(context.Background())
 	rollerManager.Stop()
 
 	// Setup new coordinator and ws server.
 	newRollerManager, newHandler := setupCoordinator(t, cfg.DBConfig, 1, wsURL, false)
-	// Wait for coordinator verify task
+	
+	// verify that task is proved
+	for i := range ids {
+		info, err := newRollerManager.GetSessionInfo(ids[i])
+		assert.Equal(t, orm.ProvingTaskAssigned.String(), info.Status)
+		assert.NoError(t, err)
+
+		// at this point, roller haven't submitted
+		status, err := l2db.GetProvingStatusByID(ids[i])
+		assert.NoError(t, err)
+		assert.Equal(t, orm.ProvingTaskProved, status)
+	}
+
+	// Wait for coordinator to verify task
 	<-time.After(6 * time.Second)
 	defer func() {
 		newHandler.Shutdown(context.Background())