Instrumentation: Handle context.Canceled (#75867)

Ref #68480 Co-authored-by: Giuseppe Guerra <giuseppe.guerra@grafana.com>
2023-10-10 12:28:39 +02:00
parent aee8c91ac8
commit 90631360eb
8 changed files with 72 additions and 0 deletions
@@ -41,6 +41,10 @@ func TestQueryData(t *testing.T) {
 				err:           errors.New("surprise surprise"),
 				expectedError: plugins.ErrPluginDownstreamErrorBase,
 			},
+			{
+				err:           context.Canceled,
+				expectedError: plugins.ErrPluginRequestCanceledErrorBase,
+			},
 		}

 		for _, tc := range tcs {
@@ -99,6 +103,10 @@ func TestCheckHealth(t *testing.T) {
 				err:           errors.New("surprise surprise"),
 				expectedError: plugins.ErrPluginHealthCheck,
 			},
+			{
+				err:           context.Canceled,
+				expectedError: plugins.ErrPluginRequestCanceledErrorBase,
+			},
 		}

 		for _, tc := range tcs {