From 12cd99ae8d2e0d9e53e0b08d07b9b723b1664f07 Mon Sep 17 00:00:00 2001
From: Jaroslav Benc <jaroslav.benc@grafana.com>
Date: Wed, 14 Jun 2023 13:18:24 +0200
Subject: [PATCH] Pass testing status to the data source plugin extension
 (#69988)

---
 packages/grafana-data/src/types/pluginExtensions.ts         | 6 ++++++
 .../features/datasources/components/EditDataSource.test.tsx | 1 +
 .../app/features/datasources/components/EditDataSource.tsx  | 1 +
 3 files changed, 8 insertions(+)

diff --git a/packages/grafana-data/src/types/pluginExtensions.ts b/packages/grafana-data/src/types/pluginExtensions.ts
index 33029172501..c297e24ad69 100644
--- a/packages/grafana-data/src/types/pluginExtensions.ts
+++ b/packages/grafana-data/src/types/pluginExtensions.ts
@@ -121,6 +121,12 @@ export type PluginExtensionDataSourceConfigContext<JsonData extends DataSourceJs
   // Meta information about the datasource plugin
   dataSourceMeta: DataSourcePluginMeta;
 
+  // Testing status
+  testingStatus?: {
+    message?: string | null;
+    status?: string | null;
+  };
+
   // Can be used to update the `jsonData` field on the datasource
   // (Only updates the form, it still needs to be saved by the user)
   setJsonData: (jsonData: JsonData) => void;
diff --git a/public/app/features/datasources/components/EditDataSource.test.tsx b/public/app/features/datasources/components/EditDataSource.test.tsx
index 1c55294da8c..8ba9ff47546 100644
--- a/public/app/features/datasources/components/EditDataSource.test.tsx
+++ b/public/app/features/datasources/components/EditDataSource.test.tsx
@@ -359,6 +359,7 @@ describe('<EditDataSource>', () => {
       expect(props.context.dataSource).toBeDefined();
       expect(props.context.dataSourceMeta).toBeDefined();
       expect(props.context.setJsonData).toBeDefined();
+      expect(props.context.testingStatus).toBeDefined();
     });
   });
 });
diff --git a/public/app/features/datasources/components/EditDataSource.tsx b/public/app/features/datasources/components/EditDataSource.tsx
index e1ed826edcd..055f8ce3e1d 100644
--- a/public/app/features/datasources/components/EditDataSource.tsx
+++ b/public/app/features/datasources/components/EditDataSource.tsx
@@ -217,6 +217,7 @@ export function EditDataSourceView({
               context={{
                 dataSource: omit(dataSource, ['secureJsonData']),
                 dataSourceMeta: dataSourceMeta,
+                testingStatus,
                 setJsonData: (jsonData) =>
                   onOptionsChange({
                     ...dataSource,