add pause/resume test

ChainSafe · Jun 15, 2021 · 749da75 · 749da75
1 parent 7a5ce9e
commit 749da75
Show file tree

Hide file tree

Showing 5 changed files with 342 additions and 0 deletions.
diff --git a/dot/network/mock_block_state.go b/dot/network/mock_block_state.go
diff --git a/dot/network/mock_syncer.go b/dot/network/mock_syncer.go
diff --git a/dot/network/mock_transaction_handler.go b/dot/network/mock_transaction_handler.go
diff --git a/lib/babe/babe_test.go b/lib/babe/babe_test.go
@@ -253,3 +253,33 @@ func TestStartAndStop(t *testing.T) {
 	err = bs.Stop()
 	require.NoError(t, err)
 }
+
+func TestService_PauseAndResume(t *testing.T) {
+	bs := createTestService(t, &ServiceConfig{
+		LogLvl: log.LvlCrit,
+	})
+	err := bs.Start()
+	require.NoError(t, err)
+	time.Sleep(time.Second)
+
+	go func() {
+		_ = bs.Pause()
+	}()
+
+	go func() {
+		_ = bs.Pause()
+	}()
+
+	go func() {
+		err := bs.Resume()
+		require.NoError(t, err)
+	}()
+
+	go func() {
+		err := bs.Resume()
+		require.NoError(t, err)
+	}()
+
+	err = bs.Stop()
+	require.NoError(t, err)
+}
diff --git a/lib/runtime/mock_transaction_state.go b/lib/runtime/mock_transaction_state.go