[api] ServerV2.Stop() use correct context

iotexproject · Mar 28, 2023 · 0bf0c35 · 0bf0c35
1 parent a7ef663
commit 0bf0c35
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/api/coreservice_test.go b/api/coreservice_test.go
@@ -31,7 +31,7 @@ import (
 
 func TestLogsInRange(t *testing.T) {
 	require := require.New(t)
-	svr, _, _, _, cleanCallback := setupTestCoreSerivce()
+	svr, _, _, _, cleanCallback := setupTestCoreService()
 	defer cleanCallback()
 
 	t.Run("blocks with four logs", func(t *testing.T) {
@@ -107,7 +107,7 @@ func TestLogsInRange(t *testing.T) {
 }
 
 func BenchmarkLogsInRange(b *testing.B) {
-	svr, _, _, _, cleanCallback := setupTestCoreSerivce()
+	svr, _, _, _, cleanCallback := setupTestCoreService()
 	defer cleanCallback()
 
 	ctrl := gomock.NewController(b)
@@ -151,7 +151,7 @@ func getTopicsAddress(addr []string, topics [][]string) (iotexapi.LogsFilter, er
 	return filter, nil
 }
 
-func setupTestCoreSerivce() (CoreService, blockchain.Blockchain, blockdao.BlockDAO, actpool.ActPool, func()) {
+func setupTestCoreService() (CoreService, blockchain.Blockchain, blockdao.BlockDAO, actpool.ActPool, func()) {
 	cfg := newConfig()
 
 	// TODO (zhi): revise
@@ -188,7 +188,7 @@ func TestEstimateGasForAction(t *testing.T) {
 	require := require.New(t)
 	ctrl := gomock.NewController(t)
 	defer ctrl.Finish()
-	svr, _, _, _, cleanCallback := setupTestCoreSerivce()
+	svr, _, _, _, cleanCallback := setupTestCoreService()
 	defer cleanCallback()
 
 	estimatedGas, err := svr.EstimateGasForAction(context.Background(), getAction())

diff --git a/api/serverV2.go b/api/serverV2.go
@@ -100,7 +100,7 @@ func (svr *ServerV2) Start(ctx context.Context) error {
 // Stop stops the GRPC server and the CoreService
 func (svr *ServerV2) Stop(ctx context.Context) error {
 	if svr.tracer != nil {
-		if err := svr.tracer.Shutdown(context.Background()); err != nil {
+		if err := svr.tracer.Shutdown(ctx); err != nil {
 			return errors.Wrap(err, "failed to shutdown api tracer")
 		}
 	}